storage: Implement __iter__ in Directory classes

2 files changed, 6 insertions, 0 deletions

diff --git a/src/buildstream/storage/_casbaseddirectory.py b/src/buildstream/storage/_casbaseddirectory.py
index 6fe200de1..99cf1113f 100644
--- a/src/buildstream/storage/_casbaseddirectory.py
+++ b/src/buildstream/storage/_casbaseddirectory.py
@@ -843,6 +843,9 @@ class CasBasedDirectory(Directory):
         except VirtualDirectoryError:
             return False
 
+    def __iter__(self):
+        yield from self.index.keys()
+
     def _set_subtree_read_only(self, read_only):
         self.__node_properties = list(filter(lambda prop: prop.name != "SubtreeReadOnly", self.__node_properties))
         node_property = remote_execution_pb2.NodeProperty()
diff --git a/src/buildstream/storage/_filebaseddirectory.py b/src/buildstream/storage/_filebaseddirectory.py
index f9a830590..1799a84b8 100644
--- a/src/buildstream/storage/_filebaseddirectory.py
+++ b/src/buildstream/storage/_filebaseddirectory.py
@@ -277,6 +277,9 @@ class FileBasedDirectory(Directory):
 
             return utils.save_file_atomic(newpath, mode=mode, encoding=encoding)
 
+    def __iter__(self):
+        yield from os.listdir(self.external_directory)
+
     def __str__(self):
         # This returns the whole path (since we don't know where the directory started)
         # which exposes the sandbox directory; we will have to assume for the time being