From 351499113398fc2b99c034f10263bfd081a46ed8 Mon Sep 17 00:00:00 2001 From: Micael Jarniac Date: Sun, 15 Oct 2023 23:36:26 -0300 Subject: [PATCH] Rich support (#508) Closes #507 https://rich.readthedocs.io/en/latest/pretty.html#rich-repr-protocol --- src/betterproto/__init__.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/src/betterproto/__init__.py b/src/betterproto/__init__.py index df8f3c6c..da6e57fa 100644 --- 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: