cti-python-stix2/stix2/environment.py

18 lines
508 B
Python
Raw Normal View History

class ObjectFactory(object):
2017-07-12 20:44:52 +02:00
def __init__(self, created_by=None, markings=None):
self.created_by = created_by
2017-07-12 20:44:52 +02:00
self.markings = markings
def create(self, cls=None, **kwargs):
if cls is None:
raise ValueError('No STIX object class provided')
if self.created_by is not None:
kwargs['created_by_ref'] = self.created_by
2017-07-12 20:44:52 +02:00
if self.markings is not None:
kwargs['object_marking_refs'] = self.markings
return cls(**kwargs)