Merge branch 'data_migrations' of gimli.ms.mff.cuni.cz:/akce/mam/git/mamweb into data_migrations
This commit is contained in:
Kateřina Čížková
commit
2b4ade6ab6
1 changed files with 26 additions and 26 deletions
|
|
@ -13,7 +13,7 @@ from django.contrib.sites.models import Site
|
|||
|
||||
User = django.contrib.auth.get_user_model()
|
||||
|
||||
def gen_osoby(size):
|
||||
def gen_osoby(rnd, 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']
|
||||
|
|
@ -61,7 +61,7 @@ def gen_osoby(size):
|
|||
|
||||
|
||||
|
||||
def gen_skoly():
|
||||
def gen_skoly(): #TODO někdy to přepsat, aby jich bylo více
|
||||
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)
|
||||
|
|
@ -77,13 +77,13 @@ def gen_skoly():
|
|||
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',
|
||||
skoly.append(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):
|
||||
def gen_resitele(rnd, osoby, skoly):
|
||||
resitele = []
|
||||
for os in osoby:
|
||||
rand = rnd.randint(0, 8)
|
||||
|
|
@ -93,30 +93,30 @@ def gen_resitele(osoby, skoly):
|
|||
zasilat=rnd.choice(Resitel.ZASILAT_CHOICES)))
|
||||
return resitele
|
||||
|
||||
def gen_prijemci(osoby, kolik=10):
|
||||
def gen_prijemci(rnd, osoby, kolik=10):
|
||||
prijemci = []
|
||||
for i in range(kolik):
|
||||
rand_os = rnd.choice(osoby)
|
||||
prijemci.add(Prijemce.objects.create(osoba=rand_os))
|
||||
prijemci.append(Prijemce.objects.create(osoba=rand_os))
|
||||
return prijemci
|
||||
|
||||
def gen_organizatori(osoby, last_rocnik):
|
||||
def gen_organizatori(rnd, 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
|
||||
od = 1993 + 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,
|
||||
# user=rnd.choice(users),
|
||||
# organizuje_od=od, organizuje_do=do))
|
||||
return organizatori
|
||||
#aktualni organizatori jeste nemaji vyplnene organizuje_do
|
||||
if do > datetime.datetime.now().year:
|
||||
do = None
|
||||
organizatori.append(Organizator.objects.create(osoba=os,
|
||||
user=rnd.choice(users),
|
||||
organizuje_od=od, organizuje_do=do))
|
||||
return organizatri
|
||||
|
||||
def gen_ulohy_do_cisla(cislo, organizatori, resitele, slovnik_cisel, size):
|
||||
def gen_ulohy_do_cisla(rnd, 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á",
|
||||
|
|
@ -163,11 +163,11 @@ def gen_ulohy_do_cisla(cislo, organizatori, resitele, slovnik_cisel, size):
|
|||
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)
|
||||
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
|
||||
|
||||
def gen_soustredeni():
|
||||
|
|
@ -213,15 +213,15 @@ def create_test_data(size = 6, rnd = None):
|
|||
skoly = gen_skoly()
|
||||
|
||||
# osoby
|
||||
osoby = gen_osoby()
|
||||
osoby = gen_osoby(rnd, size)
|
||||
|
||||
# resitele a organizatori
|
||||
last_rocnik = 25
|
||||
resitele = gen_resitele(osoby)
|
||||
organizatori = gen_organizatori(osoby)
|
||||
resitele = gen_resitele(rnd, osoby, skoly)
|
||||
organizatori = gen_organizatori(rnd, osoby, last_rocnik)
|
||||
|
||||
# prijemci
|
||||
prijemci = gen_prijemci()
|
||||
prijemci = gen_prijemci(rnd)
|
||||
|
||||
zlinska.kontaktni_osoba=rnd.choice(osoby)
|
||||
zlinska.save()
|
||||
|
|
@ -251,7 +251,7 @@ def create_test_data(size = 6, rnd = None):
|
|||
slovnik_cisel[ci] = cislo
|
||||
|
||||
# generovani uloh
|
||||
gen_ulohy_do_cisla(ci, organizatori, resitele, size)
|
||||
gen_ulohy_do_cisla(rnd, ci, organizatori, resitele, slovnik_cisel, size)
|
||||
|
||||
# generovani temat
|
||||
if ci <= 3:
|
||||
|
|
|
|||
Loading…
Reference in a new issue