mamweb/seminar/management/commands/testdata.py

# -*- coding: utf-8 -*-

import datetime
import os
import random

from django.core.management.base import BaseCommand
from django.core.management import call_command
from django.conf import settings

from seminar.models import Skola, Resitel, Rocnik, Cislo, Problem, Reseni, PrilohaReseni, Nastaveni
from seminar.testutils import create_test_data
import django.contrib.auth
User = django.contrib.auth.get_user_model()


class Command(BaseCommand):
    help = "Clear database and load testing data."

    def handle(self, *args, **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', no_input=True)
        self.stdout.write('Vytvarim uzivatele "admin" (heslo "admin") a pseudo-nahodna data ...')
        create_test_data(size=8)
        self.stdout.write('Vytvoreno {} uzivatelu, {} skol, {} resitelu, {} rocniku, {} cisel,' 
               ' {} problemu, {} reseni.'.format(User.objects.count(), Skola.objects.count(), 
                    Resitel.objects.count(), Rocnik.objects.count(), Cislo.objects.count(), 
                    Problem.objects.count(), Reseni.objects.count()))