Source code for requests_cache.models.base

from datetime import datetime
from typing import List

from attrs 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})'