RinUI 0.1.0__py3-none-any.whl → 0.1.1__py3-none-any.whl

RinUI/core/theme.py

@@ -1,339 +1,339 @@
-import ctypes
-import platform
-import time
-
-from PySide6.QtCore import QObject, Signal, Slot, QThread
-
-from .config import DEFAULT_CONFIG, RinConfig, is_win10, is_windows, is_win11, BackdropEffect
-import sys
-import darkdetect
-
-
-def check_darkdetect_support():
-    system = platform.system()
-    if system == "Darwin":
-        mac_ver = platform.mac_ver()[0]
-        major, minor, *_ = map(int, mac_ver.split("."))
-        return (major == 10 and minor >= 14) or major > 10
-
-    elif system == "Windows":
-        return platform.release() >= "10"
-    else:
-        return False
-
-
-ACCENT_STATES = {
-    "acrylic": 3,
-    "mica": 2,
-    "tabbed": 4,
-    "none": 0
-}
-
-ACCENT_SUPPORT = {
-    "acrylic": is_win10(),
-    "mica": is_win11(),
-    "tabbed": is_win10(),
-    "none": True
-}
-
-
-class ACCENT_POLICY(ctypes.Structure):
-    _fields_ = [
-        ("AccentState", ctypes.c_int),
-        ("AccentFlags", ctypes.c_int),
-        ("GradientColor", ctypes.c_int),
-        ("AnimationId", ctypes.c_int),
-    ]
-
-
-class WINDOWCOMPOSITIONATTRIBDATA(ctypes.Structure):
-    _fields_ = [
-        ("Attrib", ctypes.c_int),
-        ("pvData", ctypes.c_void_p),
-        ("cbData", ctypes.c_size_t),
-    ]
-
-
-class ThemeListener(QThread):
-    """
-    监听系统颜色模式
-    """
-    themeChanged = Signal(str)
-
-    def run(self):
-        last_theme = darkdetect.theme()
-        while True:
-            current_theme = darkdetect.theme()
-            if current_theme != last_theme:
-                last_theme = current_theme
-                self.themeChanged.emit(current_theme)
-                print(f"Theme changed: {current_theme}")
-            time.sleep(1)
-
-    def stop(self):
-        self.terminate()
-
-
-class ThemeManager(QObject):
-    themeChanged = Signal(str)
-    backdropChanged = Signal(str)
-    windows = []  # 窗口句柄们（
-    _instance = None
-
-    # DWM 常量保持不变
-    DWMWA_USE_IMMERSIVE_DARK_MODE = 20
-    DWMWA_WINDOW_CORNER_PREFERENCE = 33
-    DWMWA_NCRENDERING_POLICY = 2
-    DWMNCRENDERINGPOLICY_ENABLED = 2
-    DWMWA_SYSTEMBACKDROP_TYPE = 38
-    WCA_ACCENT_POLICY = 19
-
-    # 圆角
-    DWMWCP_DEFAULT = 0
-    DWMWCP_DONOTROUND = 1
-    DWMWCP_ROUND = 2
-    DWMWCP_ROUNDSMALL = 3
-
-    def clean_up(self):
-        """
-        清理资源并停止主题监听。
-        """
-        if self.listener:
-            RinConfig.save_config()
-            print("Save config.")
-            self.listener.stop()
-            self.listener.wait()  # 等待线程结束
-            print("Theme listener stopped.")
-
-    def __new__(cls, *args, **kwargs):
-        """
-        单例管理，共享主题状态
-        :param args:
-        :param kwargs:
-        """
-        if cls._instance is None:
-            cls._instance = super().__new__(cls)
-        return cls._instance
-
-    def __init__(self):
-        if hasattr(self, "_initialized") and self._initialized:
-            return
-        self._initialized = True
-        super().__init__()
-        self.theme_dict = {
-            "Light": 0,
-            "Dark": 1
-        }
-
-        self.listener = None  # 监听线程
-        self.current_theme = DEFAULT_CONFIG["theme"]["current_theme"]  # 当前主题
-        self.is_darkdetect_supported = check_darkdetect_support()
-
-        try:
-            self.current_theme = RinConfig["theme"]["current_theme"]
-        except Exception as e:
-            print(f"Failed to load config because of {e}, using default config")
-
-        # self.hwnd = None  # 窗口句柄
-
-        self.start_listener()
-
-    def start_listener(self):
-        if not self.is_darkdetect_supported:
-            print("darkdetect not supported on this platform")
-            return
-        self.listener = ThemeListener()
-        self.listener.themeChanged.connect(self._handle_system_theme)
-        self.listener.start()
-
-    def set_window(self, window):  # 绑定窗口句柄
-        hwnd = int(window.winId())
-        self.windows.append(hwnd)
-        print(f"Window handle set: {hwnd}")
-
-    def _handle_system_theme(self):
-        if self.current_theme == "Auto":
-            self._update_window_theme()
-            self.themeChanged.emit(self._actual_theme())
-        else:
-            # 保持当前背景效果不变
-            self._update_window_theme()
-
-    @Slot(str)
-    def apply_backdrop_effect(self, effect_type: str):
-        """
-        应用背景效果
-        :param effect_type: str, 背景效果类型（acrylic, mica, tabbed, none）
-        """
-        self._update_window_theme()
-        if not is_windows() or not self.windows:
-            print(f"Cannot apply effect \"{effect_type}\" on this platform")
-            return -2  # 非 windows或未绑定窗口
-        self.backdropChanged.emit(effect_type)
-
-        accent_state = ACCENT_STATES.get(effect_type, 0)
-        if not ACCENT_SUPPORT.get(effect_type, False):
-            print(f"Effect \"{effect_type}\" not supported on this platform")
-            return -1  # 效果不支持
-
-        for hwnd in self.windows:
-            if is_win11():
-                ctypes.windll.dwmapi.DwmSetWindowAttribute(
-                    hwnd,
-                    self.DWMWA_SYSTEMBACKDROP_TYPE,
-                    ctypes.byref(ctypes.c_int(accent_state)),
-                    ctypes.sizeof(ctypes.c_int)
-                )
-            elif is_win10() and effect_type == BackdropEffect.Acrylic.value:
-                self._apply_win10_effect(effect_type, hwnd)
-
-        RinConfig["backdrop_effect"] = effect_type
-        print(
-            f"Applied \"{effect_type.strip().capitalize()}\" effect with "
-            f"{platform.system() + '11' if is_win11() else '10'}"
-        )
-        return 0  # 成功
-
-    def _apply_win10_effect(self, effect_type, hwnd):
-        """
-        应用 Windows 10 背景效果
-        :param effect_type: str, 背景效果类型（acrylic, tabbed(actually blur)
-        """
-        backdrop_color = RinConfig["win10_feat"]["backdrop_dark" if self.is_dark_theme() else "backdrop_light"]
-
-        accent = ACCENT_POLICY()
-        accent.AccentState = ACCENT_STATES[effect_type]
-        accent.AccentFlags = 2
-        accent.GradientColor = backdrop_color
-        data = WINDOWCOMPOSITIONATTRIBDATA()
-        data.Attrib = self.WCA_ACCENT_POLICY
-        data.pvData = ctypes.cast(ctypes.pointer(accent), ctypes.c_void_p)
-        data.cbData = ctypes.sizeof(accent)
-
-        try:
-            set_window_composition = ctypes.windll.user32.SetWindowCompositionAttribute
-            set_window_composition(hwnd, ctypes.byref(data))
-        except Exception as e:
-            print(f"Failed to apply acrylic on Win10: {e}")
-
-    def apply_window_effects(self):  # 启用圆角阴影
-        if sys.platform != "win32" or not self.windows:
-            return
-
-        dwm = ctypes.windll.dwmapi
-
-        # 启用非客户端渲染策略（让窗口边框具备阴影）
-        ncrp = ctypes.c_int(self.DWMNCRENDERINGPOLICY_ENABLED)
-        for hwnd in self.windows:
-            dwm.DwmSetWindowAttribute(
-                hwnd,
-                self.DWMWA_NCRENDERING_POLICY,
-                ctypes.byref(ncrp),
-                ctypes.sizeof(ncrp)
-            )
-
-            # 启用圆角效果
-            corner_preference = ctypes.c_int(self.DWMWCP_ROUND)
-            dwm.DwmSetWindowAttribute(
-                hwnd,
-                self.DWMWA_WINDOW_CORNER_PREFERENCE,
-                ctypes.byref(corner_preference),
-                ctypes.sizeof(corner_preference)
-            )
-        print("Enabled Rounded and Shadows")
-
-    def _update_window_theme(self):  # 更新窗口的颜色模式
-        if sys.platform != "win32" or not self.windows:
-            return
-        actual_theme = self._actual_theme()
-        for hwnd in self.windows:
-            if is_win11():
-                ctypes.windll.dwmapi.DwmSetWindowAttribute(
-                    hwnd,
-                    self.DWMWA_USE_IMMERSIVE_DARK_MODE,
-                    ctypes.byref(ctypes.c_int(self.theme_dict[actual_theme])),
-                    ctypes.sizeof(ctypes.c_int)
-                )
-            elif is_win10() and RinConfig["backdrop_effect"] == BackdropEffect.Acrylic.value:
-                self._apply_win10_effect(RinConfig["backdrop_effect"], hwnd)
-            else:
-                print(f"Cannot apply backdrop on {platform.system()}")
-
-        print(f"Window theme updated to {actual_theme}")
-
-    def is_dark_theme(self):
-        """是否为暗黑主题"""
-        return self._actual_theme() == "Dark"
-
-    def _actual_theme(self):
-        """实际应用的主题"""
-        if self.current_theme == "Auto":
-            return darkdetect.theme() if self.is_darkdetect_supported else "Light"
-        return self.current_theme
-
-    @Slot(str)
-    def toggle_theme(self, theme: str):  # 切换主题
-        if theme not in ["Auto", "Light", "Dark"]:  # 三状态
-            return
-        if self.current_theme != theme:
-            print(f"Switching to '{theme}' theme")
-            self.current_theme = theme
-            RinConfig["theme"]["current_theme"] = theme
-            self._update_window_theme()
-            self.themeChanged.emit(self._actual_theme())
-
-    @Slot(result=str)
-    def get_theme(self):
-        return self._actual_theme()
-
-    @Slot(result=str)
-    def get_theme_name(self):
-        """获取当前主题名称"""
-        return self.current_theme
-
-    @Slot(str)
-    def receive(self, message):
-        print(message)
-
-    @Slot(result=str)
-    def get_backdrop_effect(self):
-        """获取当前背景效果"""
-        return RinConfig["backdrop_effect"]
-
-    @Slot(str)
-    def set_theme_color(self, color):
-        """设置当前主题颜色"""
-        RinConfig["theme_color"] = color
-        RinConfig.save_config()
-
-    @Slot(result=str)
-    def get_theme_color(self):
-        """获取当前主题颜色"""
-        return RinConfig["theme_color"]
-
-    @Slot(QObject, result=int)
-    def getWindowId(self, window):
-        """获取窗口的句柄"""
-        print(f"GetWindowId: {window.winId()}")
-        return int(window.winId())
-
-    @Slot(int)
-    def dragWindowEvent(self, hwnd):
-        """ 在Windows 用原生方法拖动"""
-        if not is_windows() or hwnd not in self.windows:
-            print(
-                f"Use Qt method to drag window on: {platform.system()}"
-                if not is_windows() else f"Invalid window handle: {hwnd}"
-            )
-            return
-
-        import win32con
-        from win32gui import ReleaseCapture
-        from win32api import SendMessage
-
-        ReleaseCapture()
-        SendMessage(
-            hwnd,
-            win32con.WM_SYSCOMMAND,
-            win32con.SC_MOVE | win32con.HTCAPTION, 0
-        )
+import ctypes
+import platform
+import time
+
+from PySide6.QtCore import QObject, Signal, Slot, QThread
+
+from .config import DEFAULT_CONFIG, RinConfig, is_win10, is_windows, is_win11, BackdropEffect
+import sys
+import darkdetect
+
+
+def check_darkdetect_support():
+    system = platform.system()
+    if system == "Darwin":
+        mac_ver = platform.mac_ver()[0]
+        major, minor, *_ = map(int, mac_ver.split("."))
+        return (major == 10 and minor >= 14) or major > 10
+
+    elif system == "Windows":
+        return platform.release() >= "10"
+    else:
+        return False
+
+
+ACCENT_STATES = {
+    "acrylic": 3,
+    "mica": 2,
+    "tabbed": 4,
+    "none": 0
+}
+
+ACCENT_SUPPORT = {
+    "acrylic": is_win10(),
+    "mica": is_win11(),
+    "tabbed": is_win10(),
+    "none": True
+}
+
+
+class ACCENT_POLICY(ctypes.Structure):
+    _fields_ = [
+        ("AccentState", ctypes.c_int),
+        ("AccentFlags", ctypes.c_int),
+        ("GradientColor", ctypes.c_int),
+        ("AnimationId", ctypes.c_int),
+    ]
+
+
+class WINDOWCOMPOSITIONATTRIBDATA(ctypes.Structure):
+    _fields_ = [
+        ("Attrib", ctypes.c_int),
+        ("pvData", ctypes.c_void_p),
+        ("cbData", ctypes.c_size_t),
+    ]
+
+
+class ThemeListener(QThread):
+    """
+    监听系统颜色模式
+    """
+    themeChanged = Signal(str)
+
+    def run(self):
+        last_theme = darkdetect.theme()
+        while True:
+            current_theme = darkdetect.theme()
+            if current_theme != last_theme:
+                last_theme = current_theme
+                self.themeChanged.emit(current_theme)
+                print(f"Theme changed: {current_theme}")
+            time.sleep(1)
+
+    def stop(self):
+        self.terminate()
+
+
+class ThemeManager(QObject):
+    themeChanged = Signal(str)
+    backdropChanged = Signal(str)
+    windows = []  # 窗口句柄们（
+    _instance = None
+
+    # DWM 常量保持不变
+    DWMWA_USE_IMMERSIVE_DARK_MODE = 20
+    DWMWA_WINDOW_CORNER_PREFERENCE = 33
+    DWMWA_NCRENDERING_POLICY = 2
+    DWMNCRENDERINGPOLICY_ENABLED = 2
+    DWMWA_SYSTEMBACKDROP_TYPE = 38
+    WCA_ACCENT_POLICY = 19
+
+    # 圆角
+    DWMWCP_DEFAULT = 0
+    DWMWCP_DONOTROUND = 1
+    DWMWCP_ROUND = 2
+    DWMWCP_ROUNDSMALL = 3
+
+    def clean_up(self):
+        """
+        清理资源并停止主题监听。
+        """
+        if self.listener:
+            RinConfig.save_config()
+            print("Save config.")
+            self.listener.stop()
+            self.listener.wait()  # 等待线程结束
+            print("Theme listener stopped.")
+
+    def __new__(cls, *args, **kwargs):
+        """
+        单例管理，共享主题状态
+        :param args:
+        :param kwargs:
+        """
+        if cls._instance is None:
+            cls._instance = super().__new__(cls)
+        return cls._instance
+
+    def __init__(self):
+        if hasattr(self, "_initialized") and self._initialized:
+            return
+        self._initialized = True
+        super().__init__()
+        self.theme_dict = {
+            "Light": 0,
+            "Dark": 1
+        }
+
+        self.listener = None  # 监听线程
+        self.current_theme = DEFAULT_CONFIG["theme"]["current_theme"]  # 当前主题
+        self.is_darkdetect_supported = check_darkdetect_support()
+
+        try:
+            self.current_theme = RinConfig["theme"]["current_theme"]
+        except Exception as e:
+            print(f"Failed to load config because of {e}, using default config")
+
+        # self.hwnd = None  # 窗口句柄
+
+        self.start_listener()
+
+    def start_listener(self):
+        if not self.is_darkdetect_supported:
+            print("darkdetect not supported on this platform")
+            return
+        self.listener = ThemeListener()
+        self.listener.themeChanged.connect(self._handle_system_theme)
+        self.listener.start()
+
+    def set_window(self, window):  # 绑定窗口句柄
+        hwnd = int(window.winId())
+        self.windows.append(hwnd)
+        print(f"Window handle set: {hwnd}")
+
+    def _handle_system_theme(self):
+        if self.current_theme == "Auto":
+            self._update_window_theme()
+            self.themeChanged.emit(self._actual_theme())
+        else:
+            # 保持当前背景效果不变
+            self._update_window_theme()
+
+    @Slot(str)
+    def apply_backdrop_effect(self, effect_type: str):
+        """
+        应用背景效果
+        :param effect_type: str, 背景效果类型（acrylic, mica, tabbed, none）
+        """
+        self._update_window_theme()
+        if not is_windows() or not self.windows:
+            print(f"Cannot apply effect \"{effect_type}\" on this platform")
+            return -2  # 非 windows或未绑定窗口
+        self.backdropChanged.emit(effect_type)
+
+        accent_state = ACCENT_STATES.get(effect_type, 0)
+        if not ACCENT_SUPPORT.get(effect_type, False):
+            print(f"Effect \"{effect_type}\" not supported on this platform")
+            return -1  # 效果不支持
+
+        for hwnd in self.windows:
+            if is_win11():
+                ctypes.windll.dwmapi.DwmSetWindowAttribute(
+                    hwnd,
+                    self.DWMWA_SYSTEMBACKDROP_TYPE,
+                    ctypes.byref(ctypes.c_int(accent_state)),
+                    ctypes.sizeof(ctypes.c_int)
+                )
+            elif is_win10() and effect_type == BackdropEffect.Acrylic.value:
+                self._apply_win10_effect(effect_type, hwnd)
+
+        RinConfig["backdrop_effect"] = effect_type
+        print(
+            f"Applied \"{effect_type.strip().capitalize()}\" effect with "
+            f"{platform.system() + '11' if is_win11() else '10'}"
+        )
+        return 0  # 成功
+
+    def _apply_win10_effect(self, effect_type, hwnd):
+        """
+        应用 Windows 10 背景效果
+        :param effect_type: str, 背景效果类型（acrylic, tabbed(actually blur)
+        """
+        backdrop_color = RinConfig["win10_feat"]["backdrop_dark" if self.is_dark_theme() else "backdrop_light"]
+
+        accent = ACCENT_POLICY()
+        accent.AccentState = ACCENT_STATES[effect_type]
+        accent.AccentFlags = 2
+        accent.GradientColor = backdrop_color
+        data = WINDOWCOMPOSITIONATTRIBDATA()
+        data.Attrib = self.WCA_ACCENT_POLICY
+        data.pvData = ctypes.cast(ctypes.pointer(accent), ctypes.c_void_p)
+        data.cbData = ctypes.sizeof(accent)
+
+        try:
+            set_window_composition = ctypes.windll.user32.SetWindowCompositionAttribute
+            set_window_composition(hwnd, ctypes.byref(data))
+        except Exception as e:
+            print(f"Failed to apply acrylic on Win10: {e}")
+
+    def apply_window_effects(self):  # 启用圆角阴影
+        if sys.platform != "win32" or not self.windows:
+            return
+
+        dwm = ctypes.windll.dwmapi
+
+        # 启用非客户端渲染策略（让窗口边框具备阴影）
+        ncrp = ctypes.c_int(self.DWMNCRENDERINGPOLICY_ENABLED)
+        for hwnd in self.windows:
+            dwm.DwmSetWindowAttribute(
+                hwnd,
+                self.DWMWA_NCRENDERING_POLICY,
+                ctypes.byref(ncrp),
+                ctypes.sizeof(ncrp)
+            )
+
+            # 启用圆角效果
+            corner_preference = ctypes.c_int(self.DWMWCP_ROUND)
+            dwm.DwmSetWindowAttribute(
+                hwnd,
+                self.DWMWA_WINDOW_CORNER_PREFERENCE,
+                ctypes.byref(corner_preference),
+                ctypes.sizeof(corner_preference)
+            )
+        print("Enabled Rounded and Shadows")
+
+    def _update_window_theme(self):  # 更新窗口的颜色模式
+        if sys.platform != "win32" or not self.windows:
+            return
+        actual_theme = self._actual_theme()
+        for hwnd in self.windows:
+            if is_win11():
+                ctypes.windll.dwmapi.DwmSetWindowAttribute(
+                    hwnd,
+                    self.DWMWA_USE_IMMERSIVE_DARK_MODE,
+                    ctypes.byref(ctypes.c_int(self.theme_dict[actual_theme])),
+                    ctypes.sizeof(ctypes.c_int)
+                )
+            elif is_win10() and RinConfig["backdrop_effect"] == BackdropEffect.Acrylic.value:
+                self._apply_win10_effect(RinConfig["backdrop_effect"], hwnd)
+            else:
+                print(f"Cannot apply backdrop on {platform.system()}")
+
+        print(f"Window theme updated to {actual_theme}")
+
+    def is_dark_theme(self):
+        """是否为暗黑主题"""
+        return self._actual_theme() == "Dark"
+
+    def _actual_theme(self):
+        """实际应用的主题"""
+        if self.current_theme == "Auto":
+            return darkdetect.theme() if self.is_darkdetect_supported else "Light"
+        return self.current_theme
+
+    @Slot(str)
+    def toggle_theme(self, theme: str):  # 切换主题
+        if theme not in ["Auto", "Light", "Dark"]:  # 三状态
+            return
+        if self.current_theme != theme:
+            print(f"Switching to '{theme}' theme")
+            self.current_theme = theme
+            RinConfig["theme"]["current_theme"] = theme
+            self._update_window_theme()
+            self.themeChanged.emit(self._actual_theme())
+
+    @Slot(result=str)
+    def get_theme(self):
+        return self._actual_theme()
+
+    @Slot(result=str)
+    def get_theme_name(self):
+        """获取当前主题名称"""
+        return self.current_theme
+
+    @Slot(str)
+    def receive(self, message):
+        print(message)
+
+    @Slot(result=str)
+    def get_backdrop_effect(self):
+        """获取当前背景效果"""
+        return RinConfig["backdrop_effect"]
+
+    @Slot(str)
+    def set_theme_color(self, color):
+        """设置当前主题颜色"""
+        RinConfig["theme_color"] = color
+        RinConfig.save_config()
+
+    @Slot(result=str)
+    def get_theme_color(self):
+        """获取当前主题颜色"""
+        return RinConfig["theme_color"]
+
+    @Slot(QObject, result=int)
+    def getWindowId(self, window):
+        """获取窗口的句柄"""
+        print(f"GetWindowId: {window.winId()}")
+        return int(window.winId())
+
+    @Slot(int)
+    def dragWindowEvent(self, hwnd):
+        """ 在Windows 用原生方法拖动"""
+        if not is_windows() or hwnd not in self.windows:
+            print(
+                f"Use Qt method to drag window on: {platform.system()}"
+                if not is_windows() else f"Invalid window handle: {hwnd}"
+            )
+            return
+
+        import win32con
+        from win32gui import ReleaseCapture
+        from win32api import SendMessage
+
+        ReleaseCapture()
+        SendMessage(
+            hwnd,
+            win32con.WM_SYSCOMMAND,
+            win32con.SC_MOVE | win32con.HTCAPTION, 0
+        )