From 14fadcb3c49f0e1d02dd87f819c2c753b04a14fe Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Rapha=C3=ABl=20Vinot?= Date: Thu, 5 Apr 2018 11:36:24 +0200 Subject: [PATCH] fix: Add support for uuid --- pymispgalaxies/api.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/pymispgalaxies/api.py b/pymispgalaxies/api.py index 39a0e24..661e6fc 100644 --- a/pymispgalaxies/api.py +++ b/pymispgalaxies/api.py @@ -164,10 +164,13 @@ class ClusterValue(): def __init__(self, v): if not v['value']: raise PyMISPGalaxiesError("Invalid cluster (no value): {}".format(v)) + self.uuid = v.get('uuid', None) self.value = v['value'] self.description = v.get('description') self.meta = self.__init_meta(v.get('meta')) self.searchable = [self.value] + if self.uuid: + self.searchable.append(self.uuid) if self.meta and self.meta.synonyms: self.searchable += self.meta.synonyms @@ -181,6 +184,8 @@ class ClusterValue(): def to_dict(self): to_return = {'value': self.value} + if self.uuid: + to_return['uuid'] = self.uuid if self.description: to_return['description'] = self.description if self.meta: