Browse Source

Replace subprocess with call_command

pull/1697/head
Hironsan 2 years ago
parent
commit
01e9cfb504
1 changed files with 9 additions and 19 deletions
  1. 28
      backend/cli.py

28
backend/cli.py

@ -2,16 +2,18 @@ import argparse
import multiprocessing import multiprocessing
import os import os
import platform import platform
import subprocess
import sys import sys
import django
from django.core import management
from .config.celery import app from .config.celery import app
os.environ["STANDALONE"] = "True" os.environ["STANDALONE"] = "True"
os.environ.setdefault("DJANGO_SETTINGS_MODULE", "config.settings.production") os.environ.setdefault("DJANGO_SETTINGS_MODULE", "config.settings.production")
base = os.path.abspath(os.path.dirname(__file__)) base = os.path.abspath(os.path.dirname(__file__))
sys.path.append(base) sys.path.append(base)
manage_path = os.path.join(base, "manage.py")
django.setup()
parser = argparse.ArgumentParser(description="doccano, text annotation for machine learning practitioners.") parser = argparse.ArgumentParser(description="doccano, text annotation for machine learning practitioners.")
@ -56,27 +58,15 @@ def run_on_windows(args):
def command_db_init(args): def command_db_init(args):
print("Setup Database.") print("Setup Database.")
subprocess.call([sys.executable, manage_path, "wait_for_db"], shell=False)
subprocess.call([sys.executable, manage_path, "migrate"], shell=False)
subprocess.call([sys.executable, manage_path, "create_roles"], shell=False)
management.call_command("wait_for_db")
management.call_command("migrate")
management.call_command("create_roles")
def command_user_create(args): def command_user_create(args):
print("Create admin user.") print("Create admin user.")
subprocess.call(
[
sys.executable,
manage_path,
"create_admin",
"--username",
args.username,
"--password",
args.password,
"--email",
args.email,
"--noinput",
],
shell=False,
management.call_command(
"create_admin", "--noinput", username=args.username, password=args.password, email=args.email
) )

Loading…
Cancel
Save