testutils.py: drobné změny

seminar/testutils.py

@@ -7,7 +7,7 @@ import django.contrib.auth
 from django.db import transaction
 import unidecode
 
-from seminar.models import Skola, Resitel, Rocnik, Cislo, Problem, Reseni, PrilohaReseni, Nastaveni, Soustredeni, Soustredeni_Ucastnici, Soustredeni_Organizatori, Osoba, Organizator, Prijemce, Tema, Uloha, Konfera, KonferaNode, RocnikNode
+from seminar.models import Skola, Resitel, Rocnik, Cislo, Problem, Reseni, PrilohaReseni, Nastaveni, Soustredeni, Soustredeni_Ucastnici, Soustredeni_Organizatori, Osoba, Organizator, Prijemce, Tema, Uloha, Konfera, KonferaNode, TextNode, UlohaVzorakNode, RocnikNode, CisloNode, TemaVCisleNode
 
 from django.contrib.flatpages.models import FlatPage
 from django.contrib.sites.models import Site
@@ -148,7 +148,7 @@ def gen_ulohy_do_cisla(rnd, cislo, organizatori, resitele, slovnik_cisel, size):
 		"sporné", "nepopsatelně jednoduché", "pokud jste na to nepřišli,"
 		"tak jste fakt hloupí"]
 				
-	if cislo >= 3:
+	if cislo >= 3: #TODO nagenerovat i nody!!!
 		for pi in range(1, ((size + 1) // 2) + 1):
 			poc_op = rnd.randint(1, 4) # počet opravovatelů
 			poc_oboru = rnd.randint(1, 2)
@@ -371,6 +371,9 @@ def gen_ulohy_k_tematum(rocniky, slovnik_rocnik_cisla, slovnik_rocnik_temata): #
 			for cislo in slovnik_cisel:
 				syn = cislo.CisloNode.first_child
 				while syn != None:
+					if type(syn) != TemaVCisleNode
+						continue
+					else:
 						tema = syn.tema
 					for i in range(1, rnd.randint(1, 4))
 						poc_op = rnd.randint(1, 4)
@@ -393,7 +396,6 @@ def gen_ulohy_k_tematum(rocniky, slovnik_rocnik_cisla, slovnik_rocnik_temata): #
 						p.zadani = " ".join([rnd.choice(sloveso), rnd.choice(koho), 
 							rnd.choice(ceho), rnd.choice(jmeno), rnd.choice(kde)])
 						
-                                        #Tomův pokus o opravu bugu:
 						text_vzoraku = Text.objects.create(
 							na_web = rnd.choice(reseni),
 							do_cisla = rnd.choice(reseni)
@@ -401,11 +403,10 @@ def gen_ulohy_k_tematum(rocniky, slovnik_rocnik_cisla, slovnik_rocnik_temata): #
 						vzorak = TextNode.objects.create(text = text_vzoraku)
 						uloha_vzorak = UlohaVzorakNode.objects.create(uloha=p, first_child = vzorak)
 						p.UlohaVzorakNode = uloha_vzorak
-                                        
-                                        #p.vzorak = " - ".join([p.zadani, rnd.choice(reseni)]) 
 						p.save()
 						
 						
+				
 				syn = syn.succ
 				
 				
@@ -482,11 +483,6 @@ def create_test_data(size = 6, rnd = None):
 	# 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
-		
-					
-				
-
 			# FIXME: misto typu ruzne typy objektu a vnoreni do sebe
 			# TODO: vytvorit temata s ruznymi vlakny
 			# TODO: nagenerovat starsim rocnikum pohadku