zulip/zilencer/management/commands/print_initial_password.py

36 lines
1.1 KiB
Python

from argparse import ArgumentParser
from typing import Any
from typing_extensions import override
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"
@override
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)
@override
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)))