Cache object information when checking if a file exists.

src/databricks/labs/blueprint/paths.py

@@ -592,7 +592,7 @@ def exists(self, *, follow_symlinks: bool = True) -> bool:
         if not follow_symlinks:
             raise NotImplementedError("follow_symlinks=False is not supported for DBFS")
         try:
-            self._ws.dbfs.get_status(self.as_posix())
+            self._cached_file_info = self._ws.dbfs.get_status(self.as_posix())
             return True
         except NotFound:
             return False
@@ -741,7 +741,7 @@ def exists(self, *, follow_symlinks: bool = True) -> bool:
         if not follow_symlinks:
             raise NotImplementedError("follow_symlinks=False is not supported for Databricks Workspace")
         try:
-            self._ws.workspace.get_status(self.as_posix())
+            self._cached_object_info = self._ws.workspace.get_status(self.as_posix())
             return True
         except NotFound:
             return False

tests/unit/test_paths.py

@@ -424,10 +424,21 @@ def test_iterdir() -> None:
 def test_exists_when_path_exists() -> None:
     ws = create_autospec(WorkspaceClient)
     workspace_path = WorkspacePath(ws, "/test/path")
-    ws.workspace.get_status.return_value = True
+    ws.workspace.get_status.return_value = ObjectInfo(path="/test/path")
     assert workspace_path.exists()
 
 
+def test_exists_caches_info() -> None:
+    ws = create_autospec(WorkspaceClient)
+    workspace_path = WorkspacePath(ws, "/test/path")
+    ws.workspace.get_status.return_value = ObjectInfo(path="/test/path", object_type=ObjectType.FILE)
+    _ = workspace_path.exists()
+
+    ws.workspace.get_status.reset_mock()
+    _ = workspace_path.is_file()
+    assert not ws.workspace.get_status.called
+
+
 def test_exists_when_path_does_not_exist() -> None:
     ws = create_autospec(WorkspaceClient)
     workspace_path = WorkspacePath(ws, "/test/path")