zulip/zephyr/management/commands/update_mit_fullnames.py

30 lines
1.2 KiB
Python

from __future__ import absolute_import
from optparse import make_option
from django.core.management.base import BaseCommand
from zephyr.models import UserProfile
from zephyr.lib.actions import compute_mit_user_fullname
# Helper to be used with manage.py shell to fix bad names on prod.
def update_mit_fullnames(change=False):
for u in UserProfile.objects.select_related().all():
if (u.is_active or u.realm.domain != "mit.edu"):
# Don't change fullnames for non-MIT users or users who
# actually have an account (is_active) and thus have
# presumably set their fullname how they like it.
continue
computed_name = compute_mit_user_fullname(u.email)
if u.full_name != computed_name:
print "%s: %s => %s" % (u.email, u.full_name, computed_name)
if change:
u.full_name = computed_name
u.save()
class Command(BaseCommand):
option_list = BaseCommand.option_list + (
make_option('--dry-run', '-n', dest='dry_run', default=False, action='store_true'),)
def handle(self, *args, **options):
update_mit_fullnames(change=not options['dry_run'])