aboutsummaryrefslogtreecommitdiffstats
path: root/client/views.py
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)