blob: 4c9f59ca0543899f8b22cd087e4a8fe5872ecd24 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
|
from datetime import datetime
from typing import List
from attr import Factory
class RichMixin:
"""Mixin that customizes output when pretty-printed with rich. Compared to default rich behavior
for attrs classes, this does the following:
* Inform rich about all default values so they will be excluded from output
* Handle default value factories
* Stringify datetime objects
* Does not handle positional-only args (since we don't currently have any)
* Add a base repr that excludes default values even if rich isn't installed
"""
def __rich_repr__(self):
public_attrs = [a for a in self.__attrs_attrs__ if a.repr]
for a in public_attrs:
default = a.default.factory() if isinstance(a.default, Factory) else a.default
value = getattr(self, a.name)
value = str(value) if isinstance(value, datetime) else value
yield a.name, value, default
def __repr__(self):
tokens: List[str] = []
for arg in self.__rich_repr__():
key, value, default = arg
tokens.append(f'{key}={value!r}' if value != default else None)
repr_attrs = ', '.join([t for t in tokens if t])
return f'{self.__class__.__name__}({repr_attrs})'
|