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

some testing

pull/1/head
Raphaël Vinot 2014-03-20 11:10:52 +01:00
parent b27bb44e61
commit a920333544
1 changed files with 38 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

46
pymisp/api.py
View File

@ -13,17 +13,21 @@ URL_TMPL = URL + '/{}'
URL_XML_DOWNLOAD = URL + '/xml/download'
URL_XML_DOWNLOAD_TMPL = URL_XML_DOWNLOAD + '/{}'
OUTPUT_TYPE = 'json'
def __prepare_session():
def __prepare_session(output_type=OUTPUT_TYPE):
"""
Prepare the headers of the session
"""
session = requests.Session()
session.headers.update({'Authorization': key, 'Accept': 'application/xml'})
session.headers.update({'Authorization': key,
'Accept': 'application/' + output_type})
return session
################ REST API ################
# supports JSON and XML output
def get_index():
"""
Return the index.
@ -65,18 +69,21 @@ def delete_event(event_id):
############### XML Export ###############
# Only XML,
def download_all():
"""
Download all event from the instance
"""
session = __prepare_session()
session = __prepare_session('xml')
return session.get(URL_XML_DOWNLOAD, verify=False)
def download(event_id):
"""
Download one event in XML
"""
session = __prepare_session()
session = __prepare_session('xml')
return session.get(URL_XML_DOWNLOAD_TMPL.format(event_id), verify=False)
######### REST Search #########
@ -87,25 +94,48 @@ def __prepare_rest_search(values, not_values):
"""
to_return = ''
if values is not None:
to_return += '&&'.join(values)
if type(values) != type([]):
to_return += values
else:
to_return += '&&'.join(values)
if not_values is not None:
if len(to_return) > 0 :
to_return += '&&!'
else:
to_return += '!'
to_return += '&&!'.join(not_values)
if type(values) != type([]):
to_return += not_values
else:
to_return += '&&!'.join(not_values)
return to_return
URL_SEARCH_TMPL = 'https://misp.circl.lu/attributes/restSearch/download/{}/{}/{}/{}/{}'
URL_SEARCH_TMPL = 'https://misp.circl.lu/events/restSearch/download/{}/{}/{}/{}/{}'
# NOTE: searching by tags works, not the other options
def search(values=None, not_values=None, type_attribute=None,
category=None, org=None, tags=None, not_tags=None):
v = __prepare_rest_search(values, not_values).replace('/', '|')
t = __prepare_rest_search(tags, not_tags).replace(':', ';')
if len(v) == 0:
v = 'null'
if len(t) == 0:
t = 'null'
if type_attribute is None:
type_attribute = 'null'
if category is None:
category = 'null'
if org is None:
org = 'null'
session = __prepare_session()
session = __prepare_session('xml')
return session.get(URL_SEARCH_TMPL.format(v, type_attribute,
category, org, t), verify=False)
##########################################
if __name__ == '__main__':
r = search(values='77.67.80.31', tags='OSINT')
print unicode(r.text)
#r = get_index()
#print r.text