mirror of https://github.com/zulip/zulip.git
130 lines
5.0 KiB
Python
130 lines
5.0 KiB
Python
import os
|
|
import re
|
|
import tempfile
|
|
from argparse import ArgumentParser, RawTextHelpFormatter
|
|
from typing import Any
|
|
|
|
from django.conf import settings
|
|
from django.db import connection
|
|
from django.utils.timezone import now as timezone_now
|
|
|
|
from scripts.lib.zulip_tools import TIMESTAMP_FORMAT, parse_os_release, run
|
|
from version import ZULIP_VERSION
|
|
from zerver.lib.management import ZulipBaseCommand
|
|
from zerver.logging_handlers import try_git_describe
|
|
|
|
|
|
class Command(ZulipBaseCommand):
|
|
# Fix support for multi-line usage strings
|
|
def create_parser(self, *args: Any, **kwargs: Any) -> ArgumentParser:
|
|
parser = super().create_parser(*args, **kwargs)
|
|
parser.formatter_class = RawTextHelpFormatter
|
|
return parser
|
|
|
|
def add_arguments(self, parser: ArgumentParser) -> None:
|
|
parser.add_argument(
|
|
"--output", default=None, nargs="?", help="Filename of output tarball",
|
|
)
|
|
parser.add_argument("--skip-db", action='store_true', help="Skip database backup")
|
|
parser.add_argument("--skip-uploads", action='store_true', help="Skip uploads backup")
|
|
|
|
def handle(self, *args: Any, **options: Any) -> None:
|
|
timestamp = timezone_now().strftime(TIMESTAMP_FORMAT)
|
|
with tempfile.TemporaryDirectory(
|
|
prefix=f"zulip-backup-{timestamp}-",
|
|
) as tmp:
|
|
os.mkdir(os.path.join(tmp, "zulip-backup"))
|
|
members = []
|
|
paths = []
|
|
|
|
with open(os.path.join(tmp, "zulip-backup", "zulip-version"), "w") as f:
|
|
print(ZULIP_VERSION, file=f)
|
|
git = try_git_describe()
|
|
if git:
|
|
print(git, file=f)
|
|
members.append("zulip-backup/zulip-version")
|
|
|
|
with open(os.path.join(tmp, "zulip-backup", "os-version"), "w") as f:
|
|
print(
|
|
"{ID} {VERSION_ID}".format(**parse_os_release()),
|
|
file=f,
|
|
)
|
|
members.append("zulip-backup/os-version")
|
|
|
|
with open(os.path.join(tmp, "zulip-backup", "postgres-version"), "w") as f:
|
|
print(connection.pg_version, file=f)
|
|
members.append("zulip-backup/postgres-version")
|
|
|
|
if settings.DEVELOPMENT:
|
|
members.append(
|
|
os.path.join(settings.DEPLOY_ROOT, "zproject", "dev-secrets.conf"),
|
|
)
|
|
paths.append(
|
|
("zproject", os.path.join(settings.DEPLOY_ROOT, "zproject")),
|
|
)
|
|
else:
|
|
members.append("/etc/zulip")
|
|
paths.append(("settings", "/etc/zulip"))
|
|
|
|
if not options['skip_db']:
|
|
pg_dump_command = [
|
|
"pg_dump",
|
|
"--format=directory",
|
|
"--file", os.path.join(tmp, "zulip-backup", "database"),
|
|
"--host", settings.DATABASES["default"]["HOST"],
|
|
"--port", settings.DATABASES["default"]["PORT"],
|
|
"--username", settings.DATABASES["default"]["USER"],
|
|
"--dbname", settings.DATABASES["default"]["NAME"],
|
|
"--no-password",
|
|
]
|
|
os.environ["PGPASSWORD"] = settings.DATABASES["default"]["PASSWORD"]
|
|
|
|
run(
|
|
pg_dump_command,
|
|
cwd=tmp,
|
|
)
|
|
members.append("zulip-backup/database")
|
|
|
|
if not options['skip_uploads'] and settings.LOCAL_UPLOADS_DIR is not None and os.path.exists(
|
|
os.path.join(settings.DEPLOY_ROOT, settings.LOCAL_UPLOADS_DIR),
|
|
):
|
|
members.append(
|
|
os.path.join(settings.DEPLOY_ROOT, settings.LOCAL_UPLOADS_DIR),
|
|
)
|
|
paths.append(
|
|
(
|
|
"uploads",
|
|
os.path.join(settings.DEPLOY_ROOT, settings.LOCAL_UPLOADS_DIR),
|
|
),
|
|
)
|
|
|
|
assert not any("|" in name or "|" in path for name, path in paths)
|
|
transform_args = [
|
|
r"--transform=s|^{}(/.*)?$|zulip-backup/{}\1|x".format(
|
|
re.escape(path), name.replace("\\", r"\\"),
|
|
)
|
|
for name, path in paths
|
|
]
|
|
|
|
try:
|
|
if options["output"] is None:
|
|
tarball_path = tempfile.NamedTemporaryFile(
|
|
prefix=f"zulip-backup-{timestamp}-",
|
|
suffix=".tar.gz",
|
|
delete=False,
|
|
).name
|
|
else:
|
|
tarball_path = options["output"]
|
|
|
|
run(
|
|
["tar", "-C", tmp, "-cPzf", tarball_path,
|
|
*transform_args,
|
|
"--",
|
|
*members]
|
|
)
|
|
print(f"Backup tarball written to {tarball_path}")
|
|
except BaseException:
|
|
if options["output"] is None:
|
|
os.unlink(tarball_path)
|
|
raise
|