_artifactcache.py: Add get_artifact_directory() method

1 files changed, 21 insertions, 0 deletions

diff --git a/buildstream/_artifactcache.py b/buildstream/_artifactcache.py
index b73304fac..e29379dad 100644
--- a/buildstream/_artifactcache.py
+++ b/buildstream/_artifactcache.py
@@ -444,6 +444,27 @@ class ArtifactCache():
 
         return self.cas.remove(ref)
 
+    # get_artifact_directory():
+    #
+    # Get virtual directory for cached artifact of the specified Element.
+    #
+    # Assumes artifact has previously been fetched or committed.
+    #
+    # Args:
+    #     element (Element): The Element to extract
+    #     key (str): The cache key to use
+    #
+    # Raises:
+    #     ArtifactError: In cases there was an OSError, or if the artifact
+    #                    did not exist.
+    #
+    # Returns: virtual directory object
+    #
+    def get_artifact_directory(self, element, key):
+        ref = element.get_artifact_name(key)
+        digest = self.cas.resolve_ref(ref, update_mtime=True)
+        return CasBasedDirectory(self.cas, digest)
+
     # extract():
     #
     # Extract cached artifact for the specified Element if it hasn't