TMP: rich support

mistotebe · mistotebe · commit 90652b1e77c5 · 2024-09-04T11:35:32.000+01:00
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
@@ -176,17 +196,30 @@ def __repr__(self):
             optional += f", name={self.name}"
         if self.value is not None:
             optional += f", value={self.value}"
+        if self.message:
+            optional = f", message={self.message!r}"
         if self.controls is not None:
             optional += f", controls={self.controls}"
         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
@@ -264,6 +297,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
