gestioCOF/gestioncof/management/commands/loaddevdata.py

"""
Charge des données de test dans la BDD
- Utilisateurs
- Sondage
- Événement
- Petits cours
"""


import os
import random

from django.contrib.auth.models import User
from django.core.management import call_command

from gestioncof.management.base import MyBaseCommand
from gestioncof.petits_cours_models import (
    PetitCoursAbility, PetitCoursSubject, LEVELS_CHOICES,
    PetitCoursAttributionCounter
)

# Où sont stockés les fichiers json
DATA_DIR = os.path.join(os.path.dirname(os.path.dirname(__file__)),
                        'data')


class Command(MyBaseCommand):
    help = "Charge des données de test dans la BDD"

    def add_argument(self, parser):
        """
        Permet de ne pas créer l'utilisateur "root".
        """
        parser.add_argument(
            '--no-root',
            action='store_true',
            dest='no-root',
            default=False,
            help='Ne crée pas l\'utilisateur "root"'
        )

    def handle(self, *args, **options):
        # ---
        # Utilisateurs
        # ---

        # Gaulois
        gaulois = self.from_json('gaulois.json', DATA_DIR, User)
        for user in gaulois:
            user.profile.is_cof = True
            user.profile.save()

        # Romains
        self.from_json('romains.json', DATA_DIR, User)

        # Root
        no_root = options.get('no-root', False)
        if not no_root:
            self.stdout.write("Création de l'utilisateur root")
            root, _ = User.objects.get_or_create(
                username='root',
                first_name='super',
                last_name='user',
                email='root@localhost')
            root.set_password('root')
            root.is_staff = True
            root.is_superuser = True
            root.profile.is_cof = True
            root.profile.is_buro = True
            root.profile.save()
            root.save()

        # ---
        # Petits cours
        # ---

        self.stdout.write("Inscriptions au système des petits cours")
        levels = [id for (id, verbose) in LEVELS_CHOICES]
        subjects = list(PetitCoursSubject.objects.all())
        nb_of_teachers = 0
        for user in gaulois:
            if random.randint(0, 1):
                nb_of_teachers += 1
                # L'utilisateur reçoit les demandes de petits cours
                user.profile.petits_cours_accept = True
                user.save()
                # L'utilisateur est compétent dans une matière
                subject = random.choice(subjects)
                if not PetitCoursAbility.objects.filter(
                        user=user,
                        matiere=subject).exists():
                    PetitCoursAbility.objects.create(
                        user=user,
                        matiere=subject,
                        niveau=random.choice(levels),
                        agrege=bool(random.randint(0, 1))
                    )
                    # On initialise son compteur d'attributions
                    PetitCoursAttributionCounter.objects.get_or_create(
                        user=user,
                        matiere=subject
                    )
        self.stdout.write("- {:d} inscriptions".format(nb_of_teachers))

        # ---
        # Le BdA
        # ---

        call_command('loadbdadevdata')