aboutsummaryrefslogtreecommitdiffhomepage
diff options
context:
space:
mode:
authorGeogeFI2022-03-19 19:31:23 +0100
committerGeogeFI2022-03-19 19:31:23 +0100
commit4ce2e8a74759591b1bde2d437396f3375f6cd597 (patch)
tree9ea73e8b25cb36c437e9ac1c08377c73620fd976
parent7034dc63fc9747452b7609560f34ed0511d8aa7b (diff)
downloadsec-certs-4ce2e8a74759591b1bde2d437396f3375f6cd597.tar.gz
sec-certs-4ce2e8a74759591b1bde2d437396f3375f6cd597.tar.zst
sec-certs-4ce2e8a74759591b1bde2d437396f3375f6cd597.zip
format: Formatting with isort, flake8, mypy
-rw-r--r--sec_certs/sample/common_criteria.py7
1 files changed, 2 insertions, 5 deletions
diff --git a/sec_certs/sample/common_criteria.py b/sec_certs/sample/common_criteria.py
index 0b04ebd2..8ae73eba 100644
--- a/sec_certs/sample/common_criteria.py
+++ b/sec_certs/sample/common_criteria.py
@@ -2,6 +2,7 @@ import copy
import operator
from dataclasses import dataclass, field
from datetime import date, datetime
+from enum import Enum
from functools import partial
from pathlib import Path
from typing import Any, Callable, ClassVar, Dict, List, Optional, Set, Tuple, Union
@@ -11,13 +12,13 @@ from bs4 import Tag
from sec_certs import constants as constants
from sec_certs import helpers
+from sec_certs.dataset.common_criteria import CCDataset
from sec_certs.model.cpe_matching import CPEClassifier
from sec_certs.model.dependency_finder import References
from sec_certs.sample.certificate import Certificate, Heuristics, logger
from sec_certs.sample.protection_profile import ProtectionProfile
from sec_certs.serialization.json import ComplexSerializableType
from sec_certs.serialization.pandas import PandasSerializableType
-from enum import Enum
HEADERS = {
"anssi": helpers.search_only_headers_anssi,
@@ -702,15 +703,11 @@ class CommonCriteriaCert(
return None
vulnerabilities = set()
- count = 0
- special_cert_id = None
for cert_id in dependency_type_dict[dependency_type]:
for cert_obj in dset:
if cert_obj.heuristics.cert_id == cert_id:
- special_cert_id = cert_id
- count += 1
if cert_obj.heuristics.related_cves:
vulnerabilities.update(cert_obj.heuristics.related_cves)