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

Merge pull request #56 from rmarsollier/master 

throw exception for invalid uuid
pull/60/head
Raphaël Vinot 2017-03-08 18:04:44 +01:00 committed by GitHub
parent a2c05b9480 9c36afbcc6
commit c2ef188e37
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
pymisp/api.py
View File

@ -375,12 +375,16 @@ class PyMISP(object):
return self.add_event(json.dumps(misp_event, cls=EncodeUpdate)) return self.add_event(json.dumps(misp_event, cls=EncodeUpdate))
def tag(self, uuid, tag): def tag(self, uuid, tag):
if not self._valid_uuid(uuid):
raise PyMISPError('Invalid UUID')
session = self.__prepare_session() session = self.__prepare_session()
path = '/tags/attachTagToObject/{}/{}/'.format(uuid, tag) path = '/tags/attachTagToObject/{}/{}/'.format(uuid, tag)
response = session.post(urljoin(self.root_url, path)) response = session.post(urljoin(self.root_url, path))
return self._check_response(response) return self._check_response(response)
def untag(self, uuid, tag): def untag(self, uuid, tag):
if not self._valid_uuid(uuid):
raise PyMISPError('Invalid UUID')
session = self.__prepare_session() session = self.__prepare_session()
path = '/tags/removeTagFromObject/{}/{}/'.format(uuid, tag) path = '/tags/removeTagFromObject/{}/{}/'.format(uuid, tag)
response = session.post(urljoin(self.root_url, path)) response = session.post(urljoin(self.root_url, path))