Rich support (#508)

Closes #507 https://rich.readthedocs.io/en/latest/pretty.html#rich-repr-protocol
danielgtaylor · Oct 16, 2023 · 3514991 · 3514991
1 parent e3b44f4
commit 3514991
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/betterproto/__init__.py b/src/betterproto/__init__.py
@@ -790,6 +790,10 @@ def __repr__(self) -> str:
         ]
         return f"{self.__class__.__name__}({', '.join(parts)})"
 
+    def __rich_repr__(self) -> Iterable[Tuple[str, Any, Any]]:
+        for field_name in self._betterproto.sorted_field_names:
+            yield field_name, self.__raw_get(field_name), PLACEHOLDER
+
     if not TYPE_CHECKING:
 
         def __getattribute__(self, name: str) -> Any: