split source code for each command.

mdrsclient/commands/whoami.py

@@ -0,0 +1,29 @@
+from argparse import Namespace, _SubParsersAction
+from typing import 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: _SubParsersAction) -> None:
+        command = cls()
+        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=command.whoami)
+
+    def whoami(self, args: Namespace) -> None:
+        remote = self._parse_remote_host(args.remote)
+        config = ConfigFile(remote)
+        if config.url is None:
+            raise MissingConfigurationException(f"Remote host `{remote}` is not found.")
+        connection = MDRSConnection(config.remote, config.url)
+        if connection.token is not None and connection.token.is_expired:
+            connection.logout()
+        username = connection.user.username if connection.user is not None else self.ANONYMOUS_USERNAME
+        print(username)