Sith/core/management/commands/setup.py

# -*- coding:utf-8 -*
#
# Copyright 2023 © AE UTBM
# ae@utbm.fr / ae.info@utbm.fr
#
# This file is part of the website of the UTBM Student Association (AE UTBM),
# https://ae.utbm.fr.
#
# You can find the source code of the website at https://github.com/ae-utbm/sith3
#
# LICENSED UNDER THE GNU GENERAL PUBLIC LICENSE VERSION 3 (GPLv3)
# SEE : https://raw.githubusercontent.com/ae-utbm/sith3/master/LICENSE
# OR WITHIN THE LOCAL FILE "LICENSE"
#
#

import os
from django.core.management.base import BaseCommand
from django.core.management import call_command


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.mkdir(os.path.join(root_path) + "/data")
            print("Data dir created")
        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)
        call_command("migrate")
        if options["prod"]:
            call_command("populate", "--prod")
        else:
            call_command("populate")