temp

Author: nunogoncalves03

singlestoredb/management/files.py

@@ -847,9 +847,12 @@ def is_dir(self, path: PathLike) -> bool:
         bool
 
         """
-        raise ValueError(
-            'Operation not supported: directories are currently not allowed in Files API',
-        )
+        try:
+            return self.info(path).type == 'directory'
+        except ManagementError as exc:
+            if exc.errno == 404:
+                return False
+            raise
 
     # TODO: remove from FileLocation?
     def is_file(self, path: PathLike) -> bool:
@@ -873,6 +876,19 @@ def is_file(self, path: PathLike) -> bool:
                 return False
             raise
 
+    def _list_root_dir(self) -> List[str]:
+        """
+        Return the names of files in the root directory.
+
+        Parameters
+        ----------
+
+        """
+        res = self._manager._get(
+            f'files/fs/{self._location}',
+        ).json()
+        return [x['path'] for x in res['content'] or []]
+
     # TODO: remove from FileLocation?
     def listdir(
         self,
@@ -893,6 +909,9 @@ def listdir(
         List[str]
 
         """
+        if path == '/':
+            return self._list_root_dir()
+
         raise ValueError(
             'Operation not supported: directories are currently not allowed in Files API',
         )