truefoundry 0.4.4rc11__py3-none-any.whl → 0.4.4rc12__py3-none-any.whl

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/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

{truefoundry-0.4.4rc11.dist-info → truefoundry-0.4.4rc12.dist-info}/METADATA

@@ -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.dist-info → truefoundry-0.4.4rc12.dist-info}/RECORD

@@ -35,8 +35,8 @@ truefoundry/common/exceptions.py,sha256=ePpiQ_zmWe4e94gOgeMiyP_AZnKwjEBfyXsB5ScG
 truefoundry/common/request_utils.py,sha256=5xw4YGUcMf71Ncal3OfFCa-PoWDIvG3hYGCDa4Da4OI,2854
 truefoundry/common/servicefoundry_client.py,sha256=2fxmgCM-ckFHpnm6n_mL-5Z8RWN_q-dYVvFC29bkYSg,3120
 truefoundry/common/tfy_signed_url_client.py,sha256=VklNm4K4xm1CcKZ1iiT8281X6jo803AhgYdtVvvwDXY,10183
-truefoundry/common/tfy_signed_url_fs.py,sha256=Vc7Il077nZIGbDBQ8aTPNtJVcypYaoZHr13vV00REko,8486
-truefoundry/common/utils.py,sha256=XM3QsbR_kCEjyNBRpM8oRhV3kvz0xRFfAiqYG4dC1sU,3692
+truefoundry/common/tfy_signed_url_fs.py,sha256=xwUR2Vx892sVuiBN5wUNagWuGxGRMecvYsuKQ5yy1rg,8626
+truefoundry/common/utils.py,sha256=RS_YExw9PHfoxKr0BZhPQAsDBKssqL1PgdW6a6rBqu0,3764
 truefoundry/deploy/__init__.py,sha256=ugawKF2G02EmEXX35oZ2tec12d9oWN28Sf6mtGGIERY,2281
 truefoundry/deploy/auto_gen/models.py,sha256=4MaxkG2_5Wg6avaZRlK0D4JiVEM5rk3NU0BCiTx8VyU,82477
 truefoundry/deploy/builder/__init__.py,sha256=1qjHMNBE1poRCZW0WrG46dFM1f1IlivD5352qzsioMU,4953
@@ -353,7 +353,7 @@ truefoundry/workflow/map_task.py,sha256=2m3qGXQ90k9LdS45q8dqCCECc3qr8t2m_LMCVd1m
 truefoundry/workflow/python_task.py,sha256=SRXRLC4vdBqGjhkwuaY39LEWN6iPCpJAuW17URRdWTY,1128
 truefoundry/workflow/task.py,sha256=ToitYiKcNzFCtOVQwz1W8sRjbR97eVS7vQBdbgUQtKg,1779
 truefoundry/workflow/workflow.py,sha256=WaTqUjhwfAXDWu4E5ehuwAxrCbDJkoAf1oWmR2E9Qy0,4575
-truefoundry-0.4.4rc11.dist-info/METADATA,sha256=47vgw6KFJIUdkhXwbyhuVFktZ4MxdJJR9_pBuHUakGw,3102
-truefoundry-0.4.4rc11.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-truefoundry-0.4.4rc11.dist-info/entry_points.txt,sha256=TXvUxQkI6zmqJuycPsyxEIMr3oqfDjgrWj0m_9X12x4,95
-truefoundry-0.4.4rc11.dist-info/RECORD,,
+truefoundry-0.4.4rc12.dist-info/METADATA,sha256=t-97uMhttorL6bHloEkl7d4ewbPEPE5lHRz3nmCywy4,3102
+truefoundry-0.4.4rc12.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+truefoundry-0.4.4rc12.dist-info/entry_points.txt,sha256=TXvUxQkI6zmqJuycPsyxEIMr3oqfDjgrWj0m_9X12x4,95
+truefoundry-0.4.4rc12.dist-info/RECORD,,