move logic out of capa/render/__init__.py

This commit is contained in:
William Ballenthin
2021-06-09 18:06:51 -06:00
parent fc9ad6c737
commit 766dcacdbe
10 changed files with 376 additions and 374 deletions

View File

@@ -66,7 +66,7 @@ import multiprocessing.pool
import capa
import capa.main
import capa.rules
import capa.render
import capa.render.json
logger = logging.getLogger("capa")
@@ -205,7 +205,7 @@ def main(argv=None):
capabilities = result["ok"]["capabilities"]
# our renderer expects to emit a json document for a single sample
# so we deserialize the json document, store it in a larger dict, and we'll subsequently re-encode.
results[result["path"]] = json.loads(capa.render.render_json(meta, rules, capabilities))
results[result["path"]] = json.loads(capa.render.json.render(meta, rules, capabilities))
else:
raise ValueError("unexpected status: %s" % (result["status"]))