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

Alternate implementation of ObjectFactory.

stix2.1
Greg Back 2017-07-13 15:03:31 +00:00
parent 4dba41afc8
commit c29acfa3ce
1 changed files with 21 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
stix2/environment.py
View File

@ -4,24 +4,27 @@ class ObjectFactory(object):
def __init__(self, created_by_ref=None, created=None,
external_references=None, object_marking_refs=None,
granular_markings=None):
self.created_by_ref = created_by_ref
self.created = created
self.external_references = external_references
self.object_marking_refs = object_marking_refs
self.granular_markings = granular_markings
self._defaults = {}
if created_by_ref:
self._defaults['created_by_ref'] = created_by_ref
if created:
self._defaults['created'] = created
# If the user provides a default "created" time, we also want to use
# that as the modified time.
self._defaults['modified'] = created
if external_references:
self._defaults['external_references'] = external_references
if object_marking_refs:
self._defaults['object_marking_refs'] = object_marking_refs
if granular_markings:
self._defaults['granular_markings'] = granular_markings
def create(self, cls, **kwargs):
if 'created_by_ref' not in kwargs and self.created_by_ref is not None:
kwargs['created_by_ref'] = self.created_by_ref
if 'created' not in kwargs and self.created is not None:
kwargs['created'] = self.created
if 'modified' not in kwargs:
kwargs['modified'] = self.created
if 'external_references' not in kwargs and self.external_references is not None:
kwargs['external_references'] = self.external_references
if 'object_marking_refs' not in kwargs and self.object_marking_refs is not None:
kwargs['object_marking_refs'] = self.object_marking_refs
if 'granular_markings' not in kwargs and self.granular_markings is not None:
kwargs['granular_markings'] = self.granular_markings
# Use self.defaults as the base, but update with any explicit args
# provided by the user.
properties = dict(**self._defaults)
if kwargs:
properties.update(**kwargs)
return cls(**kwargs)
return cls(**properties)