Skip to content

Commit

Permalink
Merge pull request #1697 from doccano/enhancement/removeSubprocess
Browse files Browse the repository at this point in the history
Replace subprocess with call_command
  • Loading branch information
Hironsan committed Feb 21, 2022
2 parents bf5506e + 01e9cfb commit 50077b1
Showing 1 changed file with 9 additions and 19 deletions.
28 changes: 9 additions & 19 deletions backend/cli.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,16 +2,18 @@
import multiprocessing
import os
import platform
import subprocess
import sys

import django
from django.core import management

from .config.celery import app

os.environ["STANDALONE"] = "True"
os.environ.setdefault("DJANGO_SETTINGS_MODULE", "config.settings.production")
base = os.path.abspath(os.path.dirname(__file__))
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.")


Expand Down Expand Up @@ -56,27 +58,15 @@ def run_on_windows(args):

def command_db_init(args):
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):
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
)


Expand Down

0 comments on commit 50077b1

Please sign in to comment.