Tomas Gavenciak
10 years ago
7 changed files with 109 additions and 5 deletions
Binary file not shown.
@ -0,0 +1,99 @@ |
|||
# -*- coding: utf-8 -*- |
|||
|
|||
import datetime |
|||
import os |
|||
import random |
|||
|
|||
from django.core.management.base import NoArgsCommand |
|||
from django.core.management import call_command |
|||
from django.conf import settings |
|||
|
|||
class Command(NoArgsCommand): |
|||
help = "Clear database and load testing data." |
|||
|
|||
def handle_noargs(self, **options): |
|||
assert settings.DEBUG == True |
|||
dbfile = settings.DATABASES['default']['NAME'] |
|||
if os.path.exists(dbfile): |
|||
os.rename(dbfile, dbfile + '.old') |
|||
self.stderr.write('Stara databaze prejmenovana na "%s"' % (dbfile + '.old')) |
|||
call_command('migrate', noinput=True) |
|||
self.load_testing_data() |
|||
|
|||
def load_testing_data(self): |
|||
from seminar import models as sm |
|||
import django.contrib.auth |
|||
User = django.contrib.auth.get_user_model() |
|||
|
|||
self.stderr.write('Vytvarim uzivatele "admin" (heslo "admin")') |
|||
|
|||
# pevna pseudo-nahodnost |
|||
rnd = random.Random(x=42) |
|||
|
|||
# users |
|||
admin = User.objects.create_superuser(username='admin', email='', password='admin') |
|||
|
|||
self.stderr.write('Vytvarim pseudo-nahodna data') |
|||
|
|||
orgs = [] |
|||
for org in ['anet', 'bara', 'cyril', 'david', 'eva', 'filip']: |
|||
o = User.objects.create_user(username=org, password=org) |
|||
o.first_name = org.capitalize() |
|||
o.save() |
|||
orgs.append(o) |
|||
|
|||
# skoly |
|||
sm.Skola.objects.create(mesto = u'Praha', stat='CZ', psc='101 00', ulice=u'Krátká 5', nazev=u'První SŠ') |
|||
sm.Skola.objects.create(mesto = u'Praha', stat='CZ', psc='102 00', ulice=u'Dlouhá 5', nazev=u'Druhá SŠ') |
|||
sm.Skola.objects.create(mesto = u'Praha', stat='CZ', psc='103 00', ulice=u'Široká 3', nazev=u'Třetí SŠ') |
|||
sm.Skola.objects.create(mesto = u'Ostrava', stat='CZ', psc='700 00', ulice=u'Hluboká 42', nazev=u'Hutní gympl') |
|||
sm.Skola.objects.create(mesto = u'Humenné', stat='SK', psc='012 34', ulice=u'Pltká 1', nazev=u'Sredná škuola') |
|||
|
|||
# resitele |
|||
for i in range(20): |
|||
skola = rnd.choice(sm.Skola.objects.all()) |
|||
pohlavi = rnd.randint(0,1) |
|||
sm.Resitel.objects.create(skola = skola, datum_prihlaseni = datetime.date(rnd.randint(2002, 2014), rnd.randint(1,12), 1), |
|||
jmeno = u'Řešitel' if pohlavi else u'Řešitelka', prijmeni = 'Číslo-%s' % (i), rok_maturity = rnd.randint(2015, 2019), |
|||
stat = skola.stat, zasilat = sm.Resitel.ZASILAT_NIKAM, pohlavi_muz = pohlavi) |
|||
|
|||
# rocniky |
|||
for ri in range(17, 22): |
|||
r = sm.Rocnik.objects.create(prvni_rok = 1993 + ri, rocnik = ri) |
|||
|
|||
# cisla |
|||
cisel = rnd.randint(4, 6) |
|||
cs = {} |
|||
for ci in range(1, cisel + 1): |
|||
vydano = datetime.date(r.prvni_rok, ci + 6, 1) |
|||
deadline = datetime.date(r.prvni_rok, ci + 8, 1) if ci + 2 < cisel else None |
|||
c = sm.Cislo.objects.create(rocnik = r, cislo = str(ci), datum_vydani=vydano, datum_deadline=deadline) |
|||
cs[ci] = c |
|||
|
|||
# problemy resene v ci |
|||
seq='#ABCDEFGHIJKLMNOPQRSTUVWXYZ' |
|||
if ci >= 3: |
|||
for pi in range(1, 4): |
|||
p = sm.Problem.objects.create(autor = rnd.choice(orgs), cislo_zadani=cs[ci-2], cislo_reseni=cs[ci], |
|||
opravovatel = rnd.choice(orgs), kod = str(pi), nazev = u'Dummy úloha %s-%s' % (seq[ci], seq[pi]), |
|||
stav = sm.Problem.STAV_ZADANY, typ = sm.Problem.TYP_ULOHA, body = rnd.randint(1, 5)) |
|||
|
|||
for resi in range(rnd.randint(0, 7)): |
|||
res = sm.Reseni.objects.create(problem = p, resitel = rnd.choice(sm.Resitel.objects.all()), |
|||
body = rnd.randint(0, p.body), cislo_body = cs[ci]) |
|||
|
|||
nastaveni = sm.Nastaveni.objects.create(aktualni_rocnik = sm.Rocnik.objects.last(), |
|||
aktualni_cislo = sm.Cislo.objects.last()) |
|||
|
|||
self.stderr.write('Vytvoreno %d uzivatelu, %d skol, %d resitelu, %d rocniku, %d cisel, %d problemu, %d reseni.' % |
|||
(User.objects.count(), sm.Skola.objects.count(), sm.Resitel.objects.count(), sm.Rocnik.objects.count(), |
|||
sm.Cislo.objects.count(), sm.Problem.objects.count(), sm.Reseni.objects.count())) |
|||
|
|||
|
|||
|
|||
|
|||
|
|||
|
|||
|
|||
|
|||
|
Loading…
Reference in new issue