MISP
/
MISP-Taxii-Server
mirror of https://github.com/MISP/MISP-Taxii-Server
2
0
Fork 0
Code Issues Releases Wiki Activity

Include Loader in hooks.py

pull/82/head
arcsector 2020-05-06 17:00:04 -07:00 committed by GitHub
parent c86e4940af
commit 53309e3477
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
misp_taxii_hooks/hooks.py
View File

@ -9,6 +9,7 @@ import pymisp
import tempfile import tempfile
import logging import logging
from pyaml import yaml from pyaml import yaml
from yaml import Loader
from io import StringIO from io import StringIO
log = logging.getLogger("__main__") log = logging.getLogger("__main__")
@ -20,7 +21,7 @@ from opentaxii.signals import (
## CONFIG ## CONFIG
if "OPENTAXII_CONFIG" in os.environ: if "OPENTAXII_CONFIG" in os.environ:
print("Using config from {}".format(os.environ["OPENTAXII_CONFIG"])) 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: else:
print("Trying to use env variables...") print("Trying to use env variables...")
if "MISP_URL" in os.environ: if "MISP_URL" in os.environ: