feat: added path property for gcs path obj

src/ot_orchestration/utils/path.py

@@ -6,6 +6,7 @@
 import multiprocessing
 import re
 from abc import abstractmethod
+from dataclasses import dataclass
 from functools import cached_property
 from pathlib import Path
 from typing import Any, Protocol, TypedDict
@@ -37,6 +38,7 @@ class PathSegments(TypedDict):
 
 class ProtoPath(Protocol):
     segments: PathSegments
+    path: str
 
     @abstractmethod
     def dump(self, data: Any) -> None:
@@ -127,6 +129,11 @@ def exists(self) -> bool:
         """
         return self.native_path.exists()
 
+    @cached_property
+    def path(self) -> str:
+        """Get path segment."""
+        return str(self.native_path)
+
     @cached_property
     def segments(self) -> PathSegments:
         """Get path segments.
@@ -176,7 +183,7 @@ def client(self) -> storage.Client:
         adapter = HTTPAdapter(pool_connections=128, pool_maxsize=1024, max_retries=3)
         client._http.mount("https://", adapter)
         client._http._auth_request.session.mount("https://", adapter)
-        return
+        return client
 
     @cached_property
     def _match(self) -> re.Match:

tests/test_io_manager.py

@@ -188,3 +188,18 @@ def test_bucket_property(self, gcs_path: str, bucket: str) -> None:
         """Test GCSPath object bucket property."""
         gcs_path_obj = GCSPath(gcs_path)
         assert gcs_path_obj.bucket == bucket
+
+    @pytest.mark.parametrize(
+        ["gcs_path", "path"],
+        [
+            pytest.param(
+                "gs://bucket/prefix/filename",
+                "prefix/filename",
+                id="GCS path with bucket",
+            ),
+        ],
+    )
+    def test_path_property(self, gcs_path: str, path: str) -> None:
+        """Test GCSPath object bucket property."""
+        gcs_path_obj = GCSPath(gcs_path)
+        assert gcs_path_obj.path == path