from django.core.management import BaseCommand
import pydash as _; _.map = _.map_; _.filter = _.filter_
import random

from archilance import util
from specializations.models import Specialization


class Command(BaseCommand):
    def handle(self, *args, **options):
        print('---------------------------------------')
        print('Generating specializations...')
        print('---------------------------------------')
        
        
        _root = Specialization.objects.create(name='_root')
        
        stages = ('A','B','C','D')
        
        for s1 in stages:
            x = Specialization.objects.create(name='Специализация %s' % s1, parent=_root)
            for s2 in stages:
                y = Specialization.objects.create(name='Специализация %s-%s' % (s1,s2), parent=x)
                for s3 in stages:
                    z = Specialization.objects.create(name='Специализация %s-%s-%s' % (s1,s2,s3),  parent=y)
                    for s4 in stages:
                        Specialization.objects.create(name='Специализация %s-%s-%s-%s' % (s1,s2,s3,s4), parent=z)