mirror of
https://github.com/zulip/zulip.git
synced 2025-11-14 02:48:00 +00:00
2c01354569
This makes management commands more readable, since one doesn't need to know details of how the library works to read based code.
38 lines
1.5 KiB
Python
38 lines
1.5 KiB
Python
from argparse import ArgumentParser
|
|
from typing import Any, List
|
|
|
|
from zerver.lib.actions import do_create_multiuse_invite_link, ensure_stream
|
|
from zerver.lib.management import ZulipBaseCommand
|
|
from zerver.models import PreregistrationUser, Stream
|
|
|
|
|
|
class Command(ZulipBaseCommand):
|
|
help = "Generates invite link that can be used for inviting multiple users"
|
|
|
|
def add_arguments(self, parser: ArgumentParser) -> None:
|
|
self.add_realm_args(parser, required=True)
|
|
|
|
parser.add_argument("-s", "--streams", help="A comma-separated list of stream names.")
|
|
|
|
parser.add_argument(
|
|
"--referred-by",
|
|
help="Email of referrer",
|
|
required=True,
|
|
)
|
|
|
|
def handle(self, *args: Any, **options: Any) -> None:
|
|
realm = self.get_realm(options)
|
|
assert realm is not None # Should be ensured by parser
|
|
|
|
streams: List[Stream] = []
|
|
if options["streams"]:
|
|
stream_names = {stream.strip() for stream in options["streams"].split(",")}
|
|
for stream_name in set(stream_names):
|
|
stream = ensure_stream(realm, stream_name, acting_user=None)
|
|
streams.append(stream)
|
|
|
|
referred_by = self.get_user(options["referred_by"], realm)
|
|
invite_as = PreregistrationUser.INVITE_AS["MEMBER"]
|
|
invite_link = do_create_multiuse_invite_link(referred_by, invite_as, streams)
|
|
print(f"You can use {invite_link} to invite as many number of people to the organization.")
|