specify get_file for RsConnectFs

rstudio · Dec 12, 2023 · 0ea84ad · 0ea84ad
1 parent bd7055d
commit 0ea84ad
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/pins/rsconnect/fs.py b/pins/rsconnect/fs.py
@@ -2,6 +2,7 @@
 from pathlib import Path
 
 from fsspec import AbstractFileSystem
+from fsspec.utils import isfilelike
 
 from typing import Sequence
 
@@ -277,6 +278,14 @@ def get(self, rpath, lpath, recursive=False, *args, **kwargs) -> None:
                 bundle["content_guid"], bundle["id"], parsed.file_name, lpath
             )
 
+    def get_file(self, rpath, lpath, **kwargs):
+        data = self.cat_file(rpath, **kwargs)
+        if isfilelike(lpath):
+            lpath.write(data)
+        else:
+            with open(lpath, "wb") as f:
+                f.write(data)
+
     def exists(self, path: str, **kwargs) -> bool:
         try:
             self.info(path)