From c94092c42bf2a841ce1efc78f034205bfc5680ab Mon Sep 17 00:00:00 2001 From: J08nY Date: Fri, 28 Jan 2022 14:36:44 +0100 Subject: Fix date loading from dict in MaintenanceReport. --- sec_certs/sample/common_criteria.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/sec_certs/sample/common_criteria.py b/sec_certs/sample/common_criteria.py index 46816c72..99bb0e44 100644 --- a/sec_certs/sample/common_criteria.py +++ b/sec_certs/sample/common_criteria.py @@ -38,6 +38,12 @@ class CommonCriteriaCert(Certificate, PandasSerializableType, ComplexSerializabl super().__setattr__("maintenance_title", helpers.sanitize_string(self.maintenance_title)) super().__setattr__("maintenance_date", helpers.sanitize_date(self.maintenance_date)) + @classmethod + def from_dict(cls, dct: Dict) -> "MaintenanceReport": + new_dct = dct.copy() + new_dct["maintenance_date"] = date.fromisoformat(dct["maintenance_date"]) if isinstance(dct["maintenance_date"], str) else dct["maintenance_date"] + return super().from_dict(new_dct) + def __lt__(self, other): return self.maintenance_date < other.maintenance_date -- cgit v1.3.1