PATH:
opt
/
imunify360
/
venv
/
lib
/
python3.11
/
site-packages
/
pip
/
_vendor
/
rich
import inspect from functools import partial from typing import ( Any, Callable, Iterable, List, Optional, Tuple, Type, TypeVar, Union, overload, ) T = TypeVar("T") Result = Iterable[Union[Any, Tuple[Any], Tuple[str, Any], Tuple[str, Any, Any]]] RichReprResult = Result class ReprError(Exception): """An error occurred when attempting to build a repr.""" @overload def auto(cls: Optional[Type[T]]) -> Type[T]: ... @overload def auto(*, angular: bool = False) -> Callable[[Type[T]], Type[T]]: ... def auto( cls: Optional[Type[T]] = None, *, angular: Optional[bool] = None ) -> Union[Type[T], Callable[[Type[T]], Type[T]]]: """Class decorator to create __repr__ from __rich_repr__""" def do_replace(cls: Type[T], angular: Optional[bool] = None) -> Type[T]: def auto_repr(self: T) -> str: """Create repr string from __rich_repr__""" repr_str: List[str] = [] append = repr_str.append angular: bool = getattr(self.__rich_repr__, "angular", False) # type: ignore[attr-defined] for arg in self.__rich_repr__(): # type: ignore[attr-defined] if isinstance(arg, tuple): if len(arg) == 1: append(repr(arg[0])) else: key, value, *default = arg if key is None: append(repr(value)) else: if default and default[0] == value: continue append(f"{key}={value!r}") else: append(repr(arg)) if angular: return f"<{self.__class__.__name__} {' '.join(repr_str)}>" else: return f"{self.__class__.__name__}({', '.join(repr_str)})" def auto_rich_repr(self: Type[T]) -> Result: """Auto generate __rich_rep__ from signature of __init__""" try: signature = inspect.signature(self.__init__) for name, param in signature.parameters.items(): if param.kind == param.POSITIONAL_ONLY: yield getattr(self, name) elif param.kind in ( param.POSITIONAL_OR_KEYWORD, param.KEYWORD_ONLY, ): if param.default is param.empty: yield getattr(self, param.name) else: yield param.name, getattr(self, param.name), param.default except Exception as error: raise ReprError( f"Failed to auto generate __rich_repr__; {error}" ) from None if not hasattr(cls, "__rich_repr__"): auto_rich_repr.__doc__ = "Build a rich repr" cls.__rich_repr__ = auto_rich_repr # type: ignore[attr-defined] auto_repr.__doc__ = "Return repr(self)" cls.__repr__ = auto_repr # type: ignore[assignment] if angular is not None: cls.__rich_repr__.angular = angular # type: ignore[attr-defined] return cls if cls is None: return partial(do_replace, angular=angular) else: return do_replace(cls, angular=angular) @overload def rich_repr(cls: Optional[Type[T]]) -> Type[T]: ... @overload def rich_repr(*, angular: bool = False) -> Callable[[Type[T]], Type[T]]: ... def rich_repr( cls: Optional[Type[T]] = None, *, angular: bool = False ) -> Union[Type[T], Callable[[Type[T]], Type[T]]]: if cls is None: return auto(angular=angular) else: return auto(cls) if __name__ == "__main__": @auto class Foo: def __rich_repr__(self) -> Result: yield "foo" yield "bar", {"shopping": ["eggs", "ham", "pineapple"]} yield "buy", "hand sanitizer" foo = Foo() from pip._vendor.rich.console import Console console = Console() console.rule("Standard repr") console.print(foo) console.print(foo, width=60) console.print(foo, width=30) console.rule("Angular repr") Foo.__rich_repr__.angular = True # type: ignore[attr-defined] console.print(foo) console.print(foo, width=60) console.print(foo, width=30)
[-] jupyter.py
[edit]
[-] errors.py
[edit]
[-] progress_bar.py
[edit]
[-] _log_render.py
[edit]
[-] pretty.py
[edit]
[-] spinner.py
[edit]
[-] _inspect.py
[edit]
[-] _windows_renderer.py
[edit]
[-] segment.py
[edit]
[-] repr.py
[edit]
[-] live_render.py
[edit]
[-] live.py
[edit]
[-] _windows.py
[edit]
[-] panel.py
[edit]
[-] default_styles.py
[edit]
[-] ansi.py
[edit]
[-] _fileno.py
[edit]
[-] highlighter.py
[edit]
[+]
..
[-] constrain.py
[edit]
[-] measure.py
[edit]
[-] _win32_console.py
[edit]
[-] themes.py
[edit]
[-] layout.py
[edit]
[-] color.py
[edit]
[-] _timer.py
[edit]
[-] _palettes.py
[edit]
[-] _emoji_codes.py
[edit]
[-] __main__.py
[edit]
[-] _extension.py
[edit]
[-] emoji.py
[edit]
[-] theme.py
[edit]
[-] _pick.py
[edit]
[-] style.py
[edit]
[-] screen.py
[edit]
[-] syntax.py
[edit]
[-] py.typed
[edit]
[-] abc.py
[edit]
[-] region.py
[edit]
[-] palette.py
[edit]
[-] _export_format.py
[edit]
[-] protocol.py
[edit]
[-] text.py
[edit]
[-] prompt.py
[edit]
[-] status.py
[edit]
[+]
__pycache__
[-] align.py
[edit]
[-] color_triplet.py
[edit]
[-] bar.py
[edit]
[-] control.py
[edit]
[-] styled.py
[edit]
[-] progress.py
[edit]
[-] console.py
[edit]
[-] scope.py
[edit]
[-] traceback.py
[edit]
[-] _null_file.py
[edit]
[-] rule.py
[edit]
[-] logging.py
[edit]
[-] box.py
[edit]
[-] table.py
[edit]
[-] terminal_theme.py
[edit]
[-] _spinners.py
[edit]
[-] filesize.py
[edit]
[-] padding.py
[edit]
[-] __init__.py
[edit]
[-] _wrap.py
[edit]
[-] columns.py
[edit]
[-] _cell_widths.py
[edit]
[-] markup.py
[edit]
[-] containers.py
[edit]
[-] json.py
[edit]
[-] diagnose.py
[edit]
[-] _loop.py
[edit]
[-] _ratio.py
[edit]
[-] _emoji_replace.py
[edit]
[-] pager.py
[edit]
[-] tree.py
[edit]
[-] file_proxy.py
[edit]
[-] _stack.py
[edit]
[-] cells.py
[edit]