fix getitem on fips dataset

crocs-muni · Dec 5, 2022 · f14dfe3 · f14dfe3
1 parent 4712279
commit f14dfe3
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/sec_certs/dataset/fips.py b/sec_certs/dataset/fips.py
@@ -90,10 +90,10 @@ def algorithm_dataset_path(self) -> Path:
         return self.algorithms_dir / "algorithms.json"
 
     def __getitem__(self, item: str) -> FIPSCertificate:
-        if len(item) < 5:
-            return super().__getitem__(fips_dgst(item))
-        else:
+        try:
             return super().__getitem__(item)
+        except KeyError:
+            return super().__getitem__(fips_dgst(item))
 
     def _extract_data_from_html_modules(self) -> None:
         """