mirror of https://github.com/zulip/zulip.git
57 lines
1.9 KiB
Python
57 lines
1.9 KiB
Python
|
from __future__ import absolute_import
|
||
|
|
from optparse import make_option
|
||
|
|
|
||
|
|
from django.core.management.base import BaseCommand
|
||
|
|
from zerver.models import RealmEmoji, Realm
|
||
|
|
from zerver.lib.actions import do_add_realm_emoji, do_remove_realm_emoji
|
||
|
|
import sys
|
||
|
|
|
||
|
|
class Command(BaseCommand):
|
||
|
|
help = """Manage emoji for the specified realm
|
||
|
|
|
||
|
|
Usage: python manage.py realm_emoji foo.com NAME IMG_URL
|
||
|
|
|
||
|
|
Example: python manage.py realm_emoji --realm=zulip.com --op=add robotheart https://humbug-user-avatars.s3.amazonaws.com/95ffa70fe0e7aea3c052ba91b38a28d8779f5705
|
||
|
|
Example: python manage.py realm_emoji --realm=zulip.com --op=remove robotheart
|
||
|
|
Example: python manage.py realm_emoji --realm=zulip.com --op=show
|
||
|
|
"""
|
||
|
|
|
||
|
|
option_list = BaseCommand.option_list + (
|
||
|
|
make_option('-r', '--realm',
|
||
|
|
dest='domain',
|
||
|
|
type='str',
|
||
|
|
help='The name of the realm.'),
|
||
|
|
make_option('--op',
|
||
|
|
dest='op',
|
||
|
|
type='str',
|
||
|
|
default="show",
|
||
|
|
help='What operation to do (add, show, remove).'),
|
||
|
|
)
|
||
|
|
|
||
|
|
def handle(self, *args, **options):
|
||
|
|
if "domain" not in options:
|
||
|
|
self.print_help("python manage.py", "realm_emoji")
|
||
|
|
sys.exit(1)
|
||
|
|
|
||
|
|
realm = Realm.objects.get(domain=options["domain"])
|
||
|
|
if options["op"] == "show":
|
||
|
|
for name, url in realm.get_emoji().iteritems():
|
||
|
|
print name, url
|
||
|
|
sys.exit(0)
|
||
|
|
|
||
|
|
if not args:
|
||
|
|
self.print_help("python manage.py", "realm_emoji")
|
||
|
|
sys.exit(1)
|
||
|
|
|
||
|
|
name = args[0]
|
||
|
|
if options["op"] == "add":
|
||
|
|
img_url = args[1]
|
||
|
|
do_add_realm_emoji(realm, name, img_url)
|
||
|
|
sys.exit(0)
|
||
|
|
elif options["op"] == "remove":
|
||
|
|
do_remove_realm_emoji(realm, name)
|
||
|
|
sys.exit(0)
|
||
|
|
else:
|
||
|
|
self.print_help("python manage.py", "realm_emoji")
|
||
|
|
sys.exit(1)
|