zulip/zerver/management/commands/sync_api_key.py

from __future__ import absolute_import

from django.core.management.base import BaseCommand
from zerver.models import get_user_profile_by_email
import os
from ConfigParser import SafeConfigParser

class Command(BaseCommand):
    help = """Reset all colors for a person to the default grey"""

    def handle(self, *args, **options):
        config_file = os.path.join(os.environ["HOME"], ".zuliprc")
        if not os.path.exists(config_file):
            raise RuntimeError("No ~/.zuliprc found")
        config = SafeConfigParser()
        with file(config_file, 'r') as f:
            config.readfp(f, config_file)
        api_key = config.get("api", "key")
        email = config.get("api", "email")

        user_profile = get_user_profile_by_email(email)
        user_profile.api_key = api_key
        user_profile.save(update_fields=["api_key"])