feat: add __eq__ to Snowflake (#641)

* feat: add `__eq__` to Snowflake * chore: remove outdated comments
interactions-py · Mar 14, 2022 · 63656b7 · 63656b7
1 parent d43e8dc
commit 63656b7
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/interactions/api/models/misc.py b/interactions/api/models/misc.py
@@ -166,9 +166,15 @@ def timestamp(self) -> datetime.datetime:
     def __hash__(self):
         return hash(self._snowflake)
 
-    # Do we need not equals, equals, gt/lt/ge/le?
-    # If so, list them under. By Discord API this may not be needed
-    # but end users might.
+    def __eq__(self, other):
+        if isinstance(other, Snowflake):
+            return str(self) == str(other)
+        elif isinstance(other, int):
+            return int(self) == other
+        elif isinstance(other, str):
+            return str(self) == other
+
+        return NotImplemented
 
 
 class Color(object):

diff --git a/interactions/api/models/misc.pyi b/interactions/api/models/misc.pyi
@@ -42,6 +42,7 @@ class Snowflake(object):
     def __hash__(self) -> int: ...
     def __str__(self) -> str: ...
     def __int__(self) -> int: ...
+    def __eq__(self, other) -> Union[bool, NotImplemented]: ...
 
 class Format:
     USER: str