diff --git a/seminar/testutils.py b/seminar/testutils.py index 0eb04cf9..82688e37 100644 --- a/seminar/testutils.py +++ b/seminar/testutils.py @@ -13,7 +13,7 @@ from django.contrib.sites.models import Site User = django.contrib.auth.get_user_model() -def gen_osoby(size): +def gen_osoby(rnd, size): jmena_m = ['Aleš', 'Tomáš', 'Martin', 'Jakub', 'Petr', 'Lukáš', 'Cyril', 'Pavel Karel'] jmena_f = ['Eva', 'Karolína', 'Zuzana', 'Sylvie', 'Iva', 'Jana', 'Marie', 'Marta Iva', 'Shu Shan'] @@ -83,7 +83,7 @@ def gen_skoly(): kratky_nazev="GaSJŠspSJZ", je_zs=True, je_ss=True)) return skoly -def gen_resitele(osoby, skoly): +def gen_resitele(rnd, osoby, skoly): resitele = [] for os in osoby: rand = rnd.randint(0, 8) @@ -93,14 +93,14 @@ def gen_resitele(osoby, skoly): zasilat=rnd.choice(Resitel.ZASILAT_CHOICES))) return resitele -def gen_prijemci(osoby, kolik=10): +def gen_prijemci(rnd, osoby, kolik=10): prijemci = [] for i in range(kolik): rand_os = rnd.choice(osoby) - prijemci.add(Prijemce.objects.create(osoba=rand_os)) + prijemci.append(Prijemce.objects.create(osoba=rand_os)) return prijemci -def gen_organizatori(osoby, last_rocnik): +def gen_organizatori(rnd, osoby, last_rocnik): organizatori = [] for os in osoby: rand = rnd.randint(0, 8) @@ -108,15 +108,15 @@ def gen_organizatori(osoby, last_rocnik): pusobnost = rnd.randint(1, last_rocnik) od = 1993 + last_rocnik - pusobnost do = od + rnd.randint(1, 6) - # aktualni organizatori jeste nemaji vyplnene organizuje_do - #if do > datetime.datetime.now().year: - # do = None - #organizatori.append(Organizator.objects.create(osoba=os, - # user=rnd.choice(users), - # organizuje_od=od, organizuje_do=do)) - return organizatori - -def gen_ulohy_do_cisla(cislo, organizatori, resitele, slovnik_cisel, size): + #aktualni organizatori jeste nemaji vyplnene organizuje_do + if do > datetime.datetime.now().year: + do = None + organizatori.append(Organizator.objects.create(osoba=os, + user=rnd.choice(users), + organizuje_od=od, organizuje_do=do)) + return organizatri + +def gen_ulohy_do_cisla(rnd, cislo, organizatori, resitele, slovnik_cisel, size): '''cislo = cislo cisla v rocniku, nikoli objekt Cislo ''' # ulohy resene v ci jaka = ["Šachová", "Černá", "Větrná", "Dlouhá", "Křehká", "Rychlá", @@ -202,15 +202,15 @@ def create_test_data(size = 6, rnd = None): skoly = gen_skoly() # osoby - osoby = gen_osoby(size) + osoby = gen_osoby(rnd, size) # resitele a organizatori last_rocnik = 25 - resitele = gen_resitele(osoby) - organizatori = gen_organizatori(osoby) + resitele = gen_resitele(rnd, osoby, skoly) + organizatori = gen_organizatori(rnd, osoby, last_rocnik) # prijemci - prijemci = gen_prijemci() + prijemci = gen_prijemci(rnd) zlinska.kontaktni_osoba=rnd.choice(osoby) zlinska.save() @@ -240,7 +240,7 @@ def create_test_data(size = 6, rnd = None): slovnik_cisel[ci] = cislo # generovani uloh - gen_ulohy_do_cisla(ci, organizatori, resitele, size) + gen_ulohy_do_cisla(rnd, ci, organizatori, resitele, slovnik_cisel, size) # generovani temat if ci <= 3: