mirror of https://github.com/zulip/zulip.git
28 lines
1.1 KiB
Python
28 lines
1.1 KiB
Python
from argparse import ArgumentParser
|
|
from typing import Any
|
|
|
|
from zerver.actions.streams import merge_streams
|
|
from zerver.lib.management import ZulipBaseCommand
|
|
from zerver.models import get_stream
|
|
|
|
|
|
class Command(ZulipBaseCommand):
|
|
help = """Merge two streams."""
|
|
|
|
def add_arguments(self, parser: ArgumentParser) -> None:
|
|
parser.add_argument("stream_to_keep", help="name of stream to keep")
|
|
parser.add_argument(
|
|
"stream_to_destroy", help="name of stream to merge into the stream being kept"
|
|
)
|
|
self.add_realm_args(parser, required=True)
|
|
|
|
def handle(self, *args: Any, **options: str) -> None:
|
|
realm = self.get_realm(options)
|
|
assert realm is not None # Should be ensured by parser
|
|
stream_to_keep = get_stream(options["stream_to_keep"], realm)
|
|
stream_to_destroy = get_stream(options["stream_to_destroy"], realm)
|
|
stats = merge_streams(realm, stream_to_keep, stream_to_destroy)
|
|
print(f"Added { stats[0] } subscriptions")
|
|
print(f"Moved { stats[1] } messages")
|
|
print(f"Deactivated { stats[2] } subscriptions")
|