Allow to pass a pseudo file to OpenIOC loader.

pull/55/head
Raphaël Vinot 2017-02-27 11:33:07 +01:00
parent 8175a0ecf0
commit a55603e5c8
1 changed files with 26 additions and 17 deletions

View File

@ -1,6 +1,8 @@
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import os
from pymisp import MISPEvent
try:
from bs4 import BeautifulSoup
@ -79,12 +81,19 @@ def extract_field(report, field_name):
return None
def load_openioc_file(openioc_path):
if not os.path.exists(openioc_path):
raise Exception("Path doesn't exists.")
with open(openioc_path, 'r') as f:
return load_openioc(f)
def load_openioc(openioc):
# Takes a opened file, or a string
if not has_bs4:
raise Exception('You need to install BeautifulSoup: pip install bs4')
misp_event = MISPEvent()
with open(openioc, "r") as ioc_file:
iocreport = BeautifulSoup(ioc_file, "lxml")
iocreport = BeautifulSoup(openioc, "lxml")
# Set event fields
info = extract_field(iocreport, 'short_description')
if info: