blob: 5a8f80890b3e43e7ff3c97d30bf3bc66a5167e59 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
|
from random import randint
from django.shortcuts import render
from .models import UserCode
# Create your views here.
def index(request):
context = {}
return render(request, 'client/index.html', context)
def approve(request):
context = {}
if not (request.user.is_staff or request.user.has_perm('approved')):
users = UserCode.objects.filter(user=request.user)
code = None
if (len(users) == 0):
while (True):
code = randint(100000, 999999)
if len(UserCode.objects.filter(code=code)) == 0:
break
userCode = UserCode(user=request.user, code=code)
userCode.save()
else:
code = UserCode.objects.filter(user=request.user)[0].code
context = {'code': 'U' + str(code), }
return render(request, 'client/approve.html', context)
else:
return render(request, 'client/index.html', context)
|