diff options
Diffstat (limited to 'sec_certs/sample/protection_profile.py')
| -rw-r--r-- | sec_certs/sample/protection_profile.py | 22 |
1 files changed, 11 insertions, 11 deletions
diff --git a/sec_certs/sample/protection_profile.py b/sec_certs/sample/protection_profile.py index bb7cc788..d8825ced 100644 --- a/sec_certs/sample/protection_profile.py +++ b/sec_certs/sample/protection_profile.py @@ -1,11 +1,10 @@ -from dataclasses import dataclass -from typing import Optional, FrozenSet import copy -from typing import Dict import logging +from dataclasses import dataclass +from typing import Dict, FrozenSet, Optional -from sec_certs.serialization.json import ComplexSerializableType import sec_certs.helpers as helpers +from sec_certs.serialization.json import ComplexSerializableType logger = logging.getLogger(__name__) @@ -15,25 +14,26 @@ class ProtectionProfile(ComplexSerializableType): """ Object for holding protection profiles. """ - pp_name: Optional[str] + + pp_name: str pp_link: Optional[str] = None pp_ids: Optional[FrozenSet[str]] = None def __post_init__(self): - super().__setattr__('pp_name', helpers.sanitize_string(self.pp_name)) - super().__setattr__('pp_link', helpers.sanitize_link(self.pp_link)) + super().__setattr__("pp_name", helpers.sanitize_string(self.pp_name)) + super().__setattr__("pp_link", helpers.sanitize_link(self.pp_link)) @classmethod def from_dict(cls, dct): new_dct = copy.deepcopy(dct) - new_dct['pp_ids'] = frozenset(new_dct['pp_ids']) if new_dct['pp_ids'] else None + new_dct["pp_ids"] = frozenset(new_dct["pp_ids"]) if new_dct["pp_ids"] else None return cls(*tuple(new_dct.values())) @classmethod def from_old_api_dict(cls, dct: Dict): - pp_name = helpers.sanitize_string(dct['csv_scan']['cc_pp_name']) - pp_link = helpers.sanitize_link(dct['csv_scan']['link_pp_document']) - pp_ids = frozenset(dct['processed']['cc_pp_csvid']) if dct['processed']['cc_pp_csvid'] else None + pp_name = helpers.sanitize_string(dct["csv_scan"]["cc_pp_name"]) + pp_link = helpers.sanitize_link(dct["csv_scan"]["link_pp_document"]) + pp_ids = frozenset(dct["processed"]["cc_pp_csvid"]) if dct["processed"]["cc_pp_csvid"] else None return cls(pp_name, pp_link, pp_ids) def __eq__(self, other): |
