MISP
/
cti-python-stix2
mirror of https://github.com/MISP/cti-python-stix2
2
0
Fork 0
Code Issues Releases Wiki Activity

add checks for MarkingDefinition requirements

pull/1/head
Emmanuelle Vargas-Gonzalez 2020-12-23 16:15:31 -05:00
parent 6a6f13c37b
commit a86b21dbf7
1 changed files with 18 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
stix2/v21/common.py
View File

@ -3,7 +3,7 @@
from collections import OrderedDict from collections import OrderedDict
from ..custom import _custom_marking_builder from ..custom import _custom_marking_builder
from ..exceptions import InvalidValueError from ..exceptions import InvalidValueError, PropertyPresenceError
from ..markings import _MarkingsMixin from ..markings import _MarkingsMixin
from ..markings.utils import check_tlp_marking from ..markings.utils import check_tlp_marking
from ..properties import ( from ..properties import (
@ -222,6 +222,23 @@ class MarkingDefinition(_STIXBase21, _MarkingsMixin):
def _check_object_constraints(self): def _check_object_constraints(self):
super(MarkingDefinition, self)._check_object_constraints() super(MarkingDefinition, self)._check_object_constraints()
definition = self.get("definition")
definition_type = self.get("definition_type")
extensions = self.get("extensions")
if not extensions and not definition:
raise PropertyPresenceError(
"MarkingDefinition objects must have the property 'definition' "
"if 'extensions' is not present",
MarkingDefinition,
)
if not extensions and not definition_type:
raise PropertyPresenceError(
"MarkingDefinition objects must have the property 'definition_type' "
"if 'extensions' is not present",
MarkingDefinition,
)
check_tlp_marking(self, '2.1') check_tlp_marking(self, '2.1')
def serialize(self, pretty=False, include_optional_defaults=False, **kwargs): def serialize(self, pretty=False, include_optional_defaults=False, **kwargs):