kotonebot 0.3.1__py3-none-any.whl → 0.5.0__py3-none-any.whl

kotonebot/interop/win/reg.py

@@ -1,37 +1,37 @@
-import winreg
-from typing import Any, Literal
-
-RegKey = Literal["HKLM", "HKCU", "HKCR", "HKU", "HKCC"]
-
-def read_reg(key: RegKey, subkey: str, name: str, *, default: Any = None) -> Any:
-    """
-    读取注册表项的值。
-
-    :param key: 注册表键，例如 "HKLM" (HKEY_LOCAL_MACHINE), "HKCU" (HKEY_CURRENT_USER) 等。
-    :param subkey: 注册表子键的路径。
-    :param name: 要读取的值的名称。
-    :param default: 如果注册表项不存在时返回的默认值。
-    :return: 注册表项的值，如果不存在则返回默认值。
-    """
-    try:
-        hkey = {
-            "HKLM": winreg.HKEY_LOCAL_MACHINE,
-            "HKCU": winreg.HKEY_CURRENT_USER,
-            "HKCR": winreg.HKEY_CLASSES_ROOT,
-            "HKU": winreg.HKEY_USERS,
-            "HKCC": winreg.HKEY_CURRENT_CONFIG,
-        }[key]
-    except KeyError:
-        raise ValueError(f"Invalid key: {key}")
-
-    try:
-        with winreg.OpenKey(hkey, subkey) as key_handle:
-            value, _ = winreg.QueryValueEx(key_handle, name)
-            return value
-    except FileNotFoundError:
-        return default
-    except OSError as e:
-        if e.winerror == 2:  # 注册表项不存在
-            return default
-        else:
-            raise  # 其他 OSError 异常，例如权限问题，重新抛出
+import winreg
+from typing import Any, Literal
+
+RegKey = Literal["HKLM", "HKCU", "HKCR", "HKU", "HKCC"]
+
+def read_reg(key: RegKey, subkey: str, name: str, *, default: Any = None) -> Any:
+    """
+    读取注册表项的值。
+
+    :param key: 注册表键，例如 "HKLM" (HKEY_LOCAL_MACHINE), "HKCU" (HKEY_CURRENT_USER) 等。
+    :param subkey: 注册表子键的路径。
+    :param name: 要读取的值的名称。
+    :param default: 如果注册表项不存在时返回的默认值。
+    :return: 注册表项的值，如果不存在则返回默认值。
+    """
+    try:
+        hkey = {
+            "HKLM": winreg.HKEY_LOCAL_MACHINE,
+            "HKCU": winreg.HKEY_CURRENT_USER,
+            "HKCR": winreg.HKEY_CLASSES_ROOT,
+            "HKU": winreg.HKEY_USERS,
+            "HKCC": winreg.HKEY_CURRENT_CONFIG,
+        }[key]
+    except KeyError:
+        raise ValueError(f"Invalid key: {key}")
+
+    try:
+        with winreg.OpenKey(hkey, subkey) as key_handle:
+            value, _ = winreg.QueryValueEx(key_handle, name)
+            return value
+    except FileNotFoundError:
+        return default
+    except OSError as e:
+        if e.winerror == 2:  # 注册表项不存在
+            return default
+        else:
+            raise  # 其他 OSError 异常，例如权限问题，重新抛出

kotonebot/interop/win/shortcut.py

@@ -1,43 +1,43 @@
-import os
-import typing
-
-import pythoncom
-from win32comext.shell import shell, shellcon
-
-def create_shortcut(target_file: str, target_args: str, link_file: str | None, *,
-                    link_name: str | None = None,
-                    icon_path: str, description: str = ""):
-    """
-    Creates a shortcut.
-
-    :param target_file: The path to the target file.
-    :param target_args: The arguments for the target file.
-    :param link_file: The path to the shortcut file. If None, creates on the desktop.
-    :param link_name: The name of the shortcut file. If None, it is derived from the target file name.
-                        Only used when link_file is None.
-    :param icon_path: The path to the icon file.
-    :param description: The description of the shortcut.
-    """
-    pythoncom.CoInitialize()
-    if link_file is None:
-        desktop_path = shell.SHGetFolderPath(0, shellcon.CSIDL_DESKTOP, 0, 0)
-        if link_name is None:
-            link_name = os.path.splitext(os.path.basename(target_file))[0]
-        link_file = os.path.join(desktop_path, f"{link_name}.lnk")
-
-    shortcut = pythoncom.CoCreateInstance(
-        shell.CLSID_ShellLink,
-        None,
-        pythoncom.CLSCTX_INPROC_SERVER,
-        shell.IID_IShellLink
-    )
-    # TODO: 下面这些方法都没有 typing，会报错，需要一种方法加入 typing
-    shortcut.SetPath(target_file) # type: ignore
-    shortcut.SetArguments(target_args) # type: ignore
-    shortcut.SetDescription(description) # type: ignore
-    shortcut.SetIconLocation(icon_path, 0) # type: ignore
-
-    persist_file = shortcut.QueryInterface(pythoncom.IID_IPersistFile)
-    persist_file.Save(link_file, 0) # type: ignore
-    pythoncom.CoUninitialize()
-
+import os
+import typing
+
+import pythoncom
+from win32comext.shell import shell, shellcon
+
+def create_shortcut(target_file: str, target_args: str, link_file: str | None, *,
+                    link_name: str | None = None,
+                    icon_path: str, description: str = ""):
+    """
+    Creates a shortcut.
+
+    :param target_file: The path to the target file.
+    :param target_args: The arguments for the target file.
+    :param link_file: The path to the shortcut file. If None, creates on the desktop.
+    :param link_name: The name of the shortcut file. If None, it is derived from the target file name.
+                        Only used when link_file is None.
+    :param icon_path: The path to the icon file.
+    :param description: The description of the shortcut.
+    """
+    pythoncom.CoInitialize()
+    if link_file is None:
+        desktop_path = shell.SHGetFolderPath(0, shellcon.CSIDL_DESKTOP, 0, 0)
+        if link_name is None:
+            link_name = os.path.splitext(os.path.basename(target_file))[0]
+        link_file = os.path.join(desktop_path, f"{link_name}.lnk")
+
+    shortcut = pythoncom.CoCreateInstance(
+        shell.CLSID_ShellLink,
+        None,
+        pythoncom.CLSCTX_INPROC_SERVER,
+        shell.IID_IShellLink
+    )
+    # TODO: 下面这些方法都没有 typing，会报错，需要一种方法加入 typing
+    shortcut.SetPath(target_file) # type: ignore
+    shortcut.SetArguments(target_args) # type: ignore
+    shortcut.SetDescription(description) # type: ignore
+    shortcut.SetIconLocation(icon_path, 0) # type: ignore
+
+    persist_file = shortcut.QueryInterface(pythoncom.IID_IPersistFile)
+    persist_file.Save(link_file, 0) # type: ignore
+    pythoncom.CoUninitialize()
+