mirror of https://github.com/zulip/zulip.git
37 lines
1.2 KiB
Python
37 lines
1.2 KiB
Python
import argparse
|
|
import os
|
|
from typing import Any
|
|
|
|
from django.core.management.base import BaseCommand
|
|
from django.db import DEFAULT_DB_ALIAS
|
|
|
|
from scripts.lib.zulip_tools import get_dev_uuid_var_path
|
|
from zerver.lib.test_fixtures import get_migration_status
|
|
|
|
|
|
class Command(BaseCommand):
|
|
help = "Get status of migrations."
|
|
|
|
def add_arguments(self, parser: argparse.ArgumentParser) -> None:
|
|
parser.add_argument(
|
|
"app_label", nargs="?", help="App label of an application to synchronize the state."
|
|
)
|
|
|
|
parser.add_argument(
|
|
"--database",
|
|
default=DEFAULT_DB_ALIAS,
|
|
help="Nominates a database to synchronize. " 'Defaults to the "default" database.',
|
|
)
|
|
|
|
parser.add_argument("--output", help="Path to store the status to (default to stdout).")
|
|
|
|
def handle(self, *args: Any, **options: Any) -> None:
|
|
result = get_migration_status(**options)
|
|
if options["output"] is not None:
|
|
uuid_var_path = get_dev_uuid_var_path()
|
|
path = os.path.join(uuid_var_path, options["output"])
|
|
with open(path, "w") as f:
|
|
f.write(result)
|
|
else:
|
|
self.stdout.write(result)
|