Browse Source

Merge pull request #1697 from doccano/enhancement/removeSubprocess

Replace subprocess with call_command
pull/1698/head
Hiroki Nakayama 2 years ago
committed by GitHub
parent
commit
50077b181b
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
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