add folder methods to rh.cluster

runhouse/resources/hardware/cluster.py

@@ -1783,3 +1783,49 @@ def _enable_or_update_status_check(
             self.call_client_method(
                 "set_settings", {"status_check_interval": new_interval}
             )
+
+    ##############################################
+    # Folder Operations
+    ##############################################
+    def _ls(self, path: Path, recursive: bool = False):
+        return self.client.folder_operation(
+            operation="ls", path=path, recursive=recursive
+        )
+
+    def _get(
+        self,
+        path: Path,
+        mode: str = "rb",
+        serialization: str = None,
+    ):
+        return self.client.folder_operation(
+            operation="get",
+            path=path,
+            mode=mode,
+            serialization=serialization,
+        )
+
+    def _put(
+        self,
+        path: Path,
+        contents: Union[Dict[str, Any], Resource, List[Resource]],
+        mode: str = "wb",
+        overwrite: bool = False,
+        serialization: str = None,
+    ):
+        return self.client.folder_operation(
+            operation="put",
+            path=path,
+            contents=contents,
+            mode=mode,
+            overwrite=overwrite,
+            serialization=serialization,
+        )
+
+    def _rm(self, path: Path, contents: List[str] = None, recursive: bool = False):
+        return self.client.folder_operation(
+            operation="rm", path=path, contents=contents, recursive=recursive
+        )
+
+    def _mkdir(self, path: Path):
+        return self.client.folder_operation(operation="mkdir", path=path)

runhouse/servers/http/http_client.py

@@ -21,6 +21,8 @@
 from runhouse.servers.http.http_utils import (
     CallParams,
     DeleteObjectParams,
+    FolderMethod,
+    FolderParams,
     GetObjectParams,
     handle_response,
     OutputType,
@@ -272,6 +274,10 @@ def status(self, resource_address: str):
         # Note: Resource address must be specified in order to construct the cluster subtoken
         return self.request("status", req_type="get", resource_address=resource_address)
 
+    def folder_operation(self, operation: FolderMethod, **kwargs):
+        folder_params = FolderParams(operation=operation, **kwargs).dict()
+        return self.request_json("/folder", req_type="post", json_dict=folder_params)
+
     def get_certificate(self):
         cert: bytes = self.request(
             "cert",