testutils.py: náhodné úpravy

seminar/testutils.py

@@ -219,58 +219,22 @@ def gen_soustredeni():
 		Soustredeni_Ucastnici.objects.create(resitel=res, soustredeni=sous)
 	sous.save()
 
-
+def gen_rocniky(last_rocnik, size):
-@transaction.atomic
+	rocniky = []
-def create_test_data(size = 6, rnd = None):
+	node = None
-	assert size >= 1
-	# pevna pseudo-nahodnost
-	rnd = rnd or random.Random(x=42)
-
-	# static URL stranky
-	# FIXME: nakopirovat sem vsechny z produkcni databaze
-	s = Site.objects.filter(name="example.com")
-	f = FlatPage.objects.create(url="/", title="Seminář M&M", 
-		content = "<p>V&iacute;tejte na str&aacute;nce semin&aacute;ře MaM!</p>")
-	f.sites.add(s[0])
-	f.save()
-
-	# users
-	admin = User.objects.create_superuser(username='admin', email='', password='admin')
-
-	usernames = ['anet', 'bara', 'cyril', 'david', 'eva', 'filip']
-	users = []
-	for usr in usernames[:size]:
-		u = User.objects.create(username=usr, password=usr)
-		u.first_name = usr.capitalize()
-		u.save()
-		users.append(u)
-	print(users)
-
-	# skoly
-	skoly = gen_skoly()
-
-	# osoby
-	osoby = gen_osoby(rnd, size)
-
-	# resitele a organizatori	
-	last_rocnik = 25
-	resitele = gen_resitele(rnd, osoby, skoly)
-	organizatori = gen_organizatori(rnd, osoby, last_rocnik, users)
-
-	# prijemci
-	prijemci = gen_prijemci(rnd, osoby)
-	
-	global zlinska
-	zlinska.kontaktni_osoba=rnd.choice(osoby)
-	zlinska.save()
-
-	# rocniky
 	for ri in range(min(last_rocnik - size, 1), last_rocnik + 1):
-		r = Rocnik.objects.create(prvni_rok = 1993 + ri, rocnik = ri)
+		rocnik = Rocnik.objects.create(prvni_rok = 1993 + ri, rocnik = ri)
+		node2 = RocnikNode.objects.create(rocnik = rocnik, succ = node)
+		node = node2
+		rocniky.append(rocnik)
+	return rocniky
 
-		# cisla
+def gen_cisla(rocniky):
-		cisel = rnd.randint(4, 8)
+	slovnik_rocnik_cisla = {}
+	for rocnik in rocniky:
 		slovnik_cisel = {}
+		cisel = rnd.randint(4, 8)
+		node = None
 		for ci in range(1, cisel + 1):
 			# první číslo vydáváme typicky okolo prázdnin
 			# (ci - 1)*2 zaručuje první číslo v červnu a všechna
@@ -289,20 +253,25 @@ def create_test_data(size = 6, rnd = None):
 				deadline = None
 			
 			cislo = Cislo.objects.create(
-				rocnik = r,
+				rocnik = rocnik,
 				cislo = str(ci), 
 				datum_vydani=vydano,
 				datum_deadline=deadline, 
 				verejne_db=True
 			)
+			node2 = CisloNode.objects.create(cislo = cislo, succ = node)
+			node = node2
+			
 			slovnik_cisel[ci] = cislo
+		slovnik_rocnik_cisla[rocnik] = slovnik_cisel
+	return slovnik_rocnik_cisla
 
-			# generovani uloh
+def gen_temata(rocniky, slovnik_rocnik_cisla):
-			gen_ulohy_do_cisla(rnd, ci, organizatori, resitele, slovnik_cisel, size)
+	slovnik_rocnik_temata = {}
-			
+	for rocnik in rocniky:
-			# generování témat, zatím v prvních třech číslech po jednom
+		temata = {}
-			# FIXME: více témat
+		for cislo in slovnik_r[rocnik]
-			if ci <= 3:
+			if cislo <= 3:
 				jake = ["Hravé", "Fyzikální", "Nejlepší", "Totálně masakrální",
 					"Šokující", "Magnetické", "Modré", "Překvapivé", 
 					"Plasmatické", "Novoroční"]
@@ -319,56 +288,127 @@ def create_test_data(size = 6, rnd = None):
 					garant=rnd.choice(organizatori), 
 					# FIXME: téma má kód podle čísla, až jich bude
 					# více, tak udělat kód podle pořadí vygenerování
-					kod=str(ci),
+					kod=str(cislo),
- 					# atributy třídy Téma
+					# atributy třídy Téma
 					tema_typ=rnd.choice(Tema.TEMA_CHOICES),
-					rocnik=r)
+					rocnik=rocnik)
 
 				t.opravovatele=rnd.sample(organizatori, poc_op)
 				t.save()
-			
+				temata[cislo] = t
-				# FIXME: nefunguje, protože další čísla v tuhle chvíli ještě 
+		slovnik_rocnik_temata[tema] = temata
-				# 	nejsou nagenerovaná	
+
-				# nagenerovat k tématu úlohy a Nody pro vsechna cisla
+def gen_ulohy_k_tematum(rocniky, slovnik_rocnik_cisla, slovnik_rocnik_temata):
-				for i in range(ci, cisel-2):
+	for rocnik in rocniky:
-					# vytvoř úlohu pro každé ze zbývajících čísel
+		slovnik_cisel = slovnik_rocnik_cisla[rocnik]
-					poc_op = rnd.randint(1, 4)
+		cisel = len(slovnik_cisel)
-					poc_oboru = rnd.randint(1, 2)
+		slovnik_temat = slovnik_rocnik_temata[rocnik]
-					p = Uloha.objects.create(
+			for tema in slovnik_temat:
-						nazev=": ".join([t.nazev, 
+				for i in range(tema.
-							"úloha {}.".format(i-ci+1)]),
+					if #TODO
-						nadproblem=t,
+						poc_op = rnd.randint(1, 4)
-						stav=Problem.STAV_ZADANY,
+						poc_oboru = rnd.randint(1, 2)
-						zamereni=t.zamereni,
+						p = Uloha.objects.create(
-						autor=t.autor, 
+							nazev=": ".join([tema.nazev, 
-						garant=t.garant, 
+								"úloha {}.".format(i-ci+1)]),
-						opravovatele=rnd.sample(organizatori, poc_op),  
+							nadproblem=tema,
-						kod=str(i), 
+							stav=Problem.STAV_ZADANY,
-						cislo_zadani=slovnik_cisel[ci], 
+							zamereni=t.zamereni,
-						cislo_reseni=slovnik_cisel[ci+2], 
+							autor=t.autor, 
-						cislo_deadline=slovnik_cisel[ci+2], 
+							garant=t.garant, 
-						max_body = rnd.randint(1, 8))
+							opravovatele=rnd.sample(organizatori, poc_op),  
-					p.zadani = " ".join([rnd.choice(sloveso), rnd.choice(koho), 
+							kod=str(i), 
-						rnd.choice(ceho), rnd.choice(jmeno), rnd.choice(kde)])
+							cislo_zadani=slovnik_cisel[ci], 
-					
+							cislo_reseni=slovnik_cisel[ci+2], 
-					#Tomův pokus o opravu bugu:
+							cislo_deadline=slovnik_cisel[ci+2], 
-					text_vzoraku = Text.objects.create(
+							max_body = rnd.randint(1, 8))
-						na_web = rnd.choice(reseni),
+						p.zadani = " ".join([rnd.choice(sloveso), rnd.choice(koho), 
-						do_cisla = rnd.choice(reseni)
+							rnd.choice(ceho), rnd.choice(jmeno), rnd.choice(kde)])
-					)
+						
-					vzorak = TextNode.objects.create(text = text_vzoraku)
+						#Tomův pokus o opravu bugu:
-					uloha_vzorak = UlohaVzorakNode.objects.create(uloha=p, first_child = vzorak)
+						text_vzoraku = Text.objects.create(
-					p.UlohaVzorakNode = uloha_vzorak
+							na_web = rnd.choice(reseni),
-					
+							do_cisla = rnd.choice(reseni)
-					#p.vzorak = " - ".join([p.zadani, rnd.choice(reseni)]) 
+						)
-					p.save()
+						vzorak = TextNode.objects.create(text = text_vzoraku)
-					
+						uloha_vzorak = UlohaVzorakNode.objects.create(uloha=p, first_child = vzorak)
-					
+						p.UlohaVzorakNode = uloha_vzorak
-					# vytvor text
+						
-					# vytvor TemaVCisleNode
+						#p.vzorak = " - ".join([p.zadani, rnd.choice(reseni)]) 
-					if (rnd.randint(1, 3) % 3 == 0):
+						p.save()
-						# vytvor mezicislo Node
+				
-						pass
+				
+				# vytvor text
+				# vytvor TemaVCisleNode
+				if (rnd.randint(1, 3) % 3 == 0):
+					# vytvor mezicislo Node
+					pass
+
+@transaction.atomic
+def create_test_data(size = 6, rnd = None):
+	assert size >= 1
+	# pevna pseudo-nahodnost
+	rnd = rnd or random.Random(x=42)
+
+	# static URL stranky
+	# FIXME: nakopirovat sem vsechny z produkcni databaze
+	s = Site.objects.filter(name="example.com")
+	f = FlatPage.objects.create(url="/", title="Seminář M&M", 
+		content = "<p>V&iacute;tejte na str&aacute;nce semin&aacute;ře MaM!</p>")
+	f.sites.add(s[0])
+	f.save()
+
+	# users
+	admin = User.objects.create_superuser(username='admin', email='', password='admin')
+
+	usernames = ['anet', 'bara', 'cyril', 'david', 'eva', 'filip']
+	users = []
+	for usr in usernames[:size]:
+		u = User.objects.create(username=usr, password=usr)
+		u.first_name = usr.capitalize()
+		u.save()
+		users.append(u)
+	print(users)
+
+	# skoly
+	skoly = gen_skoly()
+
+	# osoby
+	osoby = gen_osoby(rnd, size)
+
+	# resitele a organizatori
+	last_rocnik = 25
+	resitele = gen_resitele(rnd, osoby, skoly)
+	organizatori = gen_organizatori(rnd, osoby, last_rocnik, users)
+
+	# prijemci
+	prijemci = gen_prijemci(rnd, osoby)
+	
+	global zlinska
+	zlinska.kontaktni_osoba=rnd.choice(osoby)
+	zlinska.save()
+
+	# rocniky
+	rocniky = gen_rocniky(last_rocnik, size)
+	
+	# cisla
+	slovnik_rocnik_cisla = gen_cisla(rocniky)	
+	
+	# generování obyčejných úloh do čísel
+	for rocnik in rocniky:
+		slovnik_cisel = slovnik_rocnik_cisla[rocnik]
+		for cislo in slovnik_cisel:
+			gen_ulohy_do_cisla(rnd, cislo, organizatori, resitele, slovnik_cisel, size)
+	
+	# generování témat, zatím v prvních třech číslech po jednom
+	# FIXME: více témat
+	slovnik_rocnik_temata = gen_temata(rocniky, slovnik_rocnik_cisla)
+	
+	# generování úloh k tématům ve všech číslech
+	gen_ulohy_k_tematum(rocniky, slovnik_rocnik_cisla, slovnik_rocnik_temata)
+	
+		# nagenerovat k tématu úlohy a Nody pro vsechna cisla
+