TMP: rich support

mistotebe · mistotebe · commit d6e8889911a1 · 2024-12-12T13:26:35.000Z
diff --git a/Lib/ldap/response.py b/Lib/ldap/response.py
@@ -77,6 +77,10 @@ def __repr__(self):
         return (f"{self.__class__.__name__}(msgid={self.msgid}, "
                 f"msgtype={self.msgtype}{optional})")
 
+    def __rich_repr__(self):
+        yield "msgid", self.msgid
+        yield "controls", self.controls, None
+
 
 class Result(Response):
     result: int
@@ -109,6 +113,13 @@ def __repr__(self):
         return (f"{self.__class__.__name__}"
                 f"(msgid={self.msgid}, result={self.result}{optional})")
 
+    def __rich_repr__(self):
+        super().__rich_repr__()
+        yield "result", self.result
+        yield "matcheddn", self.matcheddn, ""
+        yield "message", self.message, ""
+        yield "referrals", self.referrals, None
+
 
 class SearchEntry(Response):
     msgtype = ldap.RES_SEARCH_ENTRY
@@ -125,6 +136,11 @@ def __new__(cls, msgid, msgtype, controls=None, *,
 
         return instance
 
+    def __rich_repr__(self):
+        super().__rich_repr__()
+        yield "dn", self.dn
+        yield "attrs", self.attrs
+
 
 class SearchReference(Response):
     msgtype = ldap.RES_SEARCH_REFERENCE
@@ -139,6 +155,10 @@ def __new__(cls, msgid, msgtype, controls=None, *,
 
         return instance
 
+    def __rich_repr__(self):
+        super().__rich_repr__()
+        yield "referrals", self.referrals
+
 
 class SearchResult(Result):
     msgtype = ldap.RES_SEARCH_RESULT
@@ -184,12 +204,23 @@ def __repr__(self):
         return (f"{self.__class__.__name__}"
                 f"(msgid={self.msgid}{optional})")
 
+    def __rich_repr__(self):
+        # No super(), we put our values between msgid and controls
+        yield "msgid", self.msgid
+        yield "name", self.name, None
+        yield "value", self.value, None
+        yield "controls", self.controls, None
+
 
 class BindResult(Result):
     msgtype = ldap.RES_BIND
 
     servercreds: Optional[bytes]
 
+    def __rich_repr__(self):
+        super().__rich_repr__()
+        yield "servercreds", self.servercreds, None
+
 
 class ModifyResult(Result):
     msgtype = ldap.RES_MODIFY
@@ -267,6 +298,13 @@ def __repr__(self):
         return (f"{self.__class__.__name__}"
                 f"(msgid={self.msgid}, result={self.result}{optional})")
 
+    def __rich_repr__(self):
+        # No super(), we put our values between msgid and controls
+        yield "msgid", self.msgid
+        yield "name", self.name, None
+        yield "value", self.value, None
+        yield "controls", self.controls, None
+
 
 class UnsolicitedNotification(ExtendedResult):
     msgid = ldap.RES_UNSOLICITED
