def end(self, value):
if not value:
self._end = None
elif isinstance(value, DateTimeWithPrecision):
self._end = value
else:
self._end = DateTimeWithPrecision(value)
def start(self, value):
if not value:
self._start = None
elif isinstance(value, DateTimeWithPrecision):
self._start = value
else:
self._start = DateTimeWithPrecision(value)
def incident_discovery(self, value):
if value:
if isinstance(value, DateTimeWithPrecision):
self._incident_discovery = value
else:
self._incident_discovery = DateTimeWithPrecision(value=value)
else:
self._incident_discovery = None
def discovered_datetime(self, value):
if value:
if isinstance(value, DateTimeWithPrecision):
self._discovered_datetime = value
else:
self._discovered_datetime = DateTimeWithPrecision(value=value)
else:
self._discovered_datetime = None
def incident_opened(self, value):
if value:
if isinstance(value, DateTimeWithPrecision):
self._incident_opened = value
else:
self._incident_opened = DateTimeWithPrecision(value=value)
else:
self._incident_opened = None
def initial_compromise(self, value):
if value:
if isinstance(value, DateTimeWithPrecision):
self._initial_compromise = value
else:
self._initial_compromise = DateTimeWithPrecision(value=value)
else:
self._initial_compromise = None
def restoration_achieved(self, value):
if value:
if isinstance(value, DateTimeWithPrecision):
self._restoration_achieved = value
else:
self._restoration_achieved = DateTimeWithPrecision(value=value)
else:
self._restoration_achieved = None
def containment_achieved(self, value):
if value:
if isinstance(value, DateTimeWithPrecision):
self._containment_achieved = value
else:
self._containment_achieved = DateTimeWithPrecision(value=value)
else:
self._containment_achieved = None
def first_data_exfiltration(self, value):
if value:
if isinstance(value, DateTimeWithPrecision):
self._first_data_exfiltration = value
else:
self._first_data_exfiltration = DateTimeWithPrecision(
value=value)
else:
self._first_data_exfiltration = None
def first_malicious_action(self, value):
if value:
if isinstance(value, DateTimeWithPrecision):
self._first_malicious_action = value
else:
self._first_malicious_action = DateTimeWithPrecision(
value=value)
else:
self._first_malicious_action = None
def test_shortcut_dict_representation(self):
"""If precision is the default value, to_dict should return a string"""
date_str = "2013-11-17T01:03:05"
d = DateTimeWithPrecision()
d.precision = "second"
d.value = date_str
self.assertEqual(str, type(d.to_dict()))
self.assertEqual(datetime.datetime, type(d.value))
self.assertEqual(date_str, d.to_dict())
d2 = round_trip(d, output=True)
self.assertEqual(d.to_dict(), d2.to_dict())
def discovered_datetime(self, value):
"""
Sets the time this vulnerability was discovered, represented as
class:`DateTimeWithPrecision`
Default Value: ``None``
Returns:
None
"""
if value:
if isinstance(value, DateTimeWithPrecision):
self._discovered_datetime = value
else:
self._discovered_datetime = DateTimeWithPrecision(value=value)
else:
self._discovered_datetime = None
def test_construction(self):
d = DateTimeWithPrecision()
d.precision = "hour"
d.value = datetime.datetime(2014, 2, 4, 8, 21, 33)
self.assertEqual(self._full_dict, d.to_dict())
def end_time(self, value):
if isinstance(value, DateTimeWithPrecision):
self._end_time = value
else:
self._end_time = DateTimeWithPrecision(value)
