56 lines
2.2 KiB
Python
56 lines
2.2 KiB
Python
"""STIX 2.0 Bundle Representation."""
|
|
|
|
from collections import OrderedDict
|
|
|
|
from ..base import _STIXBase
|
|
from ..properties import (
|
|
IDProperty, ListProperty, STIXObjectProperty, StringProperty, TypeProperty,
|
|
)
|
|
|
|
|
|
class Bundle(_STIXBase):
|
|
"""For more detailed information on this object's properties, see
|
|
`the STIX 2.0 specification <http://docs.oasis-open.org/cti/stix/v2.0/cs01/part1-stix-core/stix-v2.0-cs01-part1-stix-core.html#_Toc496709293>`__.
|
|
"""
|
|
|
|
_type = 'bundle'
|
|
_properties = OrderedDict([
|
|
('type', TypeProperty(_type)),
|
|
('id', IDProperty(_type, spec_version='2.0')),
|
|
# Not technically correct: STIX 2.0 spec doesn't say spec_version must
|
|
# have this value, but it's all we support for now.
|
|
('spec_version', StringProperty(fixed='2.0')),
|
|
('objects', ListProperty(STIXObjectProperty(spec_version="2.0"))),
|
|
])
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
# Add any positional arguments to the 'objects' kwarg.
|
|
if args:
|
|
if isinstance(args[0], list):
|
|
kwargs['objects'] = args[0] + list(args[1:]) + kwargs.get('objects', [])
|
|
else:
|
|
kwargs['objects'] = list(args) + kwargs.get('objects', [])
|
|
|
|
self._allow_custom = kwargs.get('allow_custom', False)
|
|
self._properties['objects'].contained.allow_custom = kwargs.get('allow_custom', False)
|
|
|
|
super(Bundle, self).__init__(**kwargs)
|
|
|
|
def get_obj(self, obj_uuid):
|
|
if "objects" in self._inner:
|
|
found_objs = [elem for elem in self.objects if elem['id'] == obj_uuid]
|
|
if found_objs == []:
|
|
raise KeyError("'%s' does not match the id property of any of the bundle's objects" % obj_uuid)
|
|
return found_objs
|
|
else:
|
|
raise KeyError("There are no objects in this empty bundle")
|
|
|
|
def __getitem__(self, key):
|
|
try:
|
|
return super(Bundle, self).__getitem__(key)
|
|
except KeyError:
|
|
try:
|
|
return self.get_obj(key)
|
|
except KeyError:
|
|
raise KeyError("'%s' is neither a property on the bundle nor does it match the id property of any of the bundle's objects" % key)
|