sankalp
0a7cd2cf6d
- Replace three synced dicts with single WidgetInfo dataclass - Add type hints, unregister(), is_registered(), keys(), widgets() - Add DEFAULT_SIZE constant, duplicate-registration warning - Extract _make_fade_animator() to remove duplicated animator setup - Use public registry API in main.py instead of accessing _widgets
77 lines
2.1 KiB
Python
77 lines
2.1 KiB
Python
from dataclasses import dataclass
|
|
from types import MappingProxyType
|
|
from typing import Any
|
|
|
|
from fabric.utils import logger
|
|
|
|
|
|
@dataclass(slots=True, frozen=True)
|
|
class WidgetInfo:
|
|
widget_cls: type[Any]
|
|
size: tuple[int, int]
|
|
default_position: tuple[float, float] | None = None
|
|
|
|
|
|
class DesktopWidgetRegistry:
|
|
DEFAULT_SIZE: tuple[int, int] = (174, 174)
|
|
|
|
_registry: dict[str, WidgetInfo] = {}
|
|
_user_keys: set[str] = set()
|
|
|
|
@classmethod
|
|
def register(
|
|
cls,
|
|
key: str,
|
|
widget_cls: type[Any],
|
|
size: tuple[int, int],
|
|
default_position: tuple[float, float] | None = None,
|
|
) -> None:
|
|
if key in cls._registry:
|
|
logger.warning("Replacing widget '%s'", key)
|
|
cls._registry[key] = WidgetInfo(widget_cls, size, default_position)
|
|
|
|
@classmethod
|
|
def register_user(cls, key: str) -> None:
|
|
cls._user_keys.add(key)
|
|
|
|
@classmethod
|
|
def unregister(cls, key: str) -> None:
|
|
cls._registry.pop(key, None)
|
|
cls._user_keys.discard(key)
|
|
|
|
@classmethod
|
|
def unregister_user_keys(cls, keys: set[str]) -> None:
|
|
for key in keys:
|
|
cls.unregister(key)
|
|
|
|
@classmethod
|
|
def clear_user_widgets(cls) -> None:
|
|
cls.unregister_user_keys(set(cls._user_keys))
|
|
|
|
@classmethod
|
|
def get_widget_class(cls, key: str) -> type[Any] | None:
|
|
info = cls._registry.get(key)
|
|
return info.widget_cls if info else None
|
|
|
|
@classmethod
|
|
def get_widget_size(cls, key: str) -> tuple[int, int]:
|
|
info = cls._registry.get(key)
|
|
return info.size if info else cls.DEFAULT_SIZE
|
|
|
|
@classmethod
|
|
def get_default_position(cls, key: str) -> tuple[float, float] | None:
|
|
info = cls._registry.get(key)
|
|
return info.default_position if info else None
|
|
|
|
@classmethod
|
|
def is_registered(cls, key: str) -> bool:
|
|
return key in cls._registry
|
|
|
|
@classmethod
|
|
def keys(cls) -> tuple[str, ...]:
|
|
return tuple(cls._registry)
|
|
|
|
@classmethod
|
|
def widgets(cls) -> MappingProxyType[str, WidgetInfo]:
|
|
return MappingProxyType(cls._registry)
|