From 51c524b0c952f0509bfb7218b5f771c13d81601e Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Rapha=C3=ABl=20Vinot?= <raphael@vinot.info>
Date: Wed, 26 Jul 2017 16:37:13 +0200
Subject: [PATCH] Use dict properly

---
 pymispgalaxies/api.py | 18 +++++++++---------
 tests/tests.py        |  2 +-
 2 files changed, 10 insertions(+), 10 deletions(-)

diff --git a/pymispgalaxies/api.py b/pymispgalaxies/api.py
index c04f49c..869ba41 100644
--- a/pymispgalaxies/api.py
+++ b/pymispgalaxies/api.py
@@ -179,23 +179,23 @@ class Cluster(collections.Mapping):
         self.description = self.cluster['description']
         self.uuid = self.cluster['uuid']
         self.version = self.cluster['version']
-        self.values = {}
+        self.cluster_values = {}
         for value in self.cluster['values']:
             new_cluster_value = ClusterValue(value)
-            if self.values.get(new_cluster_value.value):
+            if self.cluster_values.get(new_cluster_value.value):
                 raise PyMISPGalaxiesError("Duplicate value ({}) in cluster: {}".format(new_cluster_value.value, self.name))
-            self.values[new_cluster_value.value] = new_cluster_value
+            self.cluster_values[new_cluster_value.value] = new_cluster_value
 
     def search(self, query):
         matching = []
-        for v in self.values.values():
+        for v in self.values():
             if [s for s in v.searchable if query.lower() in s.lower()]:
                 matching.append(v)
         return matching
 
     def machinetags(self):
         to_return = []
-        for v in self.values.values():
+        for v in self.values():
             to_return.append('misp-galaxy:{}="{}"'.format(self.type, v.value))
         return to_return
 
@@ -203,19 +203,19 @@ class Cluster(collections.Mapping):
         return '\n'.join(self.machinetags())
 
     def __getitem__(self, name):
-        return self.values[name]
+        return self.cluster_values[name]
 
     def __len__(self):
-        return len(self.values)
+        return len(self.cluster_values)
 
     def __iter__(self):
-        return iter(self.values)
+        return iter(self.cluster_values)
 
     def _json(self):
         to_return = {'name': self.name, 'type': self.type, 'source': self.source,
                      'authors': self.authors, 'description': self.description,
                      'uuid': self.uuid, 'version': self.version, 'values': []}
-        to_return['values'] = [v._json() for v in self.values.values()]
+        to_return['values'] = [v._json() for v in self.cluster_values.values()]
         return to_return
 
 
diff --git a/tests/tests.py b/tests/tests.py
index 25ecb07..234e128 100644
--- a/tests/tests.py
+++ b/tests/tests.py
@@ -44,7 +44,7 @@ class TestPyMISPGalaxies(unittest.TestCase):
     def test_meta_additional_properties(self):
         # All the properties in the meta key of the bundled-in clusters should be known
         for c in self.clusters.values():
-            for cv in c.values.values():
+            for cv in c.values():
                 if cv.meta:
                     self.assertIsNot(cv.meta.additional_properties, {})