Reword ThingyList.distinct code so that it makes more sense

Refty · Dec 14, 2022 · 7ff4839 · 7ff4839
1 parent 829c331
commit 7ff4839
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/mongo_thingy/__init__.py b/mongo_thingy/__init__.py
@@ -21,12 +21,12 @@
 
 class ThingyList(list):
     def distinct(self, key):
-        def __get_value(document):
-            if hasattr(document, "__dict__"):
-                document = document.__dict__
-            return document.get(key)
+        def __get_value(item):
+            if isinstance(item, BaseThingy):
+                item = item.view()
+            return item.get(key)
 
-        values = set(__get_value(result) for result in self)
+        values = set(__get_value(item) for item in self)
         return list(values)