from argparse import Namespace from typing import Any, Final from mdrsclient.commands.base import BaseCommand from mdrsclient.config import ConfigFile from mdrsclient.connection import MDRSConnection from mdrsclient.exceptions import MissingConfigurationException class WhoamiCommand(BaseCommand): ANONYMOUS_USERNAME: Final[str] = "(Anonymous)" @classmethod def register(cls, parsers: Any) -> None: whoami_parser = parsers.add_parser("whoami", help="show current user name") whoami_parser.add_argument("remote", help="label of remote host") whoami_parser.set_defaults(func=cls.func) @classmethod def func(cls, args: Namespace) -> None: remote = str(args.remote) cls.whoami(remote) @classmethod def whoami(cls, remote: str) -> None: from mdrsclient.client import MdrsClient remote_host = MdrsClient.parse_remote_host(remote) client = MdrsClient.from_remote(remote_host) if client.connection.token is not None and client.connection.token.is_expired: client.logout() try: user = client.whoami() username = user.username if user is not None else cls.ANONYMOUS_USERNAME except Exception: username = cls.ANONYMOUS_USERNAME print(username)