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

Allow to add a tag to a MISPEvent and MISPAttribute

pull/49/head
Raphaël Vinot 2017-01-26 14:36:01 +01:00
parent d260dc5ea5
commit 48e1211ed8
1 changed files with 16 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
pymisp/mispevent.py
View File

@ -101,6 +101,9 @@ class MISPAttribute(object):
def delete(self): def delete(self):
self.deleted = True self.deleted = True
def add_tag(self, tag):
self.Tag.append({'name': tag})
def verify(self, gpg_uid): def verify(self, gpg_uid):
if not has_pyme: if not has_pyme:
raise Exception('pyme is required, please install: pip install --pre pyme3. You will also need libgpg-error-dev and libgpgme11-dev.') raise Exception('pyme is required, please install: pip install --pre pyme3. You will also need libgpg-error-dev and libgpgme11-dev.')
@ -549,6 +552,19 @@ class MISPEvent(object):
jsonschema.validate(to_return, self.json_schema) jsonschema.validate(to_return, self.json_schema)
return to_return return to_return
def add_tag(self, tag):
self.Tag.append({'name': tag})
def add_attribute_tag(self, tag, attribute_identifier):
attribute = None
for a in self.attributes:
if a.id == attribute_identifier or a.uuid == attribute_identifier or attribute_identifier in a.value:
a.add_tag(tag)
attribute = a
if not attribute:
raise Exception('No attribute with identifier {} found.'.format(attribute_identifier))
return attribute
def publish(self): def publish(self):
self.published = True self.published = True