zulip/zilencer/management/commands/print_initial_password.py

28 lines
1.1 KiB
Python

from argparse import ArgumentParser
from typing import Any
from zerver.lib.initial_password import initial_password
from zerver.lib.management import ZulipBaseCommand
from zerver.lib.users import get_api_key
class Command(ZulipBaseCommand):
help = "Print the initial password and API key for accounts as created by populate_db"
fmt = '%-30s %-16s %-32s'
def add_arguments(self, parser: ArgumentParser) -> None:
parser.add_argument('emails', metavar='<email>', nargs='*',
help="email of user to show password and API key for")
self.add_realm_args(parser)
def handle(self, *args: Any, **options: str) -> None:
realm = self.get_realm(options)
print(self.fmt % ('email', 'password', 'API key'))
for email in options['emails']:
if '@' not in email:
print(f'ERROR: {email} does not look like an email address')
continue
user = self.get_user(email, realm)
print(self.fmt % (email, initial_password(email), get_api_key(user)))