setup.py 1.61 KB
Newer Older
Skia's avatar
Skia committed
1
import os
Skia's avatar
Skia committed
2
3
from django.core.management.base import BaseCommand, CommandError
from django.core.management import call_command
4
from django.conf import settings
Skia's avatar
Skia committed
5
6
7
8
9
from core.models import Group, User

class Command(BaseCommand):
    help = "Set up a new instance of the Sith AE"

Skia's avatar
Skia committed
10
11
12
    def add_arguments(self, parser):
        parser.add_argument('--prod', action="store_true")

Skia's avatar
Skia committed
13
14
    def handle(self, *args, **options):
        try:
Skia's avatar
Skia committed
15
            os.unlink(os.path.join(os.path.dirname(os.path.dirname(os.path.dirname(os.path.dirname(__file__)))), 'db.sqlite3'))
Skia's avatar
Skia committed
16
17
18
19
20
21
22
23
24
        except:
            pass
        call_command('migrate')
        u = User(username='root', last_name="", first_name="Bibou",
                 email="ae.info@utbm.fr",
                 date_of_birth="1942-06-12T00:00:00+01:00",
                 is_superuser=True, is_staff=True)
        u.set_password("plop")
        u.save()
25
26
        for g in settings.AE_GROUPS.values():
            Group(id=g['id'], name=g['name']).save()
Skia's avatar
Skia committed
27
28
29
30
31
32
33
34
        if not options['prod']:
            print("Dev mode, adding some test data")
            u = User(username='skia', last_name="Kia", first_name="S'",
                     email="skia@git.an",
                     date_of_birth="1942-06-12T00:00:00+01:00",
                     is_superuser=True, is_staff=True)
            u.set_password("plop")
            u.save()
35
36
37
38
39
40
            u = User(username='guy', last_name="Carlier", first_name="Guy",
                     email="guy@git.an",
                     date_of_birth="1942-06-12T00:00:00+01:00",
                     is_superuser=False, is_staff=False)
            u.set_password("plop")
            u.save()
Skia's avatar
Skia committed
41