mirror of
https://github.com/zulip/zulip.git
synced 2025-11-08 16:01:58 +00:00
8e7ce7cc79
This is a preparatory commit for using isort for sorting all of our imports, merging changes to files where we can easily review the changes as something we're happy with. These are also files with relatively little active development, which means we don't expect much merge conflict risk from these changes.
67 lines
2.3 KiB
Python
67 lines
2.3 KiB
Python
from argparse import ArgumentParser
|
|
from typing import Any
|
|
|
|
from zerver.lib.actions import ensure_stream
|
|
from zerver.lib.management import ZulipBaseCommand
|
|
from zerver.models import DefaultStreamGroup
|
|
|
|
|
|
class Command(ZulipBaseCommand):
|
|
help = """
|
|
Create default stream groups which the users can choose during sign up.
|
|
|
|
./manage.py create_default_stream_groups -s gsoc-1,gsoc-2,gsoc-3 -d "Google summer of code" -r zulip
|
|
"""
|
|
|
|
def add_arguments(self, parser: ArgumentParser) -> None:
|
|
self.add_realm_args(parser, True)
|
|
|
|
parser.add_argument(
|
|
'-n', '--name',
|
|
dest='name',
|
|
type=str,
|
|
required=True,
|
|
help='Name of the group you want to create.'
|
|
)
|
|
|
|
parser.add_argument(
|
|
'-d', '--description',
|
|
dest='description',
|
|
type=str,
|
|
required=True,
|
|
help='Description of the group.'
|
|
)
|
|
|
|
parser.add_argument(
|
|
'-s', '--streams',
|
|
dest='streams',
|
|
type=str,
|
|
required=True,
|
|
help='A comma-separated list of stream names.')
|
|
|
|
def handle(self, *args: Any, **options: Any) -> None:
|
|
realm = self.get_realm(options)
|
|
assert realm is not None # Should be ensured by parser
|
|
|
|
streams = []
|
|
stream_names = set([stream.strip() for stream in options["streams"].split(",")])
|
|
for stream_name in set(stream_names):
|
|
stream = ensure_stream(realm, stream_name)
|
|
streams.append(stream)
|
|
|
|
try:
|
|
default_stream_group = DefaultStreamGroup.objects.get(
|
|
name=options["name"], realm=realm, description=options["description"])
|
|
except DefaultStreamGroup.DoesNotExist:
|
|
default_stream_group = DefaultStreamGroup.objects.create(
|
|
name=options["name"], realm=realm, description=options["description"])
|
|
default_stream_group.streams.set(streams)
|
|
|
|
default_stream_groups = DefaultStreamGroup.objects.all()
|
|
for default_stream_group in default_stream_groups:
|
|
print(default_stream_group.name)
|
|
print(default_stream_group.description)
|
|
for stream in default_stream_group.streams.all():
|
|
print(stream.name)
|
|
print("")
|