From bc07d195bfc17c1110e5c64a2277d83e8a7d0fef Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jon=C3=A1=C5=A1=20Havelka?= Date: Mon, 31 Jan 2022 20:18:08 +0100 Subject: [PATCH] =?UTF-8?q?P=C5=99id=C3=A1n=C3=AD=20skupin=20org=20a=20res?= =?UTF-8?q?itel=20testdat=C5=AFm?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- seminar/testutils.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/seminar/testutils.py b/seminar/testutils.py index 8ab8fd5f..efeea49f 100644 --- a/seminar/testutils.py +++ b/seminar/testutils.py @@ -3,6 +3,7 @@ import datetime from django.contrib.auth.models import Permission +from django.contrib.auth.models import Group from pytz import timezone import random import lorem @@ -129,6 +130,7 @@ def gen_resitele(rnd, osoby, skoly): resitele = [] x = 0 resitel_perm = Permission.objects.filter(codename__exact='resitel').first() + resitel_group = Group.objects.filter(name__exact='resitel').first() for os in osoby: rand = rnd.randint(0, 8) if not (rand % 8 == 0): @@ -141,6 +143,7 @@ def gen_resitele(rnd, osoby, skoly): os.user = user os.save() os.user.user_permissions.add(resitel_perm) + os.user.groups.add(resitel_group) resitele.append(Resitel.objects.create(osoba=os, skola=rnd.choice(skoly), rok_maturity=os.datum_narozeni.year + rnd.randint(18, 21), zasilat=rnd.choice(Resitel.ZASILAT_CHOICES)[0])) @@ -165,6 +168,7 @@ def gen_organizatori(rnd, osoby, last_rocnik): x = 0 org_perm = Permission.objects.filter(codename__exact='org').first() + org_group = Group.objects.filter(name__exact='org').first() for os in osoby: rand = rnd.randint(0, 8) if (rand % 8 == 0): @@ -200,6 +204,7 @@ def gen_organizatori(rnd, osoby, last_rocnik): os.user = user os.save() os.user.user_permissions.add(org_perm) + os.user.groups.add(org_group) os.user.is_staff = True os.user.save() organizatori.append(Organizator.objects.create(osoba=os,