fix: Diffusing interoperability parameter to all included objects & r…

…eferences
MISP · Apr 15, 2020 · e4f0855 · e4f0855
1 parent 8e95dbf
commit e4f0855
Showing 1 changed file with 17 additions and 2 deletions.
diff --git a/stix2/base.py b/stix2/base.py
@@ -324,11 +324,26 @@ def sort_by(element):
 
 
 class _DomainObject(_STIXBase, _MarkingsMixin):
-    pass
+    def __init__(self, *args, **kwargs):
+        interoperability = kwargs.get('interoperability', False)
+        self.__interoperability = interoperability
+        self._properties['id'].interoperability = interoperability
+        self._properties['created_by_ref'].interoperability = interoperability
+        if kwargs.get('object_marking_refs'):
+            self._properties['object_marking_refs'].contained.interoperability = interoperability
+        super(_DomainObject, self).__init__(*args, **kwargs)
 
 
 class _RelationshipObject(_STIXBase, _MarkingsMixin):
-    pass
+    def __init__(self, *args, **kwargs):
+        interoperability = kwargs.get('interoperability', False)
+        self.__interoperability = interoperability
+        self._properties['id'].interoperability = interoperability
+        if kwargs.get('created_by_ref'):
+            self._properties['created_by_ref'].interoperability = interoperability
+        if kwargs.get('object_marking_refs'):
+            self._properties['object_marking_refs'].contained.interoperability = interoperability
+        super(_RelationshipObject, self).__init__(*args, **kwargs)
 
 
 class _Observable(_STIXBase):