Add Rich Equality Comparison to WriteAccessAllowed (python-telegram-bot#3911)

Bibo-Joshi · web-flow · commit 39d45124dfb2 · 2023-10-02T20:21:51.000+02:00
diff --git a/telegram/_writeaccessallowed.py b/telegram/_writeaccessallowed.py
@@ -28,7 +28,12 @@ class WriteAccessAllowed(TelegramObject):
     This object represents a service message about a user allowing a bot to write messages after
     adding the bot to the attachment menu or launching a Web App from a link.
 
+    Objects of this class are comparable in terms of equality. Two objects of this class are
+    considered equal, if their :attr:`web_app_name` is equal.
+
     .. versionadded:: 20.0
+    .. versionchanged:: NEXT.VERSION
+       Added custom equality comparison for objects of this class.
 
     Args:
         web_app_name (:obj:`str`, optional): Name of the Web App which was launched from a link.
@@ -50,4 +55,6 @@ def __init__(
         super().__init__(api_kwargs=api_kwargs)
         self.web_app_name: Optional[str] = web_app_name
 
+        self._id_attrs = (self.web_app_name,)
+
         self._freeze()
diff --git a/tests/test_writeaccessallowed.py b/tests/test_writeaccessallowed.py
@@ -36,3 +36,22 @@ def test_to_dict(self):
         action = WriteAccessAllowed()
         action_dict = action.to_dict()
         assert action_dict == {}
+
+    def test_equality(self):
+        a = WriteAccessAllowed()
+        b = WriteAccessAllowed()
+        c = WriteAccessAllowed(web_app_name="foo")
+        d = WriteAccessAllowed(web_app_name="foo")
+        e = WriteAccessAllowed(web_app_name="bar")
+
+        assert a == b
+        assert hash(a) == hash(b)
+
+        assert a != c
+        assert hash(a) != hash(c)
+
+        assert c == d
+        assert hash(c) == hash(d)
+
+        assert c != e
+        assert hash(c) != hash(e)
