diff --git a/seminar/testutils.py b/seminar/testutils.py index a991660e..36902417 100644 --- a/seminar/testutils.py +++ b/seminar/testutils.py @@ -13,52 +13,7 @@ from django.contrib.sites.models import Site User = django.contrib.auth.get_user_model() -@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 = "
Vítejte na stránce semináře MaM!
") - 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'] - orgs = [] - for org in usernames[:size]: - o = User.objects.create_user(username=org, password=org) - o.first_name = org.capitalize() - o.save() - orgs.append(o) - - # skoly - 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)) -# #FIXME pridat kontaktni osobu alespon nekde -# skoly.append(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)) - - # osoby +def gen_osoby(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'] @@ -89,45 +44,175 @@ def create_test_data(size = 6, rnd = None): narozeni = datetime.date(rnd.randint(1980, 2020), rnd.randint(1, 12), rnd.randint(1, 28)) ulic = rnd.choice(seznam_ulic) - print(ulic) + print(ulic) #FIXME: pozdeji smazat, jen debug cp = rnd.randint(1, 99) ulice = " ".join([ulic, str(cp)]) mesto = rnd.choice([seznam_mest]) psc = [rnd.choice([k for k in range(10)]) for i in range(6)] osoby.append(Osoba.objects.create(jmeno = jmeno, prijmeni = prijmeni, - prezdivka = prezdivka, pohlavi_muz = pohlavi, 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)))) + prezdivka = prezdivka, pohlavi_muz = pohlavi, 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? - # resitele a organizatori - last_rocnik = 25 + return osoby + + + +def gen_skoly(): + 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)) + #FIXME pridat kontaktni osobu alespon nekde + skoly.append(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)) + return skoly + +def gen_resitele(osoby, skoly): resitele = [] - organizatori = [] for os in osoby: rand = rnd.randint(0, 8) if not (rand % 8 == 0): resitele.append(Resitel.objects.create(osoba=os, skola=rnd.choice([skoly]), rok_maturity=rnd.randint(2019, 2029), zasilat=rnd.choice(Resitel.ZASILAT_CHOICES))) - else: + return resitele + +def gen_prijemci(osoby, kolik=10): + prijemci = [] + for i in range(kolik): + rand_os = rnd.choice(osoby) + prijemci.add(Prijemce.objects.create(osoba=rand_os)) + return prijemci + +def gen_organizatori(osoby, last_rocnik): + organizatori = [] + for os in osoby: + rand = rnd.randint(0, 8) + if (rand % 8 == 0): pusobnost = rnd.randint(1, last_rocnik) od = 1993 + last_rocnik - pusobnost do = od + rnd.randint(1, 6) # aktualni organizatori jeste nemaji vyplnene organizuje_do #if do > datetime.datetime.now().year: # do = None - organizatori.append(Organizator.objects.create(osoba=os)) - #organizuje_od=od)) - #, organizuje_do=do)) + #organizatori.append(Organizator.objects.create(osoba=os, + # user=rnd.choice(users), + # organizuje_od=od, organizuje_do=do)) + return organizatori + +def gen_ulohy_do_cisla(cislo, organizatori, resitele, slovnik_cisel, size): +''' +cislo = cislo cisla v rocniku, nikoli objekt Cislo +''' + # ulohy resene v ci + jaka = ["Šachová", "Černá", "Větrná", "Dlouhá", "Křehká", "Rychlá", + "Zákeřná", "Fyzikální"] + co = ["kostka", "smršť", "díra", "zrada", "toulka", "tyč", + "úloha", "blecha"] + sloveso = ["Najděte", "Spočítejte", "Zapište", "Změřte", "Odhadněte"] + koho = ["délku", "počet", "množství", "dílky"] + ceho = ["všech", "správných", "konstatních", "zelených"] + jmeno = ["řešení", "tahů", "čísel", "kalhot", "koulí", "hadů"] + kde = ["na zemi", "ve vesmíru", "ve vzduchu", "na šňůře", "v letadle"] + obor = ["M", "F", "I", "O", "B"] + reseni = ["to je přece jasné", "triviální", "omlouváme se," + "otevřený problém", "neřešitelné", "triviálně triviální", + "použitím věty z prvního semestru na matfyzu", + "jednoduše pomocí látky z druhého semestru na matfyzu", + "netriviální aplikace diferenciálních rovnic", "zadání je vnitřně" + "sporné", "nepopsatelně jednoduché", "pokud jste na to nepřišli," + "tak jste fakt hloupí"] + + if cislo >= 3: + for pi in range(1, ((size + 1) // 2) + 1): + poc_op = rnd.randint(1, 4) + poc_oboru = rnd.randint(1, 2) + p = Uloha.objects.create( + nazev=" ".join([rnd.choice(jaka), rnd.choice(co)]), + stav=Problem.STAV_ZADANY, + zamereni=rnd.sample(zamereni, poc_oboru), + autor=rnd.choice(organizatori), + garant=rnd.choice(organizatori), + opravovatele=rnd.sample(organizatori, poc_op), + kod=str(pi), + cislo_zadani=slovnik_cisel[cislo-2], + cislo_reseni=slovnik_cisel[cislo], + cislo_deadline=slovnik_cisel[cislo], + max_body = rnd.randint(1, 8)) + p.zadani = " ".join([rnd.choice(sloveso), rnd.choice(koho), + rnd.choice(ceho), rnd.choice(jmeno), rnd.choice(kde)]) + p.vzorak = " - ".join([p.zadani, rnd.choice(reseni)]) + p.save() + + # reseni ulohy + poc_reseni = rnd.randint(size // 2, size * 2) + poc_resitel = rnd.randint(1, 3) + res_sel = rnd.sample(resitele, min(poc_reseni, len(resitele) - 2)) + for resitel in res_sel: + res = Reseni.objects.create(problem = p, + resitele=[resitel], + forma=rnd.choice(Reseni.FORMA_CHOICES)) + hod = Hodnoceni.objects.create(body=rnd.randint(0, p.max_body), + cislo_body=slovnik_cisel[cislo], reseni=res, problem=p) + return + + + +@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 = "Vítejte na stránce semináře MaM!
") + 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_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() + + # resitele a organizatori + last_rocnik = 25 + resitele = gen_resitele(osoby) + organizatori = gen_organizatori(osoby) # prijemci - prijemci = [] - for i in range(10): - randos = rnd.choice(osoby) - prijemci.add(Prijemce.objects.create(osoba=randos)) + prijemci = gen_prijemci() zlinska.kontaktni_osoba=rnd.choice(osoby) zlinska.save() @@ -138,7 +223,6 @@ def create_test_data(size = 6, rnd = None): # cisla cisel = rnd.randint(4, 8) - slovnik_cisel = {} for ci in range(1, cisel + 1): if ci >= 3: @@ -153,70 +237,75 @@ def create_test_data(size = 6, rnd = None): else: deadline = None cislo = Cislo.objects.create(rocnik = r, cislo = str(ci), - datum_vydani=vydano, datum_deadline=deadline, verejne_db=True) + datum_vydani=vydano, datum_deadline=deadline, + verejne_db=True) slovnik_cisel[ci] = cislo - # FIXME: misto typu ruzne typy objektu a vnoreni do sebe - # TODO: vytvorit temata s ruznymi vlakny - # TODO: nagenerovat starsim rocnikum pohadku - - # ulohy resene v ci - jaka = ["Šachová", "Černá", "Větrná", "Dlouhá", "Křehká", "Rychlá", - "Zákeřná", "Fyzikální"] - co = ["kostka", "smršť", "díra", "zrada", "toulka", "tyč", - "úloha", "blecha"] - sloveso = ["Najděte", "Spočítejte", "Zapište", "Změřte", "Odhadněte"] - koho = ["délku", "počet", "množství", "dílky"] - ceho = ["všech", "správných", "konstatních", "zelených"] - jmeno = ["řešení", "tahů", "čísel", "kalhot", "koulí", "hadů"] - kde = ["na zemi", "ve vesmíru", "ve vzduchu", "na šňůře", "v letadle"] - obor = ["M", "F", "I", "O", "B"] - reseni = ["to je přece jasné", "triviální", "omlouváme se," - "otevřený problém", "neřešitelné", "triviálně triviální", - "použitím věty z prvního semestru na matfyzu", - "jednoduše pomocí látky z druhého semestru na matfyzu", - "netriviální aplikace diferenciálních rovnic", "zadání je vnitřně" - "sporné", "nepopsatelně jednoduché", "pokud jste na to nepřišli," - "tak jste fakt hloupí"] - - if ci >= 3: - for pi in range(1, ((size + 1) // 2) + 1): + # generovani uloh + gen_ulohy_do_cisla(ci, organizatori, resitele, size) + + # generovani temat + if ci <= 3: + jake = ["Hravé", "Fyzikální", "Nejlepší", "Totálně masakrální", + "Šokující", "Magnetické", "Modré", "Překvapivé", + "Plasmatické", "Novoroční"] + co = ["téma", "záření", "stavení", "jiskření", "jelito", + "drama", "kuře", "moře", "klání", "proudění", "čekání"] + poc_oboru = rnd.randint(1, 2) + poc_op = rnd.randint(1, 3) + t = Tema.objects.create( + nazev=" ".join([rnd.choice(jake), rnd.choice(co)]), + stav=Problem.STAV_ZADANY, + zamereni=rnd.sample(zamereni, poc_oboru), + autor=rnd.choice(organizatori), + garant=rnd.choice(organizatori), + opravovatele=rnd.sample(organizatori, poc_op), + kod=str(pi), + tema_typ=rnd.choice(Tema.TEMA_CHOICES), + rocnik=r) + # nagenerovat k tématu úlohy a Nody pro vsechna cisla + for i in range(ci, cisel): + # vytvor ulohu poc_op = rnd.randint(1, 4) poc_oboru = rnd.randint(1, 2) p = Uloha.objects.create( - nazev=" ".join([rnd.choice(jaka), rnd.choice(co)]), + nazev=": ".join([t.nazev, "úloha {}.".format(i)]), stav=Problem.STAV_ZADANY, - zamereni=rnd.sample(zamereni, poc_oboru), - autor=rnd.choice(organizatori), - garant=rnd.choice(organizatori), + zamereni=t.zamereni, + autor=t.autor, + garant=t.garant, opravovatele=rnd.sample(organizatori, poc_op), - kod=str(pi), - cislo_zadani=cs[ci-2], - cislo_reseni=cs[ci], - cislo_deadline=cs[ci], + kod=str(i), + cislo_zadani=slovnik_cisel[cislo-2], + cislo_reseni=slovnik_cisel[cislo], + cislo_deadline=slovnik_cisel[cislo], max_body = rnd.randint(1, 8)) - p.zadani = " ".join([rnd.choice(sloveso), rnd.choice(koho), - rnd.choice(ceho), rnd.choice(jmeno), rnd.choice(kde)]) - p.vzorak = " - ".join([p.zadani, rnd.choice(reseni)]) - p.save() - - # reseni ulohy - poc_reseni = rnd.randint(size // 2, size * 2) - poc_resitel = rnd.randint(1, 3) - res_sel = rnd.sample(resitele, min(poc_reseni, len(resitele) - 2)) - for resitel in res_sel: - res = Reseni.objects.create(problem = p, - resitele=[resitel], - forma=rnd.choice(Reseni.FORMA_CHOICES)) - hod = Hodnoceni.objects.create(body=rnd.randint(0, p.max_body), - cislo_body=cs[ci], reseni=res, problem=p) + p.zadani = " ".join([rnd.choice(sloveso), rnd.choice(koho), + rnd.choice(ceho), rnd.choice(jmeno), rnd.choice(kde)]) + p.vzorak = " - ".join([p.zadani, rnd.choice(reseni)]) + p.save() + + + # vytvor text + # vytvor TemaVCisleNode + if (rnd.randint(1, 3) % 3 == 0): + # vytvor mezicislo Node + + + + + # FIXME: misto typu ruzne typy objektu a vnoreni do sebe + # TODO: vytvorit temata s ruznymi vlakny + # TODO: nagenerovat starsim rocnikum pohadku # TODO: nahodne nagenerovat problemum reseni a prilohy reseni, hodnoceni # TODO: vice soustredeni a k nim nahodne podmnoziny organizatoru a ucastniku - 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)) + 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, 6): Soustredeni_Ucastnici.objects.create(resitel=res, soustredeni=sous) sous.save()