22 lines
614 B
Python
22 lines
614 B
Python
"""Decorator utility functions."""
|
|
from __future__ import annotations
|
|
|
|
from collections.abc import Callable, Hashable
|
|
from typing import TypeVar
|
|
|
|
CALLABLE_T = TypeVar("CALLABLE_T", bound=Callable) # pylint: disable=invalid-name
|
|
|
|
|
|
class Registry(dict):
|
|
"""Registry of items."""
|
|
|
|
def register(self, name: Hashable) -> Callable[[CALLABLE_T], CALLABLE_T]:
|
|
"""Return decorator to register item with a specific name."""
|
|
|
|
def decorator(func: CALLABLE_T) -> CALLABLE_T:
|
|
"""Register decorated function."""
|
|
self[name] = func
|
|
return func
|
|
|
|
return decorator
|