diff options
| author | J08nY | 2025-02-02 13:13:29 +0100 |
|---|---|---|
| committer | J08nY | 2025-02-02 13:13:29 +0100 |
| commit | 8a0cc69962baf9245170d7b481df221b1f1e3614 (patch) | |
| tree | fb45787e2fe8c34df07e69896f6dbda1405a60b9 | |
| parent | c3151fe23b8c6fa5f54fc2f59cb676ab3018b353 (diff) | |
| download | sec-certs-8a0cc69962baf9245170d7b481df221b1f1e3614.tar.gz sec-certs-8a0cc69962baf9245170d7b481df221b1f1e3614.tar.zst sec-certs-8a0cc69962baf9245170d7b481df221b1f1e3614.zip | |
Fix aux handlers super() init call.
| -rw-r--r-- | src/sec_certs/dataset/auxiliary_dataset_handling.py | 19 |
1 files changed, 11 insertions, 8 deletions
diff --git a/src/sec_certs/dataset/auxiliary_dataset_handling.py b/src/sec_certs/dataset/auxiliary_dataset_handling.py index a8a376d4..11608a2e 100644 --- a/src/sec_certs/dataset/auxiliary_dataset_handling.py +++ b/src/sec_certs/dataset/auxiliary_dataset_handling.py @@ -25,10 +25,11 @@ logger = logging.getLogger(__name__) class AuxiliaryDatasetHandler(ABC): RELATIVE_DIR: ClassVar[str | None] = None + aux_datasets_dir: Path + dset: Any def __init__(self, aux_datasets_dir: str | Path) -> None: self.aux_datasets_dir = Path(aux_datasets_dir) - self.dset: Any @property def root_dir(self) -> Path: @@ -178,9 +179,8 @@ class CCSchemeDatasetHandler(AuxiliaryDatasetHandler): aux_datasets_dir: str | Path = constants.DUMMY_NONEXISTING_PATH, only_schemes: set[str] | None = None, ): - self.aux_datasets_dir = Path(aux_datasets_dir) + super().__init__(aux_datasets_dir) self.only_schemes = only_schemes - self.dset: Any @property def dset_path(self) -> Path: @@ -203,15 +203,18 @@ class CCSchemeDatasetHandler(AuxiliaryDatasetHandler): class CCMaintenanceUpdateDatasetHandler(AuxiliaryDatasetHandler): RELATIVE_DIR: ClassVar[str] = "maintenances" + certs_with_updates: Iterable[CCCertificate] def __init__( self, aux_datasets_dir: str | Path = constants.DUMMY_NONEXISTING_PATH, - certs_with_updates: Iterable[CCCertificate] = [], + certs_with_updates: Iterable[CCCertificate] | None = None, ) -> None: - self.aux_datasets_dir = Path(aux_datasets_dir) - self.certs_with_updates = certs_with_updates - self.dset: Any + super().__init__(aux_datasets_dir) + if certs_with_updates is None: + self.certs_with_updates = [] + else: + self.certs_with_updates = certs_with_updates @property def dset_path(self) -> Path: @@ -251,7 +254,7 @@ class ProtectionProfileDatasetHandler(AuxiliaryDatasetHandler): RELATIVE_DIR: ClassVar[str] = "protection_profiles" def __init__(self, aux_datasets_dir: str | Path = constants.DUMMY_NONEXISTING_PATH): - self.aux_datasets_dir = Path(aux_datasets_dir) + super().__init__(aux_datasets_dir) @property def dset_path(self) -> Path: |