mirror of https://github.com/MISP/PyMISP
Allow to pass a pseudo file to OpenIOC loader.
parent
8175a0ecf0
commit
a55603e5c8
|
@ -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:
|
||||
|
|
Loading…
Reference in New Issue