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: