cti-python-stix2/stix2/common.py

54 lines
1.3 KiB
Python

"""STIX 2 Common Data Types and Properties"""
import re
from .base import _STIXBase
from .properties import Property
from .utils import NOW
ref_regex = ("^[a-z][a-z-]+[a-z]--[0-9a-fA-F]{8}-[0-9a-fA-F]{4}-[0-9a-fA-F]{4}"
"-[0-9a-fA-F]{4}-[0-9a-fA-F]{12}$")
REF_PROPERTY = {
'validate': (lambda x, val: re.match(ref_regex, val)),
'error_msg': "{type} {field} values must consist of a valid STIX type name and a valid UUID, separated by '--'."
}
BOOL_PROPERTY = {
'validate': (lambda x, val: isinstance(val, bool)),
'error_msg': "{type} {field} value must be a boolean."
}
COMMON_PROPERTIES = {
# 'type' and 'id' should be defined on each individual type
'created': {
'default': NOW,
},
'modified': {
'default': NOW,
},
'external_references': {},
'revoked': BOOL_PROPERTY,
'created_by_ref': REF_PROPERTY
}
class ExternalReference(_STIXBase):
_properties = {
'source_name': Property(required=True),
'description': Property(),
'url': Property(),
'external_id': Property(),
}
class KillChainPhase(_STIXBase):
_properties = {
'kill_chain_name': {
'required': True,
},
'phase_name': {
'required': True,
},
}