klmp200/Sith
1
0
Fork 0
mirror of https://github.com/ae-utbm/sith.git synced 2025-07-10 03:49:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

Improve setup script

Browse Source
This commit is contained in:
Skia
2015-12-04 16:13:20 +01:00
parent 078276cead
commit 185b35ed18
2 changed files with 15 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
core/management/commands/setup.py
View File

@ -1,3 +1,4 @@
import os
from django.core.management.base import BaseCommand, CommandError
from django.core.management import call_command
from core.models import Group, User
@ -5,10 +6,12 @@ from core.models import Group, User
class Command(BaseCommand):
help = "Set up a new instance of the Sith AE"
def add_arguments(self, parser):
parser.add_argument('--prod', action="store_true")
def handle(self, *args, **options):
try:
# TODO: really remove the DB, to make sure all the tables and indexes are dropped
call_command('flush', '--noinput', '--no-initial-data')
os.unlink(os.path.join(os.path.dirname(os.path.dirname(os.path.dirname(os.path.dirname(__file__)))), 'db.sqlite3'))
except:
pass
call_command('migrate')
@ -23,4 +26,12 @@ class Command(BaseCommand):
Group(name="bureau_restreint_ae").save()
Group(name="bureau_ae").save()
Group(name="membre_ae").save()
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()