element.py: Add _get_artifact() method

Add a getter for Element's __artifact (Artifact object).

This is required by ArtifactElement. The ArtifactElement should
have access to its Artifact.

1 files changed, 11 insertions, 0 deletions

diff --git a/src/buildstream/element.py b/src/buildstream/element.py
index bd5ca14e7..78fee5dff 100644
--- a/src/buildstream/element.py
+++ b/src/buildstream/element.py
@@ -2355,6 +2355,17 @@ class Element(Plugin):
         casbd = self.__artifact.get_files()
         return [f for f in casbd.list_relative_paths()]
 
+    # _get_artifact()
+    #
+    # Return the Element's Artifact object
+    #
+    # Returns:
+    #    (Artifact): The Artifact object of the Element
+    #
+    def _get_artifact(self):
+        assert self.__artifact, "{}: has no Artifact object".format(self.name)
+        return self.__artifact
+
     #############################################################
     #                   Private Local Methods                   #
     #############################################################