From c79cf3fe4da3dd535ccfa134a4cc7d370943ce42 Mon Sep 17 00:00:00 2001
From: "Martin Z. (Zimamazim)" <zimenm@seznam.cz>
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