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

zerver/management/commands/convert_microsoft_teams_data.py

@@ -0,0 +1,78 @@
+import argparse
+import os
+import tempfile
+from typing import Any
+
+from django.conf import settings
+from django.core.management.base import CommandError, CommandParser
+from typing_extensions import override
+
+from zerver.data_import.microsoft_teams import do_convert_directory
+from zerver.lib.management import ZulipBaseCommand
+
+
+class Command(ZulipBaseCommand):
+    help = """Convert the Microsoft Teams data into Zulip data format."""
+
+    @override
+    def add_arguments(self, parser: CommandParser) -> None:
+        parser.add_argument(
+            "microsoft_teams_data_path",
+            nargs="+",
+            metavar="<Microsoft Teams data path>",
+            help="Zipped Microsoft Teams data or directory",
+        )
+
+        parser.add_argument(
+            "--output", dest="output_dir", help="Directory to write exported data to."
+        )
+
+        parser.add_argument(
+            "--token",
+            metavar="<microsoft_graph_api_token>",
+            help="Microsoft Graph API token, see https://learn.microsoft.com/en-us/graph/auth-v2-service?tabs=http",
+        )
+
+        parser.add_argument(
+            "--processes",
+            default=settings.DEFAULT_DATA_EXPORT_IMPORT_PARALLELISM,
+            help="Processes to use in exporting UserMessage objects in parallel",
+        )
+
+        parser.formatter_class = argparse.RawTextHelpFormatter
+
+    @override
+    def handle(self, *args: Any, **options: Any) -> None:
+        output_dir = options["output_dir"]
+        if output_dir is None:
+            output_dir = tempfile.mkdtemp(prefix="converted-ms-teams-data-")
+        else:
+            output_dir = os.path.realpath(output_dir)
+
+        token = options["token"]
+        if token is None:
+            raise CommandError("Enter Microsoft Graph API token!")
+
+        num_processes = int(options["processes"])
+        if num_processes < 1:
+            raise CommandError("You must have at least one thread.")
+
+        for path in options["microsoft_teams_data_path"]:
+            if not os.path.exists(path):
+                raise CommandError(f"Microsoft Teams data file or directory not found: '{path}'")
+
+            print("Converting data ...")
+            if os.path.isdir(path):
+                print(path)
+                do_convert_directory(
+                    path,
+                    output_dir,
+                    token,
+                    processes=num_processes,
+                )
+            elif os.path.isfile(path) and path.endswith(".zip"):
+                raise ValueError(
+                    "Importing .zip Microsoft Teams data is not yet supported, please try again with the extracted data."
+                )
+            else:
+                raise ValueError(f"Don't know how to import Microsoft Teams data from {path}")