seminar/testutils.py

novinky/testutils.py

@@ -0,0 +1,27 @@
+import logging
+
+from .models import Novinky
+
+logger = logging.getLogger(__name__)
+
+
+def gen_novinky(rnd, organizatori):
+	logger.info('Generuji novinky...')
+
+	jake = ["zábavné", "veselé", "dobrodružné", "skvělé"]
+	co = ["soustředění", "Fyziklání", "víkendové setkání"]
+	kde = ["na Šumavě", "v Praze", "u Plzně", "na Marsu"]
+	kdy = ["Zítra bude", "10. 10. 2020 bude", "V prosinci bude", "V létě bude"]
+
+	for i in range(5):
+		text_novinky = " ".join([
+			rnd.choice(kdy), rnd.choice(kde),
+			rnd.choice(jake), rnd.choice(co),
+		])
+		novinka = Novinky.objects.create(
+			id=i, autor=rnd.choice(organizatori),
+			text=(text_novinky+", těšíme se na vás!"),
+			zverejneno=rnd.choice([True, False]),
+		)
+		novinka.save()
+	return

odevzdavatko/testutils.py

@@ -0,0 +1,40 @@
+import datetime
+import random
+
+from seminar.models.odevzdavatko import Reseni, Hodnoceni
+
+
+def gen_reseni_ulohy(rnd, cisla, uloha, pocet_resitelu, poradi_cisla, resitele_cisla, resitele):
+	pocet_reseni = rnd.randint(pocet_resitelu//4, pocet_resitelu * 4)
+	# generujeme náhodný počet řešení vzhledem k počtu řešitelů čísla
+	for _ in range(pocet_reseni):
+		#print("Generuji {}-té řešení".format(reseni))
+		if rnd.randint(1, 10) == 1:
+			# cca desetina řešení od více řešitelů
+			res_vyber = rnd.sample(resitele_cisla, rnd.randint(2, 5))
+		else:
+			res_vyber = rnd.sample(resitele_cisla, 1)
+		if resitele[0] in res_vyber: # speciální řešitel, který nemá žádné body
+			res_vyber.remove(resitele[0])
+
+		# Vytvoření řešení.
+		if uloha.cislo_zadani.zlomovy_deadline_pro_papirove_cislo() is not None:
+			# combine, abychom dostali plný čas a ne jen datum
+			cas_doruceni = uloha.cislo_zadani.deadline_v_cisle.first().deadline - datetime.timedelta(days=random.randint(0, 40)) - datetime.timedelta(minutes=random.randint(0, 60*24))
+			# astimezone, protože jinak vyhazuje warning o nenastavené TZ
+			res = Reseni.objects.create(forma=rnd.choice(Reseni.FORMA_CHOICES)[0], cas_doruceni=cas_doruceni.astimezone(datetime.timezone.utc))
+		else:
+			res = Reseni.objects.create(forma=rnd.choice(Reseni.FORMA_CHOICES)[0])
+		# Problém a řešitele přiřadíme později, ManyToManyField
+		# se nedá vyplnit v create().
+		res.resitele.set(res_vyber)
+		res.save()
+
+		# Vytvoření hodnocení.
+		hod = Hodnoceni.objects.create(
+			body=rnd.randint(0, uloha.max_body),
+			cislo_body=cisla[poradi_cisla - 1],
+			reseni=res,
+			problem=uloha
+		)
+	return

personalni/testutils.py

@@ -0,0 +1,235 @@
+import datetime
+import logging
+import unidecode
+
+from django.contrib.auth.models import Permission
+from django.contrib.auth.models import Group
+import django.contrib.auth
+
+from .models import Osoba, Skola, Organizator, Resitel, Prijemce
+
+logger = logging.getLogger(__name__)
+
+User = django.contrib.auth.get_user_model()
+
+zlinska = None  # tohle bude speciální škola, které později dodáme kontaktní osobu
+
+
+# testuje unikátnost vygenerovaného jména
+def __unikatni_jmeno(osoby, jmeno, prijmeni):
+	for os in osoby:
+		if os.jmeno == jmeno and os.prijmeni == prijmeni:
+			return 0
+	else:
+		return 1
+
+
+def gen_osoby(rnd, size):
+	logger.info('Generuji osoby (size={})...'.format(size))
+
+	jmena_m = ['Aleš', 'Tomáš', 'Martin', 'Jakub', 'Petr', 'Lukáš', 'Cyril', 'Pavel Karel']
+	jmena_f = ['Eva', 'Karolína', 'Zuzana', 'Sylvie', 'Iva', 'Jana', 'Marie', 'Marta Iva', 'Shu Shan']
+	prijmeni_m = ['Novotný', 'Svoboda', 'Pecha', 'Kořen', 'Holan', 'Uhlíř', 'Chytráček', 'Pokora', 'Koch', 'Szegedy', 'Rudý', "von Neumann", "d'Este"]
+	prijmeni_f = ['Novotná', 'Svobodová', 'Machová', 'Zelená', 'Yu-Xin', 'Mlsná', 'Dubná', 'Mrkvová', 'Suchá', 'Lovelace', 'Holcová', 'Rui', "Nováčková Tydlitátová"]
+	prezdivky = ['Kaki', 'Hurdur', 'Maracuja', 'Bobbo', "", "", "", "", "", "", "", 'Riki', 'Sapa', "", '', '---', 'Koko']
+	domain = ['example.com', 'dolujeme.eu', 'mff.cuni.cz', 'strcprstskrzkrk.cz', 'british.co.uk', 'splachni.to', 'haha.org']
+	seznam_ulic = ['Krátká', 'Vlhká', 'Jungmanova', '17. listopadu', '4. října', 'Roztocká', 'Forstova', 'Generála Františka Janouška', 'Náměstí Války', 'Svratecké náměstí', 'Zelená lhota', 'Z Plynu', 'K Jezeru', 'U Kocourkova', 'Uštěpačná', 'Ostrorepská', 'Zubří']
+	seznam_mest = ['Praha', 'Brno', 'Ostrava', 'Horní Jelení', 'Dolní Zábrdovice', 'Prdelkov', 'Stará myslivna', 'Kocourkov', 'Šalingrad', 'Medvědí hora', 'Basilej', 'Unterschiedlich', 'Old York', 'Lancastershire', 'Vóloďháza']
+
+	osoby = []
+	# 30 je náhodná konstanta, size je použité na víc místech a
+	# říká, jak velká asi chceme testovací data
+	for i in range(30 * size):
+		pohlavi_idx = rnd.randint(0, 2)  # 2 = nebinární
+		osloveni = [Osoba.OSLOVENI_MUZSKE, Osoba.OSLOVENI_ZENSKE, Osoba.OSLOVENI_ZADNE][pohlavi_idx]
+		jmeno = rnd.choice([jmena_m, jmena_f, jmena_m + jmena_f][pohlavi_idx])
+		prijmeni = rnd.choice([prijmeni_m, prijmeni_f, prijmeni_m + prijmeni_f][pohlavi_idx])
+		if pohlavi_idx == 2: logger.debug(f'Testdata: nebinární osoba: {jmeno} {prijmeni}.')
+		pokusy = 0
+		max_pokusy = 120*size
+		while not __unikatni_jmeno and pokusy < max_pokusy:
+			# pokud jméno a příjmení není unikátní, zkoušíme generovat nová
+			# do daného limitu (abychom se nezacyklili do nekonečna při málo jménech a příjmeních
+			# ze kterých se generuje)
+			jmeno = rnd.choice([jmena_m, jmena_f, jmena_m + jmena_f][pohlavi_idx])
+			prijmeni = rnd.choice([prijmeni_m, prijmeni_f, prijmeni_m + prijmeni_f][pohlavi_idx])
+			pokusy += 1
+		if pokusy >= max_pokusy:
+			print("Chyba, na danou velikost testovacích dat příliš málo možných jmen a příjmení")
+			exit()
+		prezdivka = rnd.choice(prezdivky)
+		email = "@".join([unidecode.unidecode(jmeno), rnd.choice(domain)])
+		telefon = "".join([str(rnd.choice([k for k in range(10)])) for _ in range(9)])
+		narozeni = datetime.date(
+			rnd.randint(1980, datetime.datetime.now().year),
+			rnd.randint(1, 12),
+			rnd.randint(1, 28)
+		)
+		ulic = rnd.choice(seznam_ulic)
+		cp = rnd.randint(1, 99)
+		ulice = " ".join([ulic, str(cp)])
+		mesto = rnd.choice(seznam_mest)
+		psc = "".join([str(rnd.choice([k for k in range(10)])) for _ in range(5)])
+
+		osoby.append(Osoba.objects.create(
+			jmeno=jmeno, prijmeni=prijmeni,
+			prezdivka=prezdivka, osloveni=osloveni,
+			email=email, telefon=telefon,
+			datum_narozeni=narozeni,
+			ulice=ulice, mesto=mesto, psc=psc,
+			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
+
+
+def gen_skoly():  # TODO někdy to přepsat, aby jich bylo více
+	logger.info('Generuji školy...')
+
+	skoly = []
+	prvnizs = Skola.objects.create(
+		mesto='Praha', stat='CZ', psc='101 00',
+		ulice='Krátká 5', nazev='První ZŠ', je_zs=True, je_ss=False,
+	)
+	skoly.append(prvnizs)
+	skoly.append(Skola.objects.create(
+		mesto='Praha', stat='CZ', psc='101 00',
+		ulice='Krátká 5', nazev='První SŠ', je_zs=False, je_ss=True,
+	))
+	skoly.append(Skola.objects.create(
+		mesto='Praha', stat='CZ', psc='102 00',
+		ulice='Dlouhá 5', nazev='Druhá SŠ', je_zs=False, je_ss=True,
+	))
+	skoly.append(Skola.objects.create(
+		mesto='Praha', stat='CZ', psc='103 00',
+		ulice='Široká 3', nazev='Třetí SŠ a ZŠ', je_zs=True, je_ss=True,
+	))
+	skoly.append(Skola.objects.create(
+		mesto='Ostrava', stat='CZ', psc='700 00',
+		ulice='Hluboká 42', nazev='Hutní gympl', je_zs=False, je_ss=True,
+	))
+	skoly.append(Skola.objects.create(
+		mesto='Humenné', stat='SK', psc='012 34',
+		ulice='Pltká 1', nazev='Sredná škuola', je_zs=False, je_ss=True,
+	))
+	global zlinska
+	zlinska = Skola.objects.create(
+		mesto='Zlín', stat='CZ', psc='76001',
+		ulice='náměstí T.G. Masaryka 2734-9',
+		nazev='Gymnázium a Střední jazyková škola s právem SJZ',
+		kratky_nazev="GaSJŠspSJZ", je_zs=True, je_ss=True,
+	)
+	skoly.append(zlinska)
+	return skoly
+
+
+def gen_resitele(rnd, osoby, skoly):
+	logger.info('Generuji řešitele...')
+
+	resitele = []
+	x = 0
+	resitel_perm = Permission.objects.filter(codename__exact='resitel').first()
+	resitel_group = Group.objects.filter(name__exact='resitel').first()
+	for os in osoby:
+		rand = rnd.randint(0, 8)
+		if not (rand % 8 == 0):
+			if not os.user:
+				if x:
+					user = User.objects.create_user(
+						username='r'+str(x), email=os.email, password='r',
+					)
+				else:
+					user = User.objects.create_user(
+						username='r', email=os.email, password='r',
+					)
+				x += 1
+				os.user = user
+				os.save()
+				os.user.user_permissions.add(resitel_perm)
+				os.user.groups.add(resitel_group)
+			resitele.append(Resitel.objects.create(
+				osoba=os, skola=rnd.choice(skoly),
+				rok_maturity=os.datum_narozeni.year + rnd.randint(18, 21),
+				zasilat=rnd.choice(Resitel.ZASILAT_CHOICES)[0],
+			))
+	return resitele
+
+
+def gen_prijemci(rnd, osoby, kolik=10):
+	logger.info('Generuji příjemce (kolik={})...'.format(kolik))
+	prijemci = []
+	for i in rnd.sample(osoby, kolik):
+		prijemci.append(Prijemce.objects.create(osoba=i))
+
+	global zlinska
+	if zlinska is not None:
+		zlinska.kontaktni_osoba=rnd.choice(osoby)
+		zlinska.save()
+
+	return prijemci
+
+
+def gen_organizatori(rnd, osoby, last_rocnik):
+	logger.info('Generuji organizátory...')
+	organizatori = []
+
+
+	seznam_konicku = ["vařím", "jezdím na kole", "řeším diferenciální rovnice", "koukám z okna", "tancuji", "programuji", "jezdím vlakem", "nedělám nic"]
+	seznam_oboru = ["matematiku", "matematiku", "matematiku", "fyziku", "literaturu", "informatiku", "informatiku", "běhání dokolečka"]
+
+	x = 0
+	org_perm = Permission.objects.filter(codename__exact='org').first()
+	org_group = Group.objects.filter(name__exact='org').first()
+	for os in osoby:
+		rand = rnd.randint(0, 8)
+		if rand % 8 == 0:
+			pusobnost = rnd.randint(1, last_rocnik)
+			od = datetime.datetime(
+				year=1993 + pusobnost,
+				month=rnd.randint(1, 12),
+				day=rnd.randint(1, 28),
+				tzinfo=datetime.timezone.utc,
+			)
+			do = datetime.datetime(
+				year=od.year + rnd.randint(1, 6),
+				month=rnd.randint(1, 12),
+				day=rnd.randint(1, 28),
+				tzinfo=datetime.timezone.utc,
+			)
+			# aktualni organizatori jeste nemaji vyplnene organizuje_do
+
+			# popis orga
+			konicek1 = rnd.choice(seznam_konicku)
+			konicek2 = rnd.choice(seznam_konicku)
+			obor = rnd.choice(seznam_oboru)
+			popis_orga = "Ve volném čase " + konicek1 + " a také " + konicek2 + ". Studuji " + obor + " a moc mě to baví."
+
+			if do.year > datetime.datetime.now().year:
+				do = None
+			if not os.user:
+				if x:
+					user = User.objects.create_user(
+						username='o'+str(x), email=os.email, password='o',
+					)
+				else:
+					user = User.objects.create_user(
+						username='o', email=os.email, password='o',
+					)
+				x += 1
+				os.user = user
+				os.save()
+				os.user.user_permissions.add(org_perm)
+				os.user.groups.add(org_group)
+				os.user.is_staff = True
+				os.user.save()
+			organizatori.append(Organizator.objects.create(
+				osoba=os,
+				organizuje_od=od, organizuje_do=do,
+				strucny_popis_organizatora=popis_orga,
+			))
+	return organizatori

seminar/testutils.py → tvorba/testutils.py

@@ -1,216 +1,22 @@
+# FIXME vypreparovat treenode
+
 import datetime
 
-from django.contrib.auth.models import Permission
-from django.contrib.auth.models import Group
-import random
 import lorem
 import django.contrib.auth
-from django.db import transaction
-import unidecode
 import logging
 
-from korektury.testutils import create_test_pdf
-from seminar.models import Skola, Resitel, Rocnik, Cislo, Deadline, Problem, Reseni, PrilohaReseni, Nastaveni, Osoba, Organizator, Prijemce, Tema, Uloha, TextNode, UlohaVzorakNode, RocnikNode, CisloNode, TemaVCisleNode, Text, Hodnoceni, UlohaZadaniNode, Novinky, TreeNode
+from seminar.models import Rocnik, Cislo, Deadline, Problem, Tema, Uloha, TextNode, UlohaVzorakNode, RocnikNode, CisloNode, TemaVCisleNode, Text, UlohaZadaniNode
 import seminar.models as m
 
-from django.contrib.flatpages.models import FlatPage
-from django.contrib.sites.models import Site
 from treenode.treelib import all_children, insert_last_child, all_children_of_type, create_node_after
 
-from soustredeni.testutils import gen_soustredeni, gen_konfery
+from odevzdavatko.testutils import gen_reseni_ulohy
 
+logger = logging.getLogger(__name__)
 
 User = django.contrib.auth.get_user_model()
-zlinska = None # tohle bude speciální škola, které později dodáme kontaktní osobu
 
-logger = logging.getLogger(__name__)
-
-# testuje unikátnost vygenerovaného jména
-def __unikatni_jmeno(osoby, jmeno, prijmeni):
-	for os in osoby:
-		if os.jmeno == jmeno and os.prijmeni == prijmeni:
-			return 0
-	else: return 1
-
-def gen_osoby(rnd, size):
-	logger.info('Generuji osoby (size={})...'.format(size))
-
-	jmena_m = ['Aleš', 'Tomáš', 'Martin', 'Jakub', 'Petr', 'Lukáš', 'Cyril', 'Pavel Karel']
-	jmena_f = ['Eva', 'Karolína', 'Zuzana', 'Sylvie', 'Iva', 'Jana', 'Marie',
-		'Marta Iva', 'Shu Shan']
-	prijmeni_m = ['Novotný', 'Svoboda', 'Pecha', 'Kořen', 'Holan', 'Uhlíř', 'Chytráček',
-			'Pokora', 'Koch', 'Szegedy', 'Rudý', "von Neumann", "d'Este"]
-	prijmeni_f = ['Novotná', 'Svobodová', 'Machová', 'Zelená', 'Yu-Xin', 'Mlsná', 'Dubná',
-			'Mrkvová', 'Suchá', 'Lovelace', 'Holcová', 'Rui', "Nováčková Tydlitátová"]
-	prezdivky = ['Kaki', 'Hurdur', 'Maracuja', 'Bobbo', "", "", "", "", "",
-			"", "", 'Riki', 'Sapa', "", '', '---', 'Koko']
-	domain = ['example.com', 'dolujeme.eu', 'mff.cuni.cz', 'strcprstskrzkrk.cz',
-			'british.co.uk', 'splachni.to', 'haha.org']
-	seznam_ulic = ['Krátká', 'Vlhká', 'Jungmanova', '17. listopadu', '4. října', 'Roztocká',
-			'Forstova', 'Generála Františka Janouška', 'Náměstí Války',
-			'Svratecké náměstí', 'Zelená lhota', 'Z Plynu', 'K Jezeru', 'U Kocourkova',
-			'Uštěpačná', 'Ostrorepská', 'Zubří']
-	seznam_mest = ['Praha', 'Brno', 'Ostrava', 'Horní Jelení', 'Dolní Zábrdovice', 'Prdelkov',
-			'Stará myslivna', 'Kocourkov', 'Šalingrad', 'Medvědí hora', 'Basilej',
-			'Unterschiedlich', 'Old York', 'Lancastershire', 'Vóloďháza']
-
-	osoby = []
-	# 30 je náhodná konstanta, size je použité na víc místech a
-	# říká, jak velká asi chceme testovací data
-	for i in range(30 * size):
-		pohlavi_idx = rnd.randint(0,2) # 2 = nebinární
-		osloveni = [Osoba.OSLOVENI_MUZSKE, Osoba.OSLOVENI_ZENSKE, Osoba.OSLOVENI_ZADNE][pohlavi_idx]
-		jmeno = rnd.choice([jmena_m, jmena_f, jmena_m + jmena_f][pohlavi_idx])
-		prijmeni = rnd.choice([prijmeni_m, prijmeni_f, prijmeni_m + prijmeni_f][pohlavi_idx])
-		if pohlavi_idx == 2: logger.debug(f'Testdata: nebinární osoba: {jmeno} {prijmeni}.')
-		pokusy = 0
-		max_pokusy = 120*size
-		while (not __unikatni_jmeno and pokusy < max_pokusy):
-		# pokud jméno a příjmení není unikátní, zkoušíme generovat nová
-		# do daného limitu (abychom se nezacyklili do nekonečna při málo jménech a příjmeních
-		# ze kterých se generuje)
-			jmeno = rnd.choice([jmena_m, jmena_f, jmena_m + jmena_f][pohlavi_idx])
-			prijmeni = rnd.choice([prijmeni_m, prijmeni_f, prijmeni_m + prijmeni_f][pohlavi_idx])
-			pokusy = pokusy + 1
-		if pokusy >= max_pokusy:
-			print("Chyba, na danou velikost testovacích dat příliš málo možných"
-				" jmen a příjmení")
-			exit()
-		prezdivka = rnd.choice(prezdivky)
-		email = "@".join([unidecode.unidecode(jmeno), rnd.choice(domain)])
-		telefon = "".join([str(rnd.choice([k for k in range(10)])) for i in range(9)])
-		narozeni = datetime.date(rnd.randint(1980, datetime.datetime.now().year), 
-				rnd.randint(1, 12), rnd.randint(1, 28))
-		ulic = rnd.choice(seznam_ulic)
-		cp = rnd.randint(1, 99)
-		ulice = " ".join([ulic, str(cp)])
-		mesto = rnd.choice(seznam_mest)
-		psc = "".join([str(rnd.choice([k for k in range(10)])) for i in range(5)])
-
-		osoby.append(Osoba.objects.create(jmeno = jmeno, prijmeni = prijmeni,
-				prezdivka = prezdivka, osloveni = osloveni, email = email,
-				telefon = telefon, datum_narozeni = narozeni, ulice = ulice,
-				mesto = mesto, psc = psc,
-				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
-
-
-
-def gen_skoly(): #TODO někdy to přepsat, aby jich bylo více
-	logger.info('Generuji školy...')
-
-	skoly = []
-	prvnizs = Skola.objects.create(mesto='Praha', stat='CZ', psc='101 00',
-		ulice='Krátká 5', nazev='První ZŠ', je_zs=True, je_ss=False)
-	skoly.append(prvnizs)
-	skoly.append(Skola.objects.create(mesto='Praha', stat='CZ', psc='101 00',
-		ulice='Krátká 5', nazev='První SŠ', je_zs=False, je_ss=True))
-	skoly.append(Skola.objects.create(mesto='Praha', stat='CZ', psc='102 00',
-		ulice='Dlouhá 5', nazev='Druhá SŠ', je_zs=False, je_ss=True))
-	skoly.append(Skola.objects.create(mesto='Praha', stat='CZ', psc='103 00',
-		ulice='Široká 3', nazev='Třetí SŠ a ZŠ', je_zs=True, je_ss=True))
-	skoly.append(Skola.objects.create(mesto='Ostrava', stat='CZ', psc='700 00',
-		ulice='Hluboká 42', nazev='Hutní gympl', je_zs=False, je_ss=True))
-	skoly.append(Skola.objects.create(mesto='Humenné', stat='SK', psc='012 34',
-		ulice='Pltká 1', nazev='Sredná škuola', je_zs=False, je_ss=True))
-	global zlinska
-	zlinska = Skola.objects.create(mesto = 'Zlín', stat='CZ', psc='76001',
-		ulice='náměstí T.G. Masaryka 2734-9',
-		nazev='Gymnázium a Střední jazyková škola s právem SJZ',
-		kratky_nazev="GaSJŠspSJZ", je_zs=True, je_ss=True)
-	skoly.append(zlinska)
-	return skoly
-
-def gen_resitele(rnd, osoby, skoly):
-	logger.info('Generuji řešitele...')
-
-	resitele = []
-	x = 0
-	resitel_perm = Permission.objects.filter(codename__exact='resitel').first()
-	resitel_group = Group.objects.filter(name__exact='resitel').first()
-	for os in osoby:
-		rand = rnd.randint(0, 8)
-		if not (rand % 8 == 0):
-			if not os.user:
-				if x:
-					user = User.objects.create_user(username='r'+str(x), email=os.email, password='r')
-				else:
-					user = User.objects.create_user(username='r', email=os.email, password='r')
-				x += 1
-				os.user = user
-				os.save()
-				os.user.user_permissions.add(resitel_perm)
-				os.user.groups.add(resitel_group)
-			resitele.append(Resitel.objects.create(osoba=os, skola=rnd.choice(skoly),
-				rok_maturity=os.datum_narozeni.year + rnd.randint(18, 21),
-				zasilat=rnd.choice(Resitel.ZASILAT_CHOICES)[0]))
-	return resitele
-
-def gen_prijemci(rnd, osoby, kolik=10):
-	logger.info('Generuji příjemce (kolik={})...'.format(kolik))
-	prijemci = []
-	for i in rnd.sample(osoby, kolik):
-		prijemci.append(Prijemce.objects.create(osoba=i))
-	return prijemci
-
-def gen_organizatori(rnd, osoby, last_rocnik):
-	logger.info('Generuji organizátory...')
-	organizatori = []
-
-	
-	seznam_konicku = ["vařím", "jezdím na kole", "řeším diferenciální rovnice", "koukám z okna",
-			"tancuji", "programuji", "jezdím vlakem", "nedělám nic"]
-	seznam_oboru = ["matematiku", "matematiku", "matematiku", "fyziku", "literaturu",
-			"informatiku", "informatiku", "běhání dokolečka"]
-
-	x = 0
-	org_perm = Permission.objects.filter(codename__exact='org').first()
-	org_group = Group.objects.filter(name__exact='org').first()
-	for os in osoby:
-		rand = rnd.randint(0, 8)
-		if (rand % 8 == 0):
-			pusobnost = rnd.randint(1, last_rocnik)
-			od = datetime.datetime(
-				year=1993 + pusobnost,
-				month=rnd.randint(1, 12),
-				day=rnd.randint(1, 28),
-				tzinfo=datetime.timezone.utc,
-				)
-			do = datetime.datetime(
-				year=od.year + rnd.randint(1, 6),
-				month=rnd.randint(1, 12),
-				day=rnd.randint(1, 28),
-				tzinfo=datetime.timezone.utc,
-				)
-			#aktualni organizatori jeste nemaji vyplnene organizuje_do
-
-			#popis orga
-			konicek1 = rnd.choice(seznam_konicku)
-			konicek2 = rnd.choice(seznam_konicku)
-			obor = rnd.choice(seznam_oboru)
-			popis_orga = "Ve volném čase " + konicek1 + " a také " + konicek2 + ". Studuji " + obor + " a moc mě to baví."
-
-			if do.year > datetime.datetime.now().year:
-				do = None
-			if not os.user:
-				if x:
-					user = User.objects.create_user(username='o'+str(x), email=os.email, password='o')
-				else:
-					user = User.objects.create_user(username='o', email=os.email, password='o')
-				x += 1
-				os.user = user
-				os.save()
-				os.user.user_permissions.add(org_perm)
-				os.user.groups.add(org_group)
-				os.user.is_staff = True
-				os.user.save()
-			organizatori.append(Organizator.objects.create(osoba=os,
-				organizuje_od=od, organizuje_do=do, strucny_popis_organizatora = popis_orga))
-	return organizatori
 
 def gen_zadani_ulohy(rnd, cisla, organizatori, pocet_oboru, poradi_cisla, poradi_problemu):
 	
@@ -282,42 +88,6 @@ def gen_vzoroveho_reseni_ulohy(rnd, organizatori, uloha, pocet_opravovatelu):
 	uloha.save()
 	return uloha_vzorak
 
-def gen_reseni_ulohy(rnd, cisla, uloha, pocet_resitelu, poradi_cisla, resitele_cisla, resitele):
-	
-	pocet_reseni = rnd.randint(pocet_resitelu//4, pocet_resitelu * 4)
-	# generujeme náhodný počet řešení vzhledem k počtu řešitelů čísla
-	for _ in range(pocet_reseni):
-		#print("Generuji {}-té řešení".format(reseni))
-		if rnd.randint(1, 10) == 1:
-		# cca desetina řešení od více řešitelů
-			res_vyber = rnd.sample(resitele_cisla, 
-				rnd.randint(2, 5))
-		else:
-			res_vyber = rnd.sample(resitele_cisla, 1)
-		if resitele[0] in res_vyber: # speciální řešitel, který nemá žádné body
-			res_vyber.remove(resitele[0])
-
-		# Vytvoření řešení.
-		if uloha.cislo_zadani.zlomovy_deadline_pro_papirove_cislo() is not None:
-			# combine, abychom dostali plný čas a ne jen datum
-			cas_doruceni = uloha.cislo_zadani.deadline_v_cisle.first().deadline - datetime.timedelta(days=random.randint(0, 40)) - datetime.timedelta(minutes=random.randint(0, 60*24))
-			# astimezone, protože jinak vyhazuje warning o nenastavené TZ
-			res = Reseni.objects.create(forma=rnd.choice(Reseni.FORMA_CHOICES)[0], cas_doruceni=cas_doruceni.astimezone(datetime.timezone.utc))
-		else:
-			res = Reseni.objects.create(forma=rnd.choice(Reseni.FORMA_CHOICES)[0])
-		# Problém a řešitele přiřadíme později, ManyToManyField
-		# se nedá vyplnit v create().
-		res.resitele.set(res_vyber)
-		res.save()
-		
-		# Vytvoření hodnocení.
-		hod = Hodnoceni.objects.create(
-			body=rnd.randint(0, uloha.max_body),
-			cislo_body=cisla[poradi_cisla - 1],
-			reseni=res,
-			problem=uloha
-		)
-	return
 
 def gen_ulohy_do_cisla(rnd, organizatori, resitele, rocnik_cisla, rocniky, size):
 	logger.info('Generuji úlohy do čísla (size={})...'.format(size))
@@ -658,22 +428,6 @@ def gen_ulohy_k_tematum(rnd, rocniky, rocnik_cisla, rocnik_temata, organizatori,
 					u.save()
 	return
 
-def gen_novinky(rnd, organizatori):
-	logger.info('Generuji novinky...')
-
-
-	jake = ["zábavné", "veselé", "dobrodružné", "skvělé"]
-	co = ["soustředění", "Fyziklání", "víkendové setkání"]
-	kde = ["na Šumavě", "v Praze", "u Plzně", "na Marsu"]
-	kdy = ["Zítra bude", "10. 10. 2020 bude", "V prosinci bude", "V létě bude"]
-
-	for i in range(5):
-		text_novinky = " ".join([rnd.choice(kdy), rnd.choice(kde), rnd.choice(jake),
-			rnd.choice(co)])
-		novinka = Novinky.objects.create(id=i,autor=rnd.choice(organizatori),
-			text=(text_novinky+", těšíme se na vás!"),zverejneno=rnd.choice([True,False]))
-		novinka.save()
-	return
 
 def otec_syn(otec, syn):
 	bratr = otec.first_child
@@ -750,117 +504,3 @@ def gen_clanek(rnd, organizatori, resitele):
 		text.save()
 		create_child(castnode, m.TextNode, text=text)
 	logger.info(f"Článek vygenerován (reseni={reseni.id}, treenode={reseninode.id})")
-
-
-
-@transaction.atomic
-def create_test_data(size = 6, rnd = None):
-	logger.info('Vyrábím testovací data (size={})...'.format(size))
-
-	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>")
-	print(s)
-	f.sites.add(s[0])
-	f.save()
-
-	# users
-	admin = User.objects.create_superuser(username='admin', email='', password='admin')
-	os_admin = Osoba.objects.create(
-		user=admin, jmeno='admin', prijmeni='admin',
-		prezdivka='admin', osloveni='', email='admin@admin.admin',
-		telefon='123 456 789', datum_narozeni=datetime.date(2000, 1, 1),
-		ulice='admin', mesto='admin', psc='100 00',
-		datum_registrace=datetime.date(2020, 9, 6)
-	)
-	or_admin = Organizator.objects.create(
-		osoba=os_admin, organizuje_od=None, organizuje_do=None,
-		strucny_popis_organizatora="Organizátor k uživateli Admin"
-	)
-
-	usernames = ['anet', 'bara', 'cyril', 'david', 'eva', 'filip']
-	users = []
-	for usr in usernames[:size]:
-		u = User.objects.create_user(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
-	organizatori = gen_organizatori(rnd, osoby, last_rocnik)
-	resitele = gen_resitele(rnd, osoby, skoly)
-
-	#generování novinek
-	novinky = gen_novinky(rnd, organizatori)
-
-	# prijemci
-	prijemci = gen_prijemci(rnd, osoby)
-
-	global zlinska
-	zlinska.kontaktni_osoba=rnd.choice(osoby)
-	zlinska.save()
-
-	# rocniky
-	rocniky = gen_rocniky(last_rocnik, size)
-
-	# cisla
-	# rocnik_cisla je pole polí čísel (typ Cislo), vnitřní pole odpovídají jednotlivým ročníkům.
-	rocnik_cisla = gen_cisla(rnd, rocniky)
-
-	# generování obyčejných úloh do čísel
-	gen_ulohy_do_cisla(rnd, organizatori, resitele, rocnik_cisla, rocniky, size)
-
-	# generování témat, zatím v prvních třech číslech po jednom
-	# FIXME: více témat
-	# rocnik_temata je pole polí trojic (první číslo :int, poslední číslo :int, téma:Tema), přičemž každé vnitřní pole odpovídá ročníku a FIXME: je to takhle fuj a když to někdo vidí poprvé, tak je z toho smutný, protože vůbec neví, co se děje a co má čekat.
-	rocnik_temata = gen_temata(rnd, rocniky, rocnik_cisla, organizatori)
-
-	rocnik = Rocnik.objects.filter(rocnik = 23).first()
-	dlouhe_tema = gen_dlouhe_tema(rnd, organizatori, rocnik, "Strašně dlouhé téma",
-		"MFI", 8)
-
-	# generování úloh k tématům ve všech číslech
-	gen_ulohy_k_tematum(rnd, rocniky, rocnik_cisla, rocnik_temata, organizatori, resitele)
-
-	#generování soustředění
-	soustredeni = gen_soustredeni(size, resitele, organizatori, rnd=rnd)
-
-	#generování konfer
-	konfery = gen_konfery(size, organizatori, soustredeni, rnd=rnd)
-
-	# vytvoreni pdf ke korekturam
-	create_test_pdf(rnd, organizatori)
-
-			# TODO: nastavi správně, kolik se čeho generuje, aby rozsahy přibližně odpovídaly
-			# FIXME: misto typu ruzne typy objektu a vnoreni do sebe (Tom nechápe, co je tímto fixme míněno)
-			# TODO: vytvorit temata s ruznymi vlakny
-			# TODO: nagenerovat starsim rocnikum pohadku
-			# TODO: nagenerovat články
-			# TODO: vecpat obrázky všude, kde to jde
-			# TODO: mezičíslo node
-			# TODO: přidat ke konferám řešení a dát je do čísel
-
-	# Dohackované vytvoření jednoho článku
-	gen_clanek(rnd, organizatori, resitele)
-
-			# TODO: přidat články včetně zařazení do struktury treenodů,
-			#	a následně otestovat konsistency check databáze z utils.py
-			# 	pomocí stránky /stav
-
-	# obecné nastavení semináře, musí být už přidané ročníky a čísla, jinak se nastaví divně
-	nastaveni = Nastaveni.objects.create(
-			aktualni_cislo = Cislo.objects.all()[1])

various/management/commands/testdata.py

@@ -1,13 +1,11 @@
-import datetime
 import os
-import random
 
 from django.core.management.base import BaseCommand
 from django.core.management import call_command
 from django.conf import settings
 
 from seminar.models import Skola, Resitel, Rocnik, Cislo, Problem, Reseni, PrilohaReseni, Nastaveni
-from seminar.testutils import create_test_data
+from various.testutils import create_test_data
 import django.contrib.auth
 User = django.contrib.auth.get_user_model()
 

various/testutils.py

@@ -0,0 +1,135 @@
+import datetime
+import random
+import logging
+
+import django.contrib.auth
+from django.contrib.flatpages.models import FlatPage
+from django.contrib.sites.models import Site
+from django.db import transaction
+
+from seminar.models import Rocnik, Cislo, Nastaveni, Osoba, Organizator
+
+from korektury.testutils import create_test_pdf
+from novinky.testutils import gen_novinky
+from personalni.testutils import gen_organizatori, gen_osoby, gen_prijemci, gen_resitele, gen_skoly
+from soustredeni.testutils import gen_soustredeni, gen_konfery
+from tvorba.testutils import gen_cisla, gen_clanek, gen_dlouhe_tema, gen_rocniky, gen_temata, gen_ulohy_do_cisla, gen_ulohy_k_tematum
+
+logger = logging.getLogger(__name__)
+
+User = django.contrib.auth.get_user_model()
+
+
+@transaction.atomic
+def create_test_data(size=6, rnd=None):
+	logger.info('Vyrábím testovací data (size={})...'.format(size))
+
+	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>",
+	)
+	print(s)
+	f.sites.add(s[0])
+	f.save()
+
+	# users
+	admin = User.objects.create_superuser(
+		username='admin', email='', password='admin',
+	)
+	os_admin = Osoba.objects.create(
+		user=admin, jmeno='admin', prijmeni='admin',
+		prezdivka='admin', osloveni='', email='admin@admin.admin',
+		telefon='123 456 789', datum_narozeni=datetime.date(2000, 1, 1),
+		ulice='admin', mesto='admin', psc='100 00',
+		datum_registrace=datetime.date(2020, 9, 6),
+	)
+	or_admin = Organizator.objects.create(
+		osoba=os_admin, organizuje_od=None, organizuje_do=None,
+		strucny_popis_organizatora="Organizátor k uživateli Admin",
+	)
+
+	usernames = ['anet', 'bara', 'cyril', 'david', 'eva', 'filip']
+	users = []
+	for usr in usernames[:size]:
+		u = User.objects.create_user(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
+	organizatori = gen_organizatori(rnd, osoby, last_rocnik)
+	resitele = gen_resitele(rnd, osoby, skoly)
+
+	# generování novinek
+	novinky = gen_novinky(rnd, organizatori)
+
+	# prijemci
+	prijemci = gen_prijemci(rnd, osoby)
+
+	# rocniky
+	rocniky = gen_rocniky(last_rocnik, size)
+
+	# cisla
+	# rocnik_cisla je pole polí čísel (typ Cislo), vnitřní pole odpovídají jednotlivým ročníkům.
+	rocnik_cisla = gen_cisla(rnd, rocniky)
+
+	# generování obyčejných úloh do čísel
+	gen_ulohy_do_cisla(rnd, organizatori, resitele, rocnik_cisla, rocniky, size)
+
+	# generování témat, zatím v prvních třech číslech po jednom
+	# FIXME: více témat
+	# rocnik_temata je pole polí trojic (první číslo :int, poslední číslo :int, téma:Tema), přičemž každé vnitřní pole odpovídá ročníku a FIXME: je to takhle fuj a když to někdo vidí poprvé, tak je z toho smutný, protože vůbec neví, co se děje a co má čekat.
+	rocnik_temata = gen_temata(rnd, rocniky, rocnik_cisla, organizatori)
+
+	rocnik = Rocnik.objects.filter(rocnik=23).first()
+	dlouhe_tema = gen_dlouhe_tema(
+		rnd, organizatori, rocnik, "Strašně dlouhé téma",
+		"MFI", 8,
+	)
+
+	# generování úloh k tématům ve všech číslech
+	gen_ulohy_k_tematum(rnd, rocniky, rocnik_cisla, rocnik_temata, organizatori, resitele)
+
+	# generování soustředění
+	soustredeni = gen_soustredeni(size, resitele, organizatori, rnd=rnd)
+
+	# generování konfer
+	konfery = gen_konfery(size, organizatori, soustredeni, rnd=rnd)
+
+	# vytvoreni pdf ke korekturam
+	create_test_pdf(rnd, organizatori)
+
+	# TODO: nastavi správně, kolik se čeho generuje, aby rozsahy přibližně odpovídaly
+	# FIXME: misto typu ruzne typy objektu a vnoreni do sebe (Tom nechápe, co je tímto fixme míněno)
+	# TODO: vytvorit temata s ruznymi vlakny
+	# TODO: nagenerovat starsim rocnikum pohadku
+	# TODO: nagenerovat články
+	# TODO: vecpat obrázky všude, kde to jde
+	# TODO: mezičíslo node
+	# TODO: přidat ke konferám řešení a dát je do čísel
+
+	# Dohackované vytvoření jednoho článku
+	gen_clanek(rnd, organizatori, resitele)
+
+	# TODO: přidat články včetně zařazení do struktury treenodů,
+	# a následně otestovat konsistency check databáze z utils.py
+	# pomocí stránky /stav
+
+	# obecné nastavení semináře, musí být už přidané ročníky a čísla, jinak se nastaví divně
+	nastaveni = Nastaveni.objects.create(
+		aktualni_cislo=Cislo.objects.all()[1])