Implement file uploading for Storage Client

iterative · Dec 29, 2024 · ec29c5d · ec29c5d
1 parent cf05881
commit ec29c5d
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/datachain/client/fsspec.py b/src/datachain/client/fsspec.py
@@ -364,6 +364,12 @@ def open_object(
         assert not file.location
         return FileWrapper(self.fs.open(self.get_full_path(file.path)), cb)  # type: ignore[return-value]
 
+    def upload(self, path: str, data: bytes) -> "File":
+        full_path = self.get_full_path(path)
+        self.fs.pipe_file(full_path, data)
+        file_info = self.fs.info(full_path)
+        return self.info_to_file(file_info, path)
+
     def download(self, file: "File", *, callback: Callback = DEFAULT_CALLBACK) -> None:
         sync(get_loop(), functools.partial(self._download, file, callback=callback))