Skip to content

Commit

Permalink
fmt
Browse files Browse the repository at this point in the history
  • Loading branch information
williballenthin committed Feb 14, 2024
1 parent b163f82 commit 0f9dd90
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 1 deletion.
4 changes: 4 additions & 0 deletions capa/loader.py
Original file line number Diff line number Diff line change
Expand Up @@ -279,20 +279,24 @@ def get_file_extractors(input_file: Path, input_format: str) -> List[FeatureExtr

if input_format == FORMAT_PE:
import capa.features.extractors.pefile

file_extractors.append(capa.features.extractors.pefile.PefileFeatureExtractor(input_file))

elif input_format == FORMAT_DOTNET:
import capa.features.extractors.pefile
import capa.features.extractors.dotnetfile

file_extractors.append(capa.features.extractors.pefile.PefileFeatureExtractor(input_file))
file_extractors.append(capa.features.extractors.dotnetfile.DotnetFileFeatureExtractor(input_file))

elif input_format == FORMAT_ELF:
import capa.features.extractors.elffile

file_extractors.append(capa.features.extractors.elffile.ElfFeatureExtractor(input_file))

elif input_format == FORMAT_CAPE:
import capa.features.extractors.cape.extractor

report = json.loads(input_file.read_text(encoding="utf-8"))
file_extractors.append(capa.features.extractors.cape.extractor.CapeExtractor.from_report(report))

Expand Down
1 change: 0 additions & 1 deletion capa/rules/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -1058,7 +1058,6 @@ def _get_ruamel_yaml_parser():
import ruamel.yaml

# use ruamel to enable nice formatting

# we use the ruamel.yaml parser because it supports roundtripping of documents with comments.
y = ruamel.yaml.YAML(typ="rt")

Expand Down

0 comments on commit 0f9dd90

Please sign in to comment.