From c79cf3fe4da3dd535ccfa134a4cc7d370943ce42 Mon Sep 17 00:00:00 2001 From: "Martin Z. (Zimamazim)" Date: Tue, 11 Jun 2019 00:25:22 +0200 Subject: [PATCH] Upravil jsem funkci gen_soustredeni. --- seminar/testutils.py | 28 ++++++++++++++++------------ 1 file changed, 16 insertions(+), 12 deletions(-) diff --git a/seminar/testutils.py b/seminar/testutils.py index edd836fa..6c80e80d 100644 --- a/seminar/testutils.py +++ b/seminar/testutils.py @@ -59,6 +59,7 @@ def gen_osoby(rnd, size): datum_registrace = datetime.date(rnd.randint(2019, 2029), rnd.randint(1, 12), rnd.randint(1, 28)))) #TODO pridat foto male a velke. Jak? + # Pavel tvrdí, že to necháme a přidáme až do adminu return osoby @@ -206,18 +207,21 @@ def gen_ulohy_do_cisla(rnd, cislo, organizatori, resitele, slovnik_cisel, size): return def gen_soustredeni(): - # TODO: vice soustredeni a k nim nahodne podmnoziny organizatoru a ucastniku - # TODO: pozor, organizatori a ucastnici jsou ManyToManyField, musí se přiřadit - # mimo create() a pak dát save() - sous = Soustredeni.objects.create(rocnik=Rocnik.objects.first(), - verejne_db=True, - misto=u'Někde', - datum_zacatku=datetime.date(2000, 11, 23), - datum_konce=datetime.date(2000, 11, 27) - ) - for res in rnd.sample(resitele, min(len(resitele), 20)): - Soustredeni_Ucastnici.objects.create(resitel=res, soustredeni=sous) - sous.save() + sousy = [] + for sousi in range(1, size): #FIXME Tu range si změňte jak chcete, nevím, co přesně znamená size (asi Anet?) + datum_zacatku=datetime.date(randint(2000, 2020), randint(1, 12), radint(1, 28)) + working_sous = Soustredeni.objects.create(rocnik=Rocnik.objects.order_by('?').first(), + verejne_db=rnd.choice([True, False]), + misto=rnd.choice(['Kremrolovice', 'Indiánov', 'U zmzliny', 'Vafláreň', 'Větrník', 'Horní Rakvička', 'Dolní cheesecake']), + datum_zacatku=datum_zacatku, + datum_konce=datum_zacatku + datetime.timedelta(weeks=1)) + for res in rnd.sample(resitele, min(len(resitele), 20)): + Soustredeni_Ucastnici.objects.create(resitel=res, soutredeni=working_sous) + for org in rnd.sample(organizatori, min(len(organizatori), 20)): + Soustredeni_Organizatori.objects.create(organizator=org, soutredeni=working_sous) + working_sous.save() + sousy.append(working_sous) + return sousy @transaction.atomic