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

add add_tag method to an event and value 5 to distribution attribute

pull/6/head
Sébastien Larinier 2016-03-11 16:53:31 +01:00
parent a67a90d1d3
commit 39c06a7d24
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
pymisp/api.py
View File

@ -233,6 +233,7 @@ class PyMISP(object):
else:
return session.post(url, data=event)
def update_event(self, event_id, event, force_out=None):
"""
Update an event
@ -265,6 +266,7 @@ class PyMISP(object):
url = urljoin(self.root_url, 'attributes/{}'.format(attribute_id))
return session.delete(url)
# ##############################################
# ######### Event handling (Json only) #########
# ##############################################
@ -303,7 +305,7 @@ class PyMISP(object):
if distribution is not None:
distribution = int(distribution)
# If None: take the default value of the event
if distribution not in [None, 0, 1, 2, 3]:
if distribution not in [None, 0, 1, 2, 3,5]:
raise NewAttributeError('{} is invalid, the distribution has to be in 0, 1, 2, 3 or None'.format(distribution))
if distribution is not None:
to_return['distribution'] = distribution
@ -354,6 +356,13 @@ class PyMISP(object):
response = self.update_event(event['Event']['id'], event, 'json')
return self._check_response(response)
def add_tag(self,event, tag):
session = self.__prepare_session('json')
to_post = {'request': {'Event':{'id': event['Event']['id'], 'tag': tag}}}
response = session.post(urljoin(self.root_url, 'events/addTag'), data=json.dumps(to_post))
return self._check_response(response)
# ##### File attributes #####
def _send_attributes(self, event, attributes, proposal=False):