mirror of
https://github.com/ae-utbm/sith.git
synced 2024-11-15 02:33:22 +00:00
30 lines
1.0 KiB
Python
Executable File
30 lines
1.0 KiB
Python
Executable File
import os
|
|
from django.core.management.base import BaseCommand, CommandError
|
|
from django.core.management import call_command
|
|
from django.conf import settings
|
|
|
|
|
|
from core.models import Group, User, Page, PageRev
|
|
from accounting.models import Customer, GeneralJournal, ProductType, Product
|
|
from club.models import Club
|
|
from subscription.models import Subscription, Subscriber
|
|
|
|
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):
|
|
root_path = os.path.dirname(os.path.dirname(os.path.dirname(os.path.dirname(__file__))))
|
|
try:
|
|
os.unlink(os.path.join(root_path, 'db.sqlite3'))
|
|
os.mkdir(os.path.join(root_path)+'/data')
|
|
except Exception as e:
|
|
print(e)
|
|
call_command('migrate')
|
|
if options['prod']:
|
|
call_command('populate', '--prod')
|
|
else:
|
|
call_command('populate')
|