split source code for each command.

mdrsclient/commands/labs.py

@@ -0,0 +1,40 @@
+from argparse import Namespace, _SubParsersAction
+
+from mdrsclient.api import LaboratoryApi
+from mdrsclient.commands.base import BaseCommand
+
+
+class LabsCommand(BaseCommand):
+    @classmethod
+    def register(cls, parsers: _SubParsersAction) -> None:
+        command = cls()
+        labs_parser = parsers.add_parser("labs", help="list all laboratories")
+        labs_parser.add_argument("remote", help="label of remote host")
+        labs_parser.set_defaults(func=command.labs)
+
+    def labs(self, args: Namespace) -> None:
+        remote = self._parse_remote_host(args.remote)
+        connection = self._create_connection(remote)
+        laboratory_api = LaboratoryApi(connection)
+        laboratories = laboratory_api.list()
+        connection.laboratories = laboratories
+        label = {"id": "ID", "name": "Name", "pi_name": "PI", "full_name": "Laboratory"}
+        length: dict[str, int] = {}
+        for key in label.keys():
+            length[key] = len(label[key])
+        for laboratory in laboratories:
+            length["id"] = max(length["id"], len(str(laboratory.id)))
+            length["name"] = max(length["name"], len(laboratory.name))
+            length["pi_name"] = max(length["pi_name"], len(laboratory.pi_name))
+            length["full_name"] = max(length["full_name"], len(laboratory.full_name))
+        header = (
+            f"{label['id']:{length['id']}}\t{label['name']:{length['name']}}\t"
+            f"{label['pi_name']:{length['pi_name']}}\t{label['full_name']:{length['full_name']}}"
+        )
+        print(header)
+        print("-" * len(header.expandtabs()))
+        for laboratory in laboratories:
+            print(
+                f"{laboratory.id:{length['id']}}\t{laboratory.name:{length['name']}}\t"
+                f"{laboratory.pi_name:{length['pi_name']}}\t{laboratory.full_name:{length['full_name']}}"
+            )