_yaml: Move '__contains__' to MappingNode

This method is only ever used in this context and should not be on the
base Node anyways

1 files changed, 3 insertions, 6 deletions

diff --git a/src/buildstream/_yaml.pyx b/src/buildstream/_yaml.pyx
index db080289b..e2593d95e 100644
--- a/src/buildstream/_yaml.pyx
+++ b/src/buildstream/_yaml.pyx
@@ -85,12 +85,6 @@ cdef class Node:
     cdef bint _shares_position_with(self, Node target):
         return self.file_index == target.file_index and self.line == target.line and self.column == target.column
 
-    def __contains__(self, what):
-        # Delegate to the inner value, though this will likely not work
-        # very well if the node is a list or string, it's unlikely that
-        # code which has access to such nodes would do this.
-        return what in (<MappingNode> self).value
-
     cpdef Node copy(self):
         raise NotImplementedError()
 
@@ -225,6 +219,9 @@ cdef class MappingNode(Node):
     def __cinit__(self, int file_index, int line, int column, dict value):
         self.value = value
 
+    def __contains__(self, what):
+        return what in self.value
+
     cpdef MappingNode copy(self):
         cdef dict copy = {}
         cdef str key