class STIXError(Exception): """Base class for errors generated in the stix2 library.""" class InvalidValueError(STIXError, ValueError): """An invalid value was provided to a STIX object's __init__.""" def __init__(self, cls, prop_name, reason): super(InvalidValueError, self).__init__() self.cls = cls self.prop_name = prop_name self.reason = reason def __str__(self): msg = "Invalid value for {0.cls.__name__} '{0.prop_name}': {0.reason}" return msg.format(self) class MissingFieldsError(STIXError, ValueError): """Missing one or more required properties when constructing STIX object.""" def __init__(self, cls, properties): super(MissingFieldsError, self).__init__() self.cls = cls self.properties = sorted(list(properties)) def __str__(self): msg = "No values for required properties for {0}: ({1})." return msg.format(self.cls.__name__, ", ".join(x for x in self.properties)) class ExtraFieldsError(STIXError, TypeError): """One or more extra properties were provided when constructing STIX object.""" def __init__(self, cls, properties): super(ExtraFieldsError, self).__init__() self.cls = cls self.properties = sorted(list(properties)) def __str__(self): msg = "Unexpected properties for {0}: ({1})." return msg.format(self.cls.__name__, ", ".join(x for x in self.properties)) class ImmutableError(STIXError, ValueError): """Attempted to modify an object after creation""" def __init__(self): super(ImmutableError, self).__init__("Cannot modify properties after creation.") class DictionaryKeyError(STIXError, ValueError): """Dictionary key does not conform to the correct format.""" def __init__(self, key, reason): super(DictionaryKeyError, self).__init__() self.key = key self.reason = reason def __str__(self): msg = "Invalid dictionary key {0.key}: ({0.reason})." return msg.format(self) class InvalidObjRefError(STIXError, ValueError): """A STIX Cyber Observable Object contains an invalid object reference.""" def __init__(self, cls, prop_name, reason): super(InvalidObjRefError, self).__init__() self.cls = cls self.prop_name = prop_name self.reason = reason def __str__(self): msg = "Invalid object reference for '{0.cls.__name__}:{0.prop_name}': {0.reason}" return msg.format(self) class UnmodifiablePropertyError(STIXError, ValueError): """Attempted to modify an unmodifiable property of object when creating a new version""" def __init__(self, unchangable_properties): super(UnmodifiablePropertyError, self).__init__() self.unchangable_properties = unchangable_properties def __str__(self): msg = "These properties cannot be changed when making a new version: {0}." return msg.format(", ".join(self.unchangable_properties)) class MutuallyExclusivePropertiesError(STIXError, TypeError): """Violating interproperty mutually exclusive constraint of a STIX object type.""" def __init__(self, cls, properties): super(MutuallyExclusivePropertiesError, self).__init__() self.cls = cls self.properties = sorted(list(properties)) def __str__(self): msg = "The field(s) for {0}: ({1}) are mutually exclusive." return msg.format(self.cls.__name__, ", ".join(x for x in self.properties)) class DependentPropertiestError(STIXError, TypeError): """Violating interproperty dependency constraint of a STIX object type.""" def __init__(self, cls, dependencies): super(DependentPropertiestError, self).__init__() self.cls = cls self.dependencies = dependencies def __str__(self): msg = "The property dependencies for {0}: ({1}) are not met." return msg.format(self.cls.__name__, ", ".join(x for x in self.dependencies)) class AtLeastOnePropertyError(STIXError, TypeError): """Violating a constraint of a STIX object type that at least one of the given properties must be populated.""" def __init__(self, cls, properties): super(AtLeastOnePropertyError, self).__init__() self.cls = cls self.properties = sorted(list(properties)) def __str__(self): msg = "At least one of the field(s) for {0}: ({1}) must be populated." return msg.format(self.cls.__name__, ", ".join(x for x in self.properties)) class RevokeError(STIXError, ValueError): """Attempted to an operation on a revoked object""" def __init__(self, called_by): super(RevokeError, self).__init__() self.called_by = called_by def __str__(self): if self.called_by == "revoke": return "Cannot revoke an already revoked object." else: return "Cannot create a new version of a revoked object."