| |
| |
|
|
| |
| |
|
|
| import functools |
| import logging |
| from enum import Enum |
|
|
|
|
| @functools.lru_cache(None) |
| def warning_once(self, *args, **kwargs): |
| """ |
| This method is identical to `logger.warning()`, but will emit the warning with the same message only once |
| |
| Note: The cache is for the function arguments, so 2 different callers using the same arguments will hit the cache. |
| The assumption here is that all warning messages are unique across the code. If they aren't then need to switch to |
| another type of cache that includes the caller frame information in the hashing function. |
| """ |
| self.warning(*args, **kwargs) |
|
|
|
|
| logging.Logger.warning_once = warning_once |
| logging.Logger.warn_once = warning_once |
|
|
|
|
| class StrEnum(str, Enum): |
| """ |
| This is equivalent to Python's :class:`enum.StrEnum` since version 3.11. |
| We include this here for compatibility with older version of Python. |
| """ |
|
|
| def __str__(self) -> str: |
| return self.value |
|
|
| def __repr__(self) -> str: |
| return f"'{str(self)}'" |
|
|
