mouse_hook.py

"""
Platform dispatcher shim for mouse hook implementations.
"""

import sys
import types

from core.mouse_hook_types import MouseEvent

if sys.platform == "win32":
    from core import mouse_hook_windows as _platform
elif sys.platform == "darwin":
    from core import mouse_hook_macos as _platform
elif sys.platform == "linux":
    from core import mouse_hook_linux as _platform
else:
    from core import mouse_hook_stub as _platform

MouseHook = _platform.MouseHook

_RESERVED = {
    "MouseHook",
    "MouseEvent",
    "_platform",
    "_MouseHookModule",
    "_RESERVED",
    "__all__",
    "__class__",
    "__doc__",
    "__file__",
    "__loader__",
    "__name__",
    "__package__",
    "__spec__",
    "__cached__",
    "__builtins__",
}


def _should_forward(name):
    return name not in _RESERVED


class _MouseHookModule(types.ModuleType):
    def __getattr__(self, name):
        if name == "MouseEvent":
            return MouseEvent
        try:
            return getattr(_platform, name)
        except AttributeError as exc:
            raise AttributeError(
                f"module {__name__!r} has no attribute {name!r}"
            ) from exc

    def __setattr__(self, name, value):
        if _should_forward(name):
            setattr(_platform, name, value)
            return
        super().__setattr__(name, value)

    def __delattr__(self, name):
        if _should_forward(name) and hasattr(_platform, name):
            delattr(_platform, name)
            return
        super().__delattr__(name)

    def __dir__(self):
        return sorted(set(super().__dir__()) | set(dir(_platform)) | {"MouseEvent"})


module = sys.modules[__name__]
module.__class__ = _MouseHookModule
module.MouseHook = MouseHook
module.MouseEvent = MouseEvent
module.__all__ = ["MouseHook", "MouseEvent"] + list(getattr(_platform, "__all__", []))