remote_file.py

python/kvikio/kvikio/__init__.py

@@ -3,7 +3,8 @@
 
 from kvikio._lib import buffer, driver_properties  # type: ignore
 from kvikio._version import __git_commit__, __version__  # noqa: F401
-from kvikio.cufile import CuFile, RemoteFile  # noqa: F401
+from kvikio.cufile import CuFile  # noqa: F401
+from kvikio.remote_file import RemoteFile  # noqa: F401
 
 
 def memory_register(buf) -> None:

python/kvikio/kvikio/remote_file.py

@@ -0,0 +1,41 @@
+# Copyright (c) 2024, NVIDIA CORPORATION. All rights reserved.
+# See file LICENSE for terms.
+
+from __future__ import annotations
+
+from typing import Optional
+
+from typing_extensions import Self
+
+from kvikio._lib import remote_handle  # type: ignore
+from kvikio.cufile import IOFuture
+
+
+class RemoteFile:
+    """File handle of a remote file"""
+
+    def __init__(self, bucket_name: str, object_name: str):
+        self._handle = remote_handle.RemoteFile.from_bucket_and_object(
+            bucket_name, object_name
+        )
+
+    @classmethod
+    def from_url(cls, url: str) -> Self:
+        ret = object.__new__(cls)
+        ret._handle = remote_handle.RemoteFile.from_url(url)
+        return ret
+
+    def __enter__(self) -> RemoteFile:
+        return self
+
+    def __exit__(self, exc_type, exc_val, exc_tb) -> None:
+        pass
+
+    def nbytes(self) -> int:
+        return self._handle.nbytes()
+
+    def pread(self, buf, size: Optional[int] = None, file_offset: int = 0) -> IOFuture:
+        return IOFuture(self._handle.pread(buf, size, file_offset))
+
+    def read(self, buf, size: Optional[int] = None, file_offset: int = 0) -> int:
+        return self.pread(buf, size, file_offset).get()