@ -1,712 +1,24 @@
import datetime import datetime
from django . contrib . auth . models import Permission
from django . contrib . auth . models import Group
from pytz import timezone
import random import random
import lorem import lorem
import django . contrib . auth
from django . db import transaction
import unidecode
import logging import logging
from korektury . testutils import create_test_pdf import django . contrib . auth
from seminar . models . nastaveni import * from django . db import transaction
from personalni . models import *
from tvorba . models import *
from odevzdavatko . models import *
from soustredeni . models import *
from seminar . models . novinky import *
from seminar . models . pomocne import *
from treenode . models import *
from django . contrib . flatpages . models import FlatPage from django . contrib . flatpages . models import FlatPage
from django . contrib . sites . models import Site from django . contrib . sites . models import Site
from treenode . treelib import all_children , insert_last_child , all_children_of_type , create_node_after
from seminar . models import *
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 = rnd . randint ( 0 , 1 )
jmeno = rnd . choice ( [ jmena_m , jmena_f ] [ pohlavi ] )
prijmeni = rnd . choice ( [ prijmeni_m , prijmeni_f ] [ pohlavi ] )
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 ] [ pohlavi ] )
prijmeni = rnd . choice ( [ prijmeni_m , prijmeni_f ] [ pohlavi ] )
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 , 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?
# 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 = timezone ( ' CET ' ) ,
)
do = datetime . datetime (
year = od . year + rnd . randint ( 1 , 6 ) ,
month = rnd . randint ( 1 , 12 ) ,
day = rnd . randint ( 1 , 28 ) ,
tzinfo = timezone ( ' CET ' ) ,
)
#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 ) :
# Proměnné pro náhodné generování názvů a zadání.
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 " ]
obory = [ " M " , " F " , " I " , " O " , " B " ]
p = Uloha . objects . create (
# atributy třídy Problem
nazev = " " . join ( [ rnd . choice ( jaka ) , rnd . choice ( co ) ] ) ,
stav = Problem . STAV_ZADANY ,
zamereni = rnd . sample ( obory , pocet_oboru ) ,
autor = rnd . choice ( organizatori ) ,
garant = rnd . choice ( organizatori ) ,
kod = str ( poradi_problemu ) ,
# atributy třídy Uloha
cislo_zadani = cisla [ poradi_cisla - 2 - 1 ] ,
cislo_reseni = cisla [ poradi_cisla - 1 ] ,
cislo_deadline = cisla [ poradi_cisla - 1 ] ,
max_body = rnd . randint ( 1 , 8 )
)
text = " " . join (
[ rnd . choice ( sloveso ) ,
rnd . choice ( koho ) ,
rnd . choice ( ceho ) ,
rnd . choice ( jmeno ) ,
rnd . choice ( kde ) ]
)
text_zadani = Text . objects . create (
na_web = text ,
do_cisla = text ,
)
zad = TextNode . objects . create ( text = text_zadani , root = p . cislo_zadani . rocnik . rocniknode )
uloha_zadani = UlohaZadaniNode . objects . create ( uloha = p , first_child = zad , root = p . cislo_zadani . rocnik . rocniknode )
p . ulohazadaninode = uloha_zadani
otec_syn ( cisla [ poradi_cisla - 2 - 1 ] . cislonode , uloha_zadani )
return p
def gen_vzoroveho_reseni_ulohy ( rnd , organizatori , uloha , pocet_opravovatelu ) :
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í " ]
# Generování vzorového řešení.
obsah = rnd . choice ( reseni )
text_vzoraku = Text . objects . create (
na_web = obsah ,
do_cisla = obsah
)
vzorak = TextNode . objects . create ( text = text_vzoraku , root = uloha . cislo_zadani . rocnik . rocniknode )
uloha_vzorak = UlohaVzorakNode . objects . create ( uloha = uloha , first_child = vzorak , root = uloha . cislo_zadani . rocnik . rocniknode )
uloha . ulohavzoraknode = uloha_vzorak
uloha . opravovatele . set ( rnd . sample ( organizatori , 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 ) )
k = 0
for rocnik in rocniky :
k + = 1
print ( " Generuji {} . číslo. " . format ( k ) )
cisla = rocnik_cisla [ k - 1 ]
for ci in range ( 3 , len ( cisla ) + 1 ) : # pro všechna čísla
resitele_size = round ( 7 / 8 * 30 * size ) # očekáváný celkový počet řešitelů
poc_res = rnd . randint ( resitele_size / / 8 , resitele_size / / 4 )
# dané číslo řeší něco mezi osminou a čtvrtinou všech řešitelů
# (náhodná hausnumera, možno změnit)
# účelem je, aby se řešení generovala z menší množiny řešitelů a tedy
# bylo více řešení od jednoho řešitele daného čísla
resitele_cisla = rnd . sample ( resitele , poc_res )
for pi in range ( 1 , ( ( size + 1 ) / / 2 ) + 1 ) : # počet problémů
poc_op = rnd . randint ( 1 , 4 ) # počet opravovatelů
poc_oboru = rnd . randint ( 1 , 2 )
# Generování zadání úlohy a UlohaZadaniNode,
# přivěšení pod dané číslo
p = gen_zadani_ulohy ( rnd , cisla , organizatori , poc_oboru , ci , pi )
# Generování vzorového řešení
uloha_vzorak = gen_vzoroveho_reseni_ulohy ( rnd , organizatori ,
p , poc_op )
insert_last_child ( cisla [ ci - 1 ] . cislonode , uloha_vzorak )
# Generování řešení a hodnocení k úloze
gen_reseni_ulohy ( rnd , cisla , p , poc_res , ci ,
resitele_cisla , resitele )
return
def gen_soustredeni ( rnd , resitele , organizatori ) :
logger . info ( ' Generuji soustředění... ' )
soustredeni = [ ]
for _ in range ( 1 , 10 ) : #FIXME Tu range si změňte jak chcete, nevím, co přesně znamená size (asi Anet?)
datum_zacatku = datetime . date ( rnd . randint ( 2000 , 2020 ) , rnd . randint ( 1 , 12 ) , rnd . randint ( 1 , 28 ) )
working_sous = Soustredeni . objects . create (
rocnik = Rocnik . objects . order_by ( ' ? ' ) . first ( ) ,
verejne_db = rnd . choice ( [ True , False ] ) ,
misto = rnd . choice ( [ ' Kremrolovice ' , ' Indiánov ' , ' U zmzliny ' , ' Vafláreň ' , ' Větrník ' , ' Horní Rakvička ' , ' Dolní cheesecake ' ] ) ,
typ = rnd . choice ( [ ' jarni ' , ' podzimni ' , ' vikend ' ] ) ,
datum_zacatku = datum_zacatku ,
datum_konce = datum_zacatku + datetime . timedelta ( days = 7 ) )
ucastnici = rnd . sample ( resitele , min ( len ( resitele ) , 20 ) )
working_sous . ucastnici . set ( ucastnici )
#for res in rnd.sample(resitele, min(len(resitele), 20)):
# Soustredeni_Ucastnici.objects.create(resitel=res, soutredeni=working_sous)
orgove_vyber = rnd . sample ( organizatori , min ( len ( organizatori ) , 20 ) )
working_sous . organizatori . set ( orgove_vyber )
#for org in rnd.sample(organizatori, min(len(organizatori), 20)):
# Soustredeni_Organizatori.objects.create(organizator=org, soutredeni=working_sous)
working_sous . save ( )
soustredeni . append ( working_sous )
return soustredeni
def gen_rocniky ( last_rocnik , size ) :
logger . info ( ' Generuji ročníky (size= {} )... ' . format ( size ) )
rocniky = [ ]
node = None
for ri in range ( min ( last_rocnik - size , 1 ) , last_rocnik + 1 ) :
rocnik = Rocnik . objects . create ( prvni_rok = 1993 + ri , rocnik = ri )
node2 = RocnikNode . objects . create ( rocnik = rocnik , succ = node )
rocnik . save ( )
node = node2
rocniky . append ( rocnik )
return rocniky
def gen_konfery ( size , rnd , organizatori , resitele , soustredeni ) :
logger . info ( ' Generuji konfery (size= {} )... ' . format ( size ) )
konfery = [ ]
for _ in range ( 1 , size ) : #FIXME Tu range si změňte jak chcete, nevím, co přesně znamená size (asi Anet?)
# Anet: size je parametr udávající velikost testovacích dat a dá se pomocí ní škálovat,
# kolik dat se nageneruje
konfera = Konfera . objects . create (
nazev = rnd . choice ( [ ' Pozorování ' , ' Zkoumání ' , ' Modelování ' , ' Počítání ' , ' Zkoušení ' ] ) + rnd . choice ( [ ' vlastností ' , ' jevů ' , ' charakteristik ' ] ) + rnd . choice ( [ ' vektorových prostorů ' , ' kinetické terorie látek ' , ' molekulární biologie ' , ' syntentických stromů ' ] ) ,
anotace = lorem . paragraph ( ) ,
abstrakt = lorem . paragraph ( ) ,
garant = rnd . choice ( organizatori ) ,
soustredeni = rnd . choice ( soustredeni ) ,
typ_prezentace = rnd . choice ( [ ' veletrh ' , ' prezentace ' ] ) )
ucastnici_sous = list ( konfera . soustredeni . ucastnici . all ( ) )
ucastnici = rnd . sample ( ucastnici_sous , min ( len ( ucastnici_sous ) , rnd . randint ( 3 , 6 ) ) )
konfera . ucastnici . set ( ucastnici )
#for res in rnd.sample(ucastnici, min(len(ucastnici), rnd.randint(3, 6))):
# Konfery_Ucastnici.objects.create(resitel=res, konfera=konfera)
konfera . save ( )
konfery . append ( konfera )
return konfery
def gen_cisla ( rnd , rocniky ) :
logger . info ( ' Generuji čísla... ' )
rocnik_cisla = [ ]
for rocnik in rocniky :
otec = True
cisla = [ ]
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
# další po dvou měsících (což je rozumná aproximace)
mesic_vydani = ( ci - 1 ) * 2 + 6
# celociselné dělení mi řekne, jestli to je první nebo druhý rok ročníku
vydano = datetime . date ( rocnik . prvni_rok + mesic_vydani / / 12 ,
( mesic_vydani - 1 ) % 12 + 1 ,
rnd . randint ( 1 , 28 ) )
deadline = datetime . date ( rocnik . prvni_rok + ( mesic_vydani + 2 ) / / 12 ,
( mesic_vydani + 1 ) % 12 + 1 ,
rnd . randint ( 1 , 28 ) )
cislo = Cislo . objects . create (
rocnik = rocnik ,
poradi = str ( ci ) ,
datum_vydani = vydano ,
verejne_db = True ,
)
node2 = CisloNode . objects . get ( cislo = cislo )
node2 . succ = node
node2 . root = rocnik . rocniknode
cislo . save ( )
deadline = Deadline . objects . create (
cislo = cislo ,
deadline = deadline ,
typ = Deadline . TYP_CISLA ,
verejna_vysledkovka = True ,
)
deadline . save ( )
node = node2
if otec :
otec = False
rocnik . rocniknode . first_child = node
rocnik . save ( )
cisla . append ( cislo )
rocnik_cisla . append ( cisla )
return rocnik_cisla
def add_first_child ( node , child ) :
node . first_child = child
node . save ( )
return
def get_text ( ) : def get_text ( ) :
odstavec = lorem . paragraph ( ) odstavec = lorem . paragraph ( )
return Text . objects . create ( na_web = odstavec , do_cisla = odstavec ) return Text . objects . create ( na_web = odstavec , do_cisla = odstavec )
def gen_dlouhe_tema ( rnd , organizatori , rocnik , nazev , obor , kod ) :
tema = Tema . objects . create (
nazev = nazev ,
stav = Problem . STAV_ZADANY ,
zamereni = " M " ,
autor = rnd . choice ( organizatori ) ,
garant = rnd . choice ( organizatori ) ,
kod = str ( kod ) ,
tema_typ = rnd . choice ( Tema . TEMA_CHOICES ) [ 0 ] ,
rocnik = rocnik ,
abstrakt = lorem . paragraph ( )
)
# Generování struktury k tématu
cisla = sorted ( rocnik . cisla . all ( ) , key = lambda cislo : cislo . poradi )
for cislo in cisla :
# Přidáme TemaVCisleNode do daného čísla
cislo_node = cislo . cislonode
tema_cislo_node = TemaVCisleNode . objects . create ( tema = tema , root = cislo_node . root )
insert_last_child ( cislo_node , tema_cislo_node )
# Přidávání obsahu do čísla
cast_node = CastNode . objects . create ( nadpis = " Příspěvek k číslu {} " . format ( cislo . kod ) , root = cislo_node . root )
add_first_child ( tema_cislo_node , cast_node )
text_node = TextNode . objects . create ( text = get_text ( ) , root = cislo_node . root )
add_first_child ( cast_node , text_node )
cast_node2 = CastNode . objects . create ( nadpis = " První podproblém " , root = cislo_node . root )
add_first_child ( text_node , cast_node2 )
text_node2 = TextNode . objects . create ( text = get_text ( ) , root = cislo_node . root )
add_first_child ( cast_node2 , text_node2 )
cast_node3 = CastNode . objects . create ( nadpis = " Druhý podproblém " , root = cislo_node . root )
add_first_child ( text_node2 , cast_node3 )
text_node3 = TextNode . objects . create ( text = get_text ( ) , root = cislo_node . root )
add_first_child ( cast_node3 , text_node3 )
cast_node4 = CastNode . objects . create ( nadpis = " Třetí podproblém " , root = cislo_node . root )
add_first_child ( text_node3 , cast_node4 )
text_node4 = TextNode . objects . create ( text = get_text ( ) , root = cislo_node . root )
add_first_child ( cast_node3 , text_node4 )
cast_node3a = CastNode . objects . create ( nadpis = " Podproblém paralelní s "
" druhým podproblémem " , root = cislo_node . root )
cast_node3 . succ = cast_node3a
cast_node3 . save ( )
text_node3a = TextNode . objects . create ( text = get_text ( ) , root = cislo_node . root )
add_first_child ( cast_node3a , text_node3a )
# Občas přidáme mezičíslo
if rnd . randint ( 1 , 3 ) == 1 :
create_node_after ( cislo_node , MezicisloNode , root = cislo_node . root )
mezicislo_node = cislo_node . succ
cast_node_mezicislo = CastNode . objects . create (
nadpis = " Příspěvek k mezičíslu " . format ( cislo . kod ) , root = cislo_node . root )
add_first_child ( mezicislo_node , cast_node_mezicislo )
odstavec = lorem . paragraph ( )
text_mezicislo = Text . objects . create ( na_web = odstavec , do_cisla = odstavec )
text_node_mezicislo = TextNode . objects . create ( text = text_mezicislo , root = cislo_node . root )
add_first_child ( cast_node_mezicislo , text_node_mezicislo )
return tema
def gen_temata ( rnd , rocniky , rocnik_cisla , organizatori ) :
logger . info ( ' Generuji témata... ' )
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 )
rocnik_temata = [ ]
# Věříme, že rocnik_cisla je pole polí čísel podle ročníků, tak si necháme dát
# vždycky jeden ročník a k němu příslušná čísla.
for rocnik , cisla in zip ( rocniky , rocnik_cisla ) :
kod = 1
letosni_temata = [ ]
# Do každého ročníku vymyslíme tři (zatím) témata, v každém z prvních čísel jedno
for zacatek_tematu in range ( 1 , 3 ) :
# Vygenerujeme téma
t = Tema . objects . create (
# atributy třídy Problem
nazev = " " . join ( [ rnd . choice ( jake ) , rnd . choice ( co ) ] ) ,
stav = Problem . STAV_ZADANY ,
zamereni = rnd . sample ( [ " M " , " F " , " I " , " O " , " B " ] , poc_oboru ) ,
autor = rnd . choice ( organizatori ) ,
garant = rnd . choice ( organizatori ) ,
kod = str ( kod ) ,
# atributy třídy Téma
tema_typ = rnd . choice ( Tema . TEMA_CHOICES ) [ 0 ] ,
rocnik = rocnik ,
abstrakt = " Abstrakt tematka {} " . format ( kod )
)
kod + = 1
# Vymyslíme, kdy skončí
konec_tematu = min ( rnd . randint ( zacatek_tematu , 7 ) , len ( cisla ) )
# Vyrobíme TemaVCisleNody pro obsah
for i in range ( zacatek_tematu , konec_tematu + 1 ) :
node = TemaVCisleNode . objects . create ( tema = t , root = rocnik . rocniknode )
# FIXME: Není to off-by-one?
otec = cisla [ i - 1 ] . cislonode
otec_syn ( otec , node )
# Vymyslíme, kdo to bude opravovat
poc_opravovatelu = rnd . randint ( 1 , 3 )
t . opravovatele . set ( rnd . sample ( organizatori , poc_opravovatelu ) )
# Uložíme všechno
t . save ( )
letosni_temata . append ( ( zacatek_tematu , konec_tematu , t ) )
rocnik_temata . append ( letosni_temata )
return rocnik_temata
def gen_ulohy_tematu ( rnd , organizatori , resitele , tema , kod , cislo , cislo_se_vzorakem ) :
""" Generování úlohy k danému tématu. """
# Proměnné pro náhodné generování názvů a zadání.
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 " ]
obory = [ " M " , " F " , " I " , " O " , " B " ]
uloha = Uloha . objects . create (
nazev = " : " . join ( [ tema . nazev ,
" úloha {} . " . format ( kod ) ] ) ,
nadproblem = tema ,
stav = Problem . STAV_ZADANY ,
zamereni = tema . zamereni ,
autor = tema . autor ,
garant = tema . garant ,
kod = str ( kod ) ,
cislo_zadani = cislo ,
cislo_reseni = cislo_se_vzorakem ,
cislo_deadline = cislo_se_vzorakem ,
max_body = rnd . randint ( 1 , 8 )
)
# Samotný obsah následně vzniklého Textu zadání
obsah = " " . join (
[ rnd . choice ( sloveso ) ,
rnd . choice ( koho ) ,
rnd . choice ( ceho ) ,
rnd . choice ( jmeno ) ,
rnd . choice ( kde ) ]
)
text_zadani = Text . objects . create (
na_web = obsah ,
do_cisla = obsah ,
)
zad = TextNode . objects . create ( text = text_zadani , root = tema . temavcislenode_set . first ( ) . root )
uloha_zadani = UlohaZadaniNode . objects . create ( uloha = uloha , first_child = zad , root = tema . temavcislenode_set . first ( ) . root )
uloha . ulohazadaninode = uloha_zadani
# Generování řešení a hodnocení k úloze
gen_reseni_ulohy ( rnd , [ cislo ] , uloha , len ( resitele ) / / 4 , 1 ,
resitele , resitele )
return uloha , uloha_zadani
def gen_ulohy_k_tematum ( rnd , rocniky , rocnik_cisla , rocnik_temata , organizatori , resitele ) :
logger . info ( ' Generuji úlohy k tématům... ' )
# Ke každému ročníku si vezmeme příslušná čísla a témata
for rocnik , cisla , temata in zip ( rocniky , rocnik_cisla , rocnik_temata ) :
# Do každého čísla nagenerujeme ke každému témátku pár úložek
for cislo in cisla :
print ( " Generuji úložky do {} -tého čísla " . format ( cislo . poradi ) )
# Vzorák bude o dvě čísla dál
cislo_se_vzorakem = Cislo . objects . filter (
rocnik = rocnik ,
poradi = str ( int ( cislo . poradi ) + 2 ) ,
)
# Pokud není číslo pro vzorák, tak se dá do posledního čísla
# (i kdyby tam mělo být zadání i řešení...)
# Tohle sice umožňuje vygenerovat vzorák do čísla dávno po konci témátka,
# ale to nám pro jednoduchost nevadí.
if len ( cislo_se_vzorakem ) == 0 :
cislo_se_vzorakem = cisla [ - 1 ]
else :
cislo_se_vzorakem = cislo_se_vzorakem . first ( )
for tema_node in all_children_of_type ( cislo . cislonode , TemaVCisleNode ) :
tema = tema_node . tema
# Pokud už témátko skončilo, žádné úložky negenerujeme
# FIXME: Bylo by hezčí, kdyby se čísla předávala jako Cislo a ne
# jako int v té trojici (start, konec, tema)
if not temata [ int ( tema . kod ) - 1 ] [ 1 ] > = int ( cislo_se_vzorakem . poradi ) :
continue
# Generujeme 1 až 4 úložky k tomuto témátku do tohoto čísla.
for kod in range ( 1 , rnd . randint ( 1 , 4 ) ) :
u , uz = gen_ulohy_tematu ( rnd , organizatori , resitele , tema , kod ,
cislo , cislo_se_vzorakem )
insert_last_child ( tema_node , uz )
poc_op = rnd . randint ( 1 , 4 )
uvz = gen_vzoroveho_reseni_ulohy ( rnd , organizatori ,
u , poc_op )
# Najdeme správný TemaVCisleNode pro vložení vzoráku
res_tema_node = None ;
for node in all_children ( cislo_se_vzorakem . cislonode ) :
if isinstance ( node , TemaVCisleNode ) :
if node . tema == tema :
res_tema_node = node
if res_tema_node is None :
raise LookupError ( " Nenalezen Node pro vložení vzoráku " )
insert_last_child ( res_tema_node , uvz )
u . save ( )
return
def gen_novinky ( rnd , organizatori ) : def gen_novinky ( rnd , organizatori ) :
logger . info ( ' Generuji novinky... ' ) logger . info ( ' Generuji novinky... ' )
@ -725,86 +37,16 @@ def gen_novinky(rnd, organizatori):
novinka . save ( ) novinka . save ( )
return return
def otec_syn ( otec , syn ) :
bratr = otec . first_child
syn . succ = bratr
otec . first_child = syn
syn . save ( )
otec . save ( )
def gen_clanek ( rnd , organizatori , resitele ) :
logger . info ( " Generuji článek do čísla 22.2 " )
clanek = Clanek . objects . create (
nazev = " Článek o Lorem ipsum " ,
nadproblem = None ,
stav = ' vyreseny ' ,
zamereni = [ ' I ' ] ,
garant = rnd . choice ( organizatori ) ,
kod = ' cl ' ,
)
clanek . save ( )
reseni = Reseni . objects . create (
zverejneno = True ,
)
reseni . resitele . add ( rnd . choice ( resitele ) )
reseni . save ( )
cislo = Cislo . objects . get ( rocnik__rocnik = 22 , poradi = 2 )
cislonode = cislo . cislonode
hodnoceni = Hodnoceni . objects . create (
body = 15.0 ,
cislo_body = cislo ,
reseni = reseni ,
problem = clanek ,
)
hodnoceni . save ( )
reseninode = ReseniNode . objects . create (
reseni = reseni
)
reseninode . save ( )
# Bude to celý text
reseni . text_cely = reseninode
reseni . save ( )
from treenode . treelib import insert_last_child , create_child
insert_last_child ( cislonode , reseninode )
# Vyrobíme nějaký obsah
# FIXME: Ten, kdo vymyslel TreeLib (mj. týž, kdo psal tenhle kód),
# nevyrobil vhodnou funkci, takže to postavíme pozpátku pomocí create_child
# (které vyrábí _prvního_ syna)
create_child ( reseninode , CastNode , nadpis = " Lorem ipsum " )
# Taky ten člověk nevyrobil vracení nových věcí...
castnode = reseninode . first_child
# Úvodní odstaveček
obsah = " Tohle je zamyšlení o textu lorem ipsum. Začneme a skončíme ukázkou. "
text = Text . objects . create (
na_web = obsah ,
do_cisla = obsah ,
)
text . save ( )
create_child ( reseninode , TextNode , text = text )
# Několik odstavců lorem ipsum
for _ in range ( rnd . randint ( 3 , 7 ) ) :
lipsum = lorem . paragraph ( )
text = Text . objects . create (
na_web = lipsum ,
do_cisla = lipsum ,
)
text . save ( )
create_child ( castnode , TextNode , text = text )
logger . info ( f " Článek vygenerován (reseni= { reseni . id } , treenode= { reseninode . id } ) " )
@transaction . atomic @transaction . atomic
def create_test_data ( size = 6 , rnd = None ) : def create_test_data ( size = 6 , rnd = None ) :
from korektury . testutils import create_test_pdf
from personalni . testutils import gen_organizatori , gen_osoby , gen_skoly , gen_prijemci , gen_resitele
from tvorba . testutils import gen_clanek , gen_cisla , gen_temata , gen_rocniky , gen_ulohy_do_cisla , gen_dlouhe_tema , gen_ulohy_k_tematum
from soustredeni . testutils import gen_soustredeni , gen_konfery
from personalni . models import Osoba , Organizator
from tvorba . models import Cislo , Rocnik
logger . info ( ' Vyrábím testovací data (size= {} )... ' . format ( size ) ) logger . info ( ' Vyrábím testovací data (size= {} )... ' . format ( size ) )
assert size > = 1 assert size > = 1
@ -844,7 +86,7 @@ def create_test_data(size = 6, rnd = None):
print ( users ) print ( users )
# skoly # skoly
skoly = gen_skoly ( ) skoly , zlinska = gen_skoly ( )
# osoby # osoby
osoby = gen_osoby ( rnd , size ) osoby = gen_osoby ( rnd , size )
@ -860,7 +102,6 @@ def create_test_data(size = 6, rnd = None):
# prijemci # prijemci
prijemci = gen_prijemci ( rnd , osoby ) prijemci = gen_prijemci ( rnd , osoby )
global zlinska
zlinska . kontaktni_osoba = rnd . choice ( osoby ) zlinska . kontaktni_osoba = rnd . choice ( osoby )
zlinska . save ( ) zlinska . save ( )
