truefoundry 0.4.4rc11__tar.gz → 0.4.4rc12__tar.gz

{truefoundry-0.4.4rc11 → truefoundry-0.4.4rc12}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: truefoundry
-Version: 0.4.4rc11
+Version: 0.4.4rc12
 Summary: Truefoundry CLI
 Author: Abhishek Choudhary
 Author-email: abhishek@truefoundry.com

{truefoundry-0.4.4rc11 → truefoundry-0.4.4rc12}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "truefoundry"
-version = "0.4.4rc11"
+version = "0.4.4rc12"
 description = "Truefoundry CLI"
 authors = ["Abhishek Choudhary <abhishek@truefoundry.com>"]
 readme = "README.md"

{truefoundry-0.4.4rc11 → truefoundry-0.4.4rc12}/truefoundry/common/tfy_signed_url_fs.py

@@ -8,7 +8,7 @@ from typing import Optional
 from fsspec.spec import DEFAULT_CALLBACK, AbstractBufferedFile, AbstractFileSystem
 
 from truefoundry.common.constants import ENV_VARS
-from truefoundry.common.tfy_signed_url_client import SignedURLClient
+from truefoundry.common.tfy_signed_url_client import LOG_PREFIX, SignedURLClient
 from truefoundry.common.utils import log_time
 
 
@@ -21,12 +21,12 @@ class SignedURLFileSystem(AbstractFileSystem):
         token = token or ENV_VARS.TFY_INTERNAL_SIGNED_URL_SERVER_TOKEN
         self.client = SignedURLClient(base_url, token)
 
-    @log_time
+    @log_time(prefix=LOG_PREFIX)
     def exists(self, path, **kwargs):
         """Check if a file exists at the given path."""
         return self.client.exists(path)
 
-    @log_time
+    @log_time(prefix=LOG_PREFIX)
     def get(
         self,
         rpath,
@@ -87,7 +87,7 @@ class SignedURLFileSystem(AbstractFileSystem):
             Path(os.path.dirname(target_local_path)).mkdir(parents=True, exist_ok=True)
             self.client.download(storage_uri=rpath, local_path=target_local_path)
 
-    @log_time
+    @log_time(prefix=LOG_PREFIX)
     def put(
         self,
         lpath,
@@ -134,7 +134,7 @@ class SignedURLFileSystem(AbstractFileSystem):
                 rpath = os.path.join(rpath, local_path.name)
             return self.client.upload(file_path=lpath, storage_uri=rpath)
 
-    @log_time
+    @log_time(prefix=LOG_PREFIX)
     def isdir(self, path):
         """Is this entry directory-like?"""
         return self.client.is_directory(path)
@@ -170,6 +170,7 @@ class SignedURLFileSystem(AbstractFileSystem):
             # Wrapping the buffer to automatically upload on close
             return io.BufferedWriter(buffer, on_close)
 
+    @log_time(prefix=LOG_PREFIX)
     def write(self, path, data, **kwargs):
         """
         Write data to the file at the specified path (this could be tied to open's close).
@@ -185,7 +186,7 @@ class SignedURLFileSystem(AbstractFileSystem):
         # Upload the content to the remote file system
         self.client.upload_from_bytes(content, storage_uri=path)
 
-    @log_time
+    @log_time(prefix=LOG_PREFIX)
     def ls(self, path, detail=True, **kwargs):
         """List objects at path."""
         return self.client.list_files(path, detail=detail)

{truefoundry-0.4.4rc11 → truefoundry-0.4.4rc12}/truefoundry/common/utils.py

@@ -105,15 +105,18 @@ def resolve_tfy_host(tfy_host: Optional[str] = None) -> str:
     return tfy_host
 
 
-def log_time(func, prefix: str = ""):
+def log_time(prefix: str = ""):
     """Decorator to log the time taken by I/O operations."""
 
-    @wraps(func)
-    def wrapper(*args, **kwargs):
-        start_time = time.time()
-        result = func(*args, **kwargs)
-        elapsed_time = time.time() - start_time
-        logger.info(f"{prefix} {func.__name__} took {elapsed_time:.2f} seconds")
-        return result
+    def decorator(func):
+        @wraps(func)
+        def wrapper(*args, **kwargs):
+            start_time = time.time()
+            result = func(*args, **kwargs)
+            elapsed_time = time.time() - start_time
+            logger.info(f"{prefix}{func.__name__} took {elapsed_time:.2f} seconds")
+            return result
+
+        return wrapper
 
-    return wrapper
+    return decorator