From 932c451dd9c66b8938676105d0ffc1823f131cf8 Mon Sep 17 00:00:00 2001 From: Will McGugan Date: Wed, 9 Jun 2021 18:28:10 +0100 Subject: [PATCH] repr example --- examples/repr.py | 15 +++++++++------ 1 file changed, 9 insertions(+), 6 deletions(-) diff --git a/examples/repr.py b/examples/repr.py index b6e5a47a..628d7d09 100644 --- a/examples/repr.py +++ b/examples/repr.py @@ -1,3 +1,7 @@ +from rich.repr import rich_repr + + +@rich_repr class Bird: def __init__(self, name, eats=None, fly=True, extinct=False): self.name = name @@ -5,9 +9,6 @@ class Bird: self.fly = fly self.extinct = extinct - def __repr__(self): - return f"Bird({self.name!r}, eats={self.eats!r}, fly={self.fly!r}, extinct={self.extinct!r})" - def __rich_repr__(self): yield self.name yield "eats", self.eats @@ -15,11 +16,13 @@ class Bird: yield "extinct", self.extinct, False __rich_repr__.angular = True - -from rich import print + + +from rich import print + BIRDS = { "gull": Bird("gull", eats=["fish", "chips", "ice cream", "sausage rolls"]), "penguin": Bird("penguin", eats=["fish"], fly=False), - "dodo": Bird("dodo", eats=["fruit"], fly=False, extinct=True) + "dodo": Bird("dodo", eats=["fruit"], fly=False, extinct=True), } print(BIRDS)