File upload (WIP)

iterative · Dec 28, 2024 · efe4850 · efe4850
1 parent cf05881
commit efe4850
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/src/datachain/client/fsspec.py b/src/datachain/client/fsspec.py
@@ -364,6 +364,18 @@ 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.write_bytes(full_path, data)
+        file_info = self.fs.info(full_path)
+        file = self.info_to_file(file_info, path)
+        self.put_in_cache(file)
+        return file
+
     def download(self, file: "File", *, callback: Callback = DEFAULT_CALLBACK) -> None:
         sync(get_loop(), functools.partial(self._download, file, callback=callback))