seminar/testutils.py
This commit is contained in:
parent
d952ab13a5
commit
348096024e
6 changed files with 443 additions and 368 deletions
27
novinky/testutils.py
Normal file
27
novinky/testutils.py
Normal file
|
|
@ -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
|
||||||
40
odevzdavatko/testutils.py
Normal file
40
odevzdavatko/testutils.py
Normal file
|
|
@ -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
|
||||||
235
personalni/testutils.py
Normal file
235
personalni/testutils.py
Normal file
|
|
@ -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
|
||||||
|
|
@ -1,216 +1,22 @@
|
||||||
|
# FIXME vypreparovat treenode
|
||||||
|
|
||||||
import datetime
|
import datetime
|
||||||
|
|
||||||
from django.contrib.auth.models import Permission
|
|
||||||
from django.contrib.auth.models import Group
|
|
||||||
import random
|
|
||||||
import lorem
|
import lorem
|
||||||
import django.contrib.auth
|
import django.contrib.auth
|
||||||
from django.db import transaction
|
|
||||||
import unidecode
|
|
||||||
import logging
|
import logging
|
||||||
|
|
||||||
from korektury.testutils import create_test_pdf
|
from seminar.models import Rocnik, Cislo, Deadline, Problem, Tema, Uloha, TextNode, UlohaVzorakNode, RocnikNode, CisloNode, TemaVCisleNode, Text, UlohaZadaniNode
|
||||||
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
|
|
||||||
import seminar.models as m
|
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 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
|
||||||
|
|
||||||
|
|
||||||
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__)
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
# testuje unikátnost vygenerovaného jména
|
User = django.contrib.auth.get_user_model()
|
||||||
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):
|
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()
|
uloha.save()
|
||||||
return uloha_vzorak
|
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):
|
def gen_ulohy_do_cisla(rnd, organizatori, resitele, rocnik_cisla, rocniky, size):
|
||||||
logger.info('Generuji úlohy do čísla (size={})...'.format(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()
|
u.save()
|
||||||
return
|
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):
|
def otec_syn(otec, syn):
|
||||||
bratr = otec.first_child
|
bratr = otec.first_child
|
||||||
|
|
@ -750,117 +504,3 @@ def gen_clanek(rnd, organizatori, resitele):
|
||||||
text.save()
|
text.save()
|
||||||
create_child(castnode, m.TextNode, text=text)
|
create_child(castnode, m.TextNode, text=text)
|
||||||
logger.info(f"Článek vygenerován (reseni={reseni.id}, treenode={reseninode.id})")
|
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ítejte na stránce seminář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])
|
|
||||||
|
|
@ -1,13 +1,11 @@
|
||||||
import datetime
|
|
||||||
import os
|
import os
|
||||||
import random
|
|
||||||
|
|
||||||
from django.core.management.base import BaseCommand
|
from django.core.management.base import BaseCommand
|
||||||
from django.core.management import call_command
|
from django.core.management import call_command
|
||||||
from django.conf import settings
|
from django.conf import settings
|
||||||
|
|
||||||
from seminar.models import Skola, Resitel, Rocnik, Cislo, Problem, Reseni, PrilohaReseni, Nastaveni
|
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
|
import django.contrib.auth
|
||||||
User = django.contrib.auth.get_user_model()
|
User = django.contrib.auth.get_user_model()
|
||||||
|
|
||||||
|
|
|
||||||
135
various/testutils.py
Normal file
135
various/testutils.py
Normal file
|
|
@ -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ítejte na stránce seminář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])
|
||||||
Loading…
Reference in a new issue