class _BaseTestMechanism(stix.Entity):
_namespace = "http://stix.mitre.org/Indicator-2"
_binding = indicator_binding
_binding_class = indicator_binding.TestMechanismType()
id_ = fields.IdField("id")
idref = fields.IdField("idref")
efficacy = fields.TypedField("Efficacy", Statement)
producer = fields.TypedField("Producer", InformationSource)
def __init__(self, id_=None, idref=None):
super(_BaseTestMechanism, self).__init__()
self.id_ = id_
self.idref = idref
self.efficacy = None
self.producer = None
def to_obj(self, ns_info=None):
obj = super(_BaseTestMechanism, self).to_obj(ns_info=ns_info)
obj.xsi_type = self._XSI_TYPE
return obj
def to_dict(self):
d = super(_BaseTestMechanism, self).to_dict()
d['xsi:type'] = self._XSI_TYPE # added by subclass
return d
class _BaseTestMechanism(stix.Entity):
_namespace = "http://stix.mitre.org/Indicator-2"
_binding = indicator_binding
_binding_class = indicator_binding.TestMechanismType()
def __init__(self, id_=None, idref=None):
self.id_ = id_ or stix.utils.create_id("testmechanism")
self.idref = idref
self.efficacy = None
self.producer = None
@property
def efficacy(self):
return self._efficacy
@efficacy.setter
def efficacy(self, value):
if not value:
self._efficacy = None
elif isinstance(value, Statement):
self._efficacy = value
else:
self._efficacy = Statement(value=value)
@property
def producer(self):
return self._producer
@producer.setter
def producer(self, value):
if not value:
self._producer = None
elif isinstance(value, InformationSource):
self._producer = value
else:
raise ValueError('Cannot set producer to type %s' % type(value))
@classmethod
def from_obj(cls, obj, return_obj=None):
if not obj:
return None
from stix.extensions.test_mechanism.snort_test_mechanism import SnortTestMechanism
from stix.extensions.test_mechanism.open_ioc_2010_test_mechanism import OpenIOCTestMechanism
if not return_obj:
klass = _BaseTestMechanism.lookup_class(obj.xml_type)
return_obj = klass.from_obj(obj)
else:
return_obj.id_ = obj.get_id()
return_obj.idref = obj.get_idref()
return_obj.efficacy = Statement.from_obj(obj.get_Efficacy())
return_obj.producer = InformationSource.from_obj(
obj.get_Producer())
return return_obj
def to_obj(self, return_obj=None):
if not return_obj:
return_obj = self._binding_class()
return_obj.set_id(self.id_)
return_obj.set_idref(self.idref)
if self.efficacy:
return_obj.set_Efficacy(self.efficacy.to_obj())
if self.producer:
return_obj.set_Producer(self.producer.to_obj())
return return_obj
@staticmethod
def lookup_class(xsi_type):
if not xsi_type:
raise ValueError("xsi:type is required")
for (k, v) in _EXTENSION_MAP.iteritems():
# TODO: for now we ignore the prefix and just check for
# a partial match
if xsi_type in k:
return v
raise ValueError("Unregistered xsi:type %s" % xsi_type)
@classmethod
def from_dict(cls, d, return_obj=None):
if not d:
return None
from stix.extensions.test_mechanism.snort_test_mechanism import SnortTestMechanism
from stix.extensions.test_mechanism.open_ioc_2010_test_mechanism import OpenIOCTestMechanism
if not return_obj:
klass = _BaseTestMechanism.lookup_class(d.get('xsi:type'))
return_obj = klass.from_dict(d)
else:
return_obj.id_ = d.get('id')
return_obj.idref = d.get('idref')
return_obj.efficacy = Statement.from_dict(d.get('efficacy'))
return_obj.producer = InformationSource.from_dict(
d.get('producer'))
return return_obj
def to_dict(self):
d = super(_BaseTestMechanism, self).to_dict()
d['xsi:type'] = self._XSI_TYPE # added by subclass
return d
class _BaseTestMechanism(Cached, stix.Entity):
_namespace = "http://stix.mitre.org/Indicator-2"
_binding = indicator_binding
_binding_class = indicator_binding.TestMechanismType()
def __init__(self, id_=None, idref=None):
self.id_ = id_
self.idref = idref
self.efficacy = None
self.producer = None
@property
def efficacy(self):
return self._efficacy
@efficacy.setter
def efficacy(self, value):
self._set_var(Statement, efficacy=value)
@property
def producer(self):
return self._producer
@producer.setter
def producer(self, value):
self._set_var(InformationSource, try_cast=False, producer=value)
@classmethod
def from_obj(cls, obj, return_obj=None):
if not obj:
return None
import stix.extensions.test_mechanism.snort_test_mechanism # noqa
import stix.extensions.test_mechanism.open_ioc_2010_test_mechanism # noqa
import stix.extensions.test_mechanism.yara_test_mechanism # noqa
import stix.extensions.test_mechanism.generic_test_mechanism # noqa
if not return_obj:
klass = stix.lookup_extension(obj)
return_obj = klass.from_obj(obj)
else:
return_obj.id_ = obj.id
return_obj.idref = obj.idref
return_obj.efficacy = Statement.from_obj(obj.Efficacy)
return_obj.producer = InformationSource.from_obj(obj.Producer)
return return_obj
def to_obj(self, return_obj=None, ns_info=None):
super(_BaseTestMechanism, self).to_obj(return_obj=return_obj,
ns_info=ns_info)
if not return_obj:
return_obj = self._binding_class()
return_obj.id = self.id_
return_obj.idref = self.idref
# return_obj.xsi_type = self._XSI_TYPE
if self.efficacy:
return_obj.Efficacy = self.efficacy.to_obj(ns_info=ns_info)
if self.producer:
return_obj.Producer = self.producer.to_obj(ns_info=ns_info)
return return_obj
@staticmethod
def lookup_class(xsi_type):
if not xsi_type:
raise ValueError("xsi:type is required")
return stix.lookup_extension(xsi_type)
@classmethod
def from_dict(cls, d, return_obj=None):
if not d:
return None
import stix.extensions.test_mechanism.snort_test_mechanism # noqa
import stix.extensions.test_mechanism.open_ioc_2010_test_mechanism # noqa
import stix.extensions.test_mechanism.yara_test_mechanism # noqa
import stix.extensions.test_mechanism.generic_test_mechanism # noqa
if not return_obj:
klass = stix.lookup_extension(d.get('xsi:type'))
return_obj = klass.from_dict(d)
else:
return_obj.id_ = d.get('id')
return_obj.idref = d.get('idref')
return_obj.efficacy = Statement.from_dict(d.get('efficacy'))
return_obj.producer = InformationSource.from_dict(
d.get('producer'))
return return_obj
def to_dict(self):
d = super(_BaseTestMechanism, self).to_dict()
d['xsi:type'] = self._XSI_TYPE # added by subclass
return d