_artifactelement.py: Add get_dependency_refs() method

1 files changed, 15 insertions, 0 deletions

diff --git a/src/buildstream/_artifactelement.py b/src/buildstream/_artifactelement.py
index d65d46173..35ef89574 100644
--- a/src/buildstream/_artifactelement.py
+++ b/src/buildstream/_artifactelement.py
@@ -20,6 +20,7 @@ from . import Element
 from . import _cachekey
 from ._exceptions import ArtifactElementError
 from ._loader.metaelement import MetaElement
+from .types import Scope
 
 
 # ArtifactElement()
@@ -55,6 +56,20 @@ class ArtifactElement(Element):
     def preflight(self):
         pass
 
+    # get_dependency_refs()
+    #
+    # Obtain the refs of a particular scope of dependencies
+    #
+    # Args:
+    #   scope (Scope): The scope of dependencies for which we want to obtain the refs
+    #
+    # Returns:
+    #   (list [str]): A list of artifact refs
+    #
+    def get_dependency_refs(self, scope=Scope.BUILD):
+        artifact = self._get_artifact()
+        return artifact.get_dependency_refs(deps=scope)
+
     # Override Element._calculate_cache_key
     def _calculate_cache_key(self, dependencies=None):
         return self._key