Sith/core/management/commands/setup.py

30 lines
951 B
Python
Raw Normal View History

2015-12-04 15:13:20 +00:00
import os
2015-12-03 17:00:08 +00:00
from django.core.management.base import BaseCommand, CommandError
from django.core.management import call_command
from django.conf import settings
2016-01-28 15:53:37 +00:00
2015-12-03 17:00:08 +00:00
class Command(BaseCommand):
help = "Set up a new instance of the Sith AE"
2015-12-04 15:13:20 +00:00
def add_arguments(self, parser):
parser.add_argument('--prod', action="store_true")
2015-12-03 17:00:08 +00:00
def handle(self, *args, **options):
2016-02-05 15:59:42 +00:00
root_path = os.path.dirname(os.path.dirname(os.path.dirname(os.path.dirname(__file__))))
2015-12-03 17:00:08 +00:00
try:
2016-02-05 15:59:42 +00:00
os.mkdir(os.path.join(root_path)+'/data')
print("Data dir created")
2016-01-28 15:53:37 +00:00
except Exception as e:
repr(e)
try:
os.remove(os.path.join(root_path, 'db.sqlite3'))
print("db.sqlite3 deleted")
except Exception as e:
repr(e)
2015-12-03 17:00:08 +00:00
call_command('migrate')
2016-03-22 08:01:24 +00:00
if options['prod']:
call_command('populate', '--prod')
else:
call_command('populate')