split source code for each command.

mdrsclient/commands/rm.py

@@ -0,0 +1,39 @@
+import os
+from argparse import Namespace, _SubParsersAction
+
+from mdrsclient.api import FileApi, FolderApi
+from mdrsclient.commands.base import BaseCommand
+from mdrsclient.exceptions import IllegalArgumentException
+
+
+class RmCommand(BaseCommand):
+    @classmethod
+    def register(cls, parsers: _SubParsersAction) -> None:
+        command = cls()
+        rm_parser = parsers.add_parser("rm", help="remove the file or folder")
+        rm_parser.add_argument(
+            "-r", "--recursive", help="remove folders and their contents recursive", action="store_true"
+        )
+        rm_parser.add_argument("remote_path", help="remote file path (remote:/lab/path/file)")
+        rm_parser.set_defaults(func=command.rm)
+
+    def rm(self, args: Namespace) -> None:
+        (remote, laboratory_name, r_path) = self._parse_remote_host_with_path(args.remote_path)
+        r_path = r_path.rstrip("/")
+        r_dirname = os.path.dirname(r_path)
+        r_basename = os.path.basename(r_path)
+        connection = self._create_connection(remote)
+        laboratory = self._find_laboratory(connection, laboratory_name)
+        parent_folder = self._find_folder(connection, laboratory, r_dirname)
+        file = parent_folder.find_file(r_basename)
+        if file is not None:
+            file_api = FileApi(connection)
+            file_api.destroy(file)
+        else:
+            folder = parent_folder.find_sub_folder(r_basename)
+            if folder is None:
+                raise IllegalArgumentException(f"Cannot remove `{r_path}`: No such file or folder.")
+            if not args.recursive:
+                raise IllegalArgumentException(f"Cannot remove `{r_path}`: Is a folder.")
+            folder_api = FolderApi(connection)
+            folder_api.destroy(folder.id)