zulip/zerver/management/commands/bulk_change_user_name.py

31 lines
1.2 KiB
Python
Raw Normal View History

from argparse import ArgumentParser
from typing import Any
from django.core.management.base import CommandError
from zerver.lib.actions import do_change_full_name
from zerver.lib.management import ZulipBaseCommand
class Command(ZulipBaseCommand):
help = """Change the names for many users."""
def add_arguments(self, parser: ArgumentParser) -> None:
parser.add_argument('data_file', metavar='<data file>', type=str,
help="file containing rows of the form <email>,<desired name>")
self.add_realm_args(parser, True)
def handle(self, *args: Any, **options: str) -> None:
data_file = options['data_file']
realm = self.get_realm(options)
with open(data_file, "r") as f:
for line in f:
email, new_name = line.strip().split(",", 1)
try:
user_profile = self.get_user(email, realm)
old_name = user_profile.full_name
print("%s: %s -> %s" % (email, old_name, new_name))
do_change_full_name(user_profile, new_name, None)
except CommandError:
print("e-mail %s doesn't exist in the realm %s, skipping" % (email, realm))