aboutsummaryrefslogtreecommitdiffhomepage
path: root/src/sec_certs/utils/extract.py
diff options
context:
space:
mode:
authorJán Jančár2025-11-22 13:52:24 +0100
committerGitHub2025-11-22 13:52:24 +0100
commit8436df07381295b05a0cc892cca27e475629ee80 (patch)
tree9b940ee0a8851baaff67567c4004cd3c2cc027fd /src/sec_certs/utils/extract.py
parent7cfa1c9fc498013d1a4807e5a11f55f39e6595aa (diff)
parent21aa562cdd1c21cc66e0b19b52b414af2782377e (diff)
downloadsec-certs-main.tar.gz
sec-certs-main.tar.zst
sec-certs-main.zip
Merge pull request #530 from crocs-muni/fix/try-spacyHEADmain
Move to uv
Diffstat (limited to 'src/sec_certs/utils/extract.py')
-rw-r--r--src/sec_certs/utils/extract.py17
1 files changed, 11 insertions, 6 deletions
diff --git a/src/sec_certs/utils/extract.py b/src/sec_certs/utils/extract.py
index 35361e94..f72fd837 100644
--- a/src/sec_certs/utils/extract.py
+++ b/src/sec_certs/utils/extract.py
@@ -6,7 +6,7 @@ import re
from collections import Counter
from enum import Enum
from pathlib import Path
-from typing import Any
+from typing import Any, no_type_check
import numpy as np
@@ -17,7 +17,8 @@ from sec_certs.constants import FILE_ERRORS_STRATEGY, LINE_SEPARATOR, MAX_ALLOWE
logger = logging.getLogger(__name__)
-def search_only_headers_anssi(filepath: Path): # noqa: C901
+@no_type_check
+def search_only_headers_anssi(filepath: Path): # type: ignore # noqa: C901
# TODO: Please, refactor me. I reallyyyyyyyyyyyyy need it!!!!!!
class HEADER_TYPE(Enum):
HEADER_FULL = 1
@@ -272,7 +273,8 @@ def search_only_headers_anssi(filepath: Path): # noqa: C901
return items_found
-def search_only_headers_bsi(filepath: Path): # noqa: C901
+@no_type_check
+def search_only_headers_bsi(filepath: Path): # type: ignore # noqa: C901
# TODO: Please, refactor me. I reallyyyyyyyyyyyyy need it!!!!!!
LINE_SEPARATOR_STRICT = " "
NUM_LINES_TO_INVESTIGATE = 15
@@ -371,7 +373,8 @@ def search_only_headers_bsi(filepath: Path): # noqa: C901
return items_found
-def search_only_headers_nscib(filepath: Path): # noqa: C901
+@no_type_check
+def search_only_headers_nscib(filepath: Path): # type: ignore # noqa: C901
# TODO: Please, refactor me. I reallyyyyyyyyyyyyy need it!!!!!!
LINE_SEPARATOR_STRICT = " "
NUM_LINES_TO_INVESTIGATE = 60
@@ -451,7 +454,8 @@ def search_only_headers_nscib(filepath: Path): # noqa: C901
return items_found
-def search_only_headers_niap(filepath: Path):
+@no_type_check
+def search_only_headers_niap(filepath: Path): # type: ignore # noqa: C901
# TODO: Please, refactor me. I reallyyyyyyyyyyyyy need it!!!!!!
LINE_SEPARATOR_STRICT = " "
NUM_LINES_TO_INVESTIGATE = 15
@@ -502,7 +506,8 @@ def search_only_headers_niap(filepath: Path):
return items_found
-def search_only_headers_canada(filepath: Path): # noqa: C901
+@no_type_check
+def search_only_headers_canada(filepath: Path): # type: ignore # noqa: C901
# TODO: Please, refactor me. I reallyyyyyyyyyyyyy need it!!!!!!
LINE_SEPARATOR_STRICT = " "
NUM_LINES_TO_INVESTIGATE = 20