Upravil jsem funkci gen_soustredeni.

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
+        sousy = []
-	# TODO: pozor, organizatori a ucastnici jsou ManyToManyField, musí se přiřadit
+        for sousi in range(1, size): #FIXME Tu range si změňte jak chcete, nevím, co přesně znamená size (asi Anet?)
-	# mimo create() a pak dát save()
+            datum_zacatku=datetime.date(randint(2000, 2020), randint(1, 12), radint(1, 28))
-	sous = Soustredeni.objects.create(rocnik=Rocnik.objects.first(), 
+            working_sous = Soustredeni.objects.create(rocnik=Rocnik.objects.order_by('?').first(),
-		verejne_db=True, 
+            verejne_db=rnd.choice([True, False]),
-		misto=u'Někde',	
+            misto=rnd.choice(['Kremrolovice', 'Indiánov', 'U zmzliny', 'Vafláreň', 'Větrník', 'Horní Rakvička', 'Dolní cheesecake']),
-		datum_zacatku=datetime.date(2000, 11, 23), 
+            datum_zacatku=datum_zacatku,
-		datum_konce=datetime.date(2000, 11, 27)
+            datum_konce=datum_zacatku + datetime.timedelta(weeks=1))
-	)
+            for res in rnd.sample(resitele, min(len(resitele), 20)):
-	for res in rnd.sample(resitele, min(len(resitele), 20)):
+                Soustredeni_Ucastnici.objects.create(resitel=res, soutredeni=working_sous)
-		Soustredeni_Ucastnici.objects.create(resitel=res, soustredeni=sous)
+            for org in rnd.sample(organizatori, min(len(organizatori), 20)):
-	sous.save()
+                Soustredeni_Organizatori.objects.create(organizator=org, soutredeni=working_sous)
+            working_sous.save()
+            sousy.append(working_sous)
+        return sousy
 
 
 @transaction.atomic