isort --profile black --length-sort --line-width 120 capa_as_library.py

This commit is contained in:
DoomedRaven
2020-12-08 21:34:42 +01:00
parent d0b2421752
commit b77d9d3738

View File

@@ -1,13 +1,13 @@
#!/usr/bin/env python3 #!/usr/bin/env python3
import collections
import json import json
import collections
import capa.main
import capa.rules
import capa.engine import capa.engine
import capa.features import capa.features
import capa.main
import capa.render.utils as rutils import capa.render.utils as rutils
import capa.rules
from capa.engine import * from capa.engine import *
from capa.render import convert_capabilities_to_result_document from capa.render import convert_capabilities_to_result_document
@@ -18,7 +18,7 @@ RULES_PATH = "/tmp/capa/rules/"
rules = capa.main.get_rules(RULES_PATH, disable_progress=True) rules = capa.main.get_rules(RULES_PATH, disable_progress=True)
rules = capa.rules.RuleSet(rules) rules = capa.rules.RuleSet(rules)
# === Render ddictionary helpers # == Render ddictionary helpers
def render_meta(doc, ostream): def render_meta(doc, ostream):
ostream["md5"] = doc["meta"]["sample"]["md5"] ostream["md5"] = doc["meta"]["sample"]["md5"]
ostream["sha1"] = doc["meta"]["sample"]["sha1"] ostream["sha1"] = doc["meta"]["sample"]["sha1"]
@@ -183,7 +183,7 @@ def render_dictionary(doc):
return ostream return ostream
# === render dictionary helpers # ==== render dictionary helpers
def capa_details(file_path: str, output_format: str="dictionary") -> dict: def capa_details(file_path: str, output_format: str="dictionary") -> dict:
# extract features and find capabilities # extract features and find capabilities