vee1e/capa
1
0
Fork 0
mirror of https://github.com/vee1e/capa.git synced 2026-09-01 10:48:51 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

scripts: add type-related assert statements

Browse source
This commit is contained in:
Yacine Elhamer 2023-06-26 21:06:35 +01:00
parent 3f5d08aedb
commit c74c8871f8
2 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

5
scripts/show-capabilities-by-function.py
View file

@ -70,7 +70,7 @@ import capa.render.result_document as rd
from capa.helpers import get_file_taste
from capa.features.common import FORMAT_AUTO
from capa.features.freeze import Address
from capa.features.extractors.base_extractor import FeatureExtractor
from capa.features.extractors.base_extractor import StaticFeatureExtractor
logger = logging.getLogger("capa.show-capabilities-by-function")
@ -167,9 +167,10 @@ def main(argv=None):
should_save_workspace = os.environ.get("CAPA_SAVE_WORKSPACE") not in ("0", "no", "NO", "n", None)
try:
extractor: FeatureExtractor = capa.main.get_extractor(
extractor = capa.main.get_extractor(
args.sample, args.format, args.os, args.backend, sig_paths, should_save_workspace
)
assert isinstance(extractor, StaticFeatureExtractor)
except capa.exceptions.UnsupportedFormatError:
capa.helpers.log_unsupported_format_error()
return -1

1
scripts/show-features.py
View file

@ -124,6 +124,7 @@ def main(argv=None):
extractor = capa.main.get_extractor(
args.sample, args.format, args.os, args.backend, sig_paths, should_save_workspace
)
assert isinstance(extractor, capa.features.extractors.base_extractor.StaticFeatureExtractor)
except capa.exceptions.UnsupportedFormatError:
capa.helpers.log_unsupported_format_error()
return -1