management: Add option to set description of default stream groups.

This commit is contained in:
Vishnu Ks 2017-11-16 04:55:38 +05:30 committed by Tim Abbott
parent c84f1686b7
commit f99fd20f8f
1 changed files with 20 additions and 10 deletions

View File

@ -17,6 +17,22 @@ Create default stream groups which the users can choose during sign up.
# type: (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',
@ -24,14 +40,6 @@ Create default stream groups which the users can choose during sign up.
required=True,
help='A comma-separated list of stream names.')
parser.add_argument(
'-d', '--default-stream-group',
dest='default_stream_group',
type=str,
required=True,
help='Name of the group you want to create.'
)
def handle(self, *args, **options):
# type: (*Any, **Any) -> None
realm = self.get_realm(options)
@ -45,15 +53,17 @@ Create default stream groups which the users can choose during sign up.
try:
default_stream_group = DefaultStreamGroup.objects.get(
name=options["default_stream_group"], realm=realm)
name=options["name"], realm=realm, description=options["description"])
except DefaultStreamGroup.DoesNotExist:
default_stream_group = DefaultStreamGroup.objects.create(
name=options["default_stream_group"], realm=realm)
name=options["name"], realm=realm, description=options["description"])
default_stream_group.streams = streams
default_stream_group.save()
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("")