snapshot of https://github.com/zulip/zulip.git @ 48c69911c7dd4e03b20552eb3bd77ad8151c1357 (2026-09-15)

zerver/management/commands/transfer_uploads_to_s3.py

@@ -0,0 +1,32 @@
+from typing import Any
+
+from django.conf import settings
+from django.core.management.base import CommandError, CommandParser
+from typing_extensions import override
+
+from zerver.lib.management import ZulipBaseCommand
+from zerver.lib.transfer import transfer_uploads_to_s3
+
+
+class Command(ZulipBaseCommand):
+    help = """Transfer uploads to S3 """
+
+    @override
+    def add_arguments(self, parser: CommandParser) -> None:
+        parser.add_argument(
+            "--processes",
+            default=settings.DEFAULT_DATA_EXPORT_IMPORT_PARALLELISM,
+            help="Processes to use for exporting uploads in parallel",
+        )
+
+    @override
+    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.")
+
+        if not settings.LOCAL_UPLOADS_DIR:
+            raise CommandError("Please set the value of LOCAL_UPLOADS_DIR.")
+
+        transfer_uploads_to_s3(num_processes)
+        print("Transfer to S3 completed successfully.")