mam/mamweb
13
1
Fork 0
Code Issues Pull requests 12 Projects Wiki Activity Actions

Přidání skupin org a resitel testdatům

Browse source
This commit is contained in:
Jonas Havelka 2022-01-31 20:18:08 +01:00
parent f15dcf2f41
commit bc07d195bf
1 changed files with 5 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
seminar/testutils.py
View file

@ -3,6 +3,7 @@
import datetime import datetime
from django.contrib.auth.models import Permission from django.contrib.auth.models import Permission
from django.contrib.auth.models import Group
from pytz import timezone from pytz import timezone
import random import random
import lorem import lorem
@ -129,6 +130,7 @@ def gen_resitele(rnd, osoby, skoly):
resitele = [] resitele = []
x = 0 x = 0
resitel_perm = Permission.objects.filter(codename__exact='resitel').first() resitel_perm = Permission.objects.filter(codename__exact='resitel').first()
resitel_group = Group.objects.filter(name__exact='resitel').first()
for os in osoby: for os in osoby:
rand = rnd.randint(0, 8) rand = rnd.randint(0, 8)
if not (rand % 8 == 0): if not (rand % 8 == 0):
@ -141,6 +143,7 @@ def gen_resitele(rnd, osoby, skoly):
os.user = user os.user = user
os.save() os.save()
os.user.user_permissions.add(resitel_perm) os.user.user_permissions.add(resitel_perm)
os.user.groups.add(resitel_group)
resitele.append(Resitel.objects.create(osoba=os, skola=rnd.choice(skoly), resitele.append(Resitel.objects.create(osoba=os, skola=rnd.choice(skoly),
rok_maturity=os.datum_narozeni.year + rnd.randint(18, 21), rok_maturity=os.datum_narozeni.year + rnd.randint(18, 21),
zasilat=rnd.choice(Resitel.ZASILAT_CHOICES)[0])) zasilat=rnd.choice(Resitel.ZASILAT_CHOICES)[0]))
@ -165,6 +168,7 @@ def gen_organizatori(rnd, osoby, last_rocnik):
x = 0 x = 0
org_perm = Permission.objects.filter(codename__exact='org').first() org_perm = Permission.objects.filter(codename__exact='org').first()
org_group = Group.objects.filter(name__exact='org').first()
for os in osoby: for os in osoby:
rand = rnd.randint(0, 8) rand = rnd.randint(0, 8)
if (rand % 8 == 0): if (rand % 8 == 0):
@ -200,6 +204,7 @@ def gen_organizatori(rnd, osoby, last_rocnik):
os.user = user os.user = user
os.save() os.save()
os.user.user_permissions.add(org_perm) os.user.user_permissions.add(org_perm)
os.user.groups.add(org_group)
os.user.is_staff = True os.user.is_staff = True
os.user.save() os.user.save()
organizatori.append(Organizator.objects.create(osoba=os, organizatori.append(Organizator.objects.create(osoba=os,