From 53309e3477234111039eb4559b758aab69daae19 Mon Sep 17 00:00:00 2001 From: arcsector Date: Wed, 6 May 2020 17:00:04 -0700 Subject: [PATCH] Include Loader in hooks.py --- misp_taxii_hooks/hooks.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/misp_taxii_hooks/hooks.py b/misp_taxii_hooks/hooks.py index 4119797..85d7566 100644 --- a/misp_taxii_hooks/hooks.py +++ b/misp_taxii_hooks/hooks.py @@ -9,6 +9,7 @@ import pymisp import tempfile import logging from pyaml import yaml +from yaml import Loader from io import StringIO log = logging.getLogger("__main__") @@ -20,7 +21,7 @@ from opentaxii.signals import ( ## CONFIG if "OPENTAXII_CONFIG" in os.environ: print("Using config from {}".format(os.environ["OPENTAXII_CONFIG"])) - CONFIG = yaml.load(open(os.environ["OPENTAXII_CONFIG"], "r")) + CONFIG = yaml.load(open(os.environ["OPENTAXII_CONFIG"], "r"), Loader=Loader) else: print("Trying to use env variables...") if "MISP_URL" in os.environ: