mirror of
https://github.com/zulip/zulip.git
synced 2025-11-11 01:16:19 +00:00
8e7ce7cc79
This is a preparatory commit for using isort for sorting all of our imports, merging changes to files where we can easily review the changes as something we're happy with. These are also files with relatively little active development, which means we don't expect much merge conflict risk from these changes.
82 lines
3.4 KiB
Python
82 lines
3.4 KiB
Python
import argparse
|
|
import os
|
|
import subprocess
|
|
from typing import Any
|
|
|
|
from django.conf import settings
|
|
from django.core.management import call_command
|
|
from django.core.management.base import BaseCommand, CommandError, \
|
|
CommandParser
|
|
|
|
from zerver.forms import check_subdomain_available
|
|
from zerver.lib.import_realm import do_import_realm, do_import_system_bots
|
|
|
|
|
|
class Command(BaseCommand):
|
|
help = """Import extracted Zulip database dump directories into a fresh Zulip instance.
|
|
|
|
This command should be used only on a newly created, empty Zulip instance to
|
|
import a database dump from one or more JSON files."""
|
|
|
|
def add_arguments(self, parser: CommandParser) -> None:
|
|
parser.add_argument('--destroy-rebuild-database',
|
|
dest='destroy_rebuild_database',
|
|
default=False,
|
|
action="store_true",
|
|
help='Destroys and rebuilds the databases prior to import.')
|
|
|
|
parser.add_argument('--import-into-nonempty',
|
|
dest='import_into_nonempty',
|
|
default=False,
|
|
action="store_true",
|
|
help='Import into an existing nonempty database.')
|
|
|
|
parser.add_argument('subdomain', metavar='<subdomain>',
|
|
type=str, help="Subdomain")
|
|
|
|
parser.add_argument('export_paths', nargs='+',
|
|
metavar='<export path>',
|
|
help="list of export directories to import")
|
|
parser.add_argument('--processes',
|
|
dest='processes',
|
|
action="store",
|
|
default=6,
|
|
help='Number of processes to use for uploading Avatars to S3 in parallel')
|
|
parser.formatter_class = argparse.RawTextHelpFormatter
|
|
|
|
def do_destroy_and_rebuild_database(self, db_name: str) -> None:
|
|
call_command('flush', verbosity=0, interactive=False)
|
|
subprocess.check_call([os.path.join(settings.DEPLOY_ROOT, "scripts/setup/flush-memcached")])
|
|
|
|
def handle(self, *args: Any, **options: Any) -> None:
|
|
num_processes = int(options['processes'])
|
|
if num_processes < 1:
|
|
raise CommandError('You must have at least one process.')
|
|
|
|
subdomain = options['subdomain']
|
|
|
|
if options["destroy_rebuild_database"]:
|
|
print("Rebuilding the database!")
|
|
db_name = settings.DATABASES['default']['NAME']
|
|
self.do_destroy_and_rebuild_database(db_name)
|
|
elif options["import_into_nonempty"]:
|
|
print("NOTE: The argument 'import_into_nonempty' is now the default behavior.")
|
|
|
|
check_subdomain_available(subdomain, from_management_command=True)
|
|
|
|
paths = []
|
|
for path in options['export_paths']:
|
|
path = os.path.realpath(os.path.expanduser(path))
|
|
if not os.path.exists(path):
|
|
raise CommandError("Directory not found: '%s'" % (path,))
|
|
if not os.path.isdir(path):
|
|
raise CommandError("Export file should be folder; if it's a "
|
|
"tarball, please unpack it first.")
|
|
paths.append(path)
|
|
|
|
for path in paths:
|
|
print("Processing dump: %s ..." % (path,))
|
|
realm = do_import_realm(path, subdomain, num_processes)
|
|
print("Checking the system bots.")
|
|
do_import_system_bots(realm)
|