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

kotonebot/ui/pushkit/wxpusher.py

@@ -1,54 +1,54 @@
-import os
-import json
-from typing import Sequence
-from cv2.typing import MatLike
-from dotenv import dotenv_values
-
-from .image_host import upload
-from .protocol import PushkitProtocol
-
-config = dotenv_values(".env")
-
-class Wxpusher(PushkitProtocol):
-    def __init__(self, app_token: str | None = None, uid: str | None = None):
-        self.app_token = app_token or config["WXPUSHER_APP_TOKEN"]
-        self.uid = uid or config["WXPUSHER_UID"]
-
-    def push(self, title: str, message: str, *, images: Sequence[str | MatLike] | None = None) -> None:
-        import requests
-        
-        summary = title
-        content = message
-
-        if images:
-            image_urls = upload(images)
-            img_md = "\n".join([f"![{img_url}]({img_url})" for img_url in image_urls])
-            content = content + "\n" + img_md
-
-        data = {
-            "appToken": self.app_token,
-            "uid": self.uid,
-            "summary": summary,
-            "content": content,
-            "contentType": 3, # 1: 文本 2: HTML 3: Markdown
-            "uids": [self.uid],
-            "verifyPay": False,
-            "verifyPayType": 0
-        }
-
-        response = requests.post(
-            "http://wxpusher.zjiecode.com/api/send/message",
-            json=data
-        )
-        result = response.json()
-        
-        if result["code"] != 1000 or not result["success"]:
-            raise RuntimeError(f"推送失败: {result['msg']}")
-
-# TODO: 极简推送 https://wxpusher.zjiecode.com/docs/#/?id=spt
-        
-if __name__ == "__main__":
-    import cv2
-    wxpusher = Wxpusher()
-    wxpusher.push("测试图片", "测试图片", images=[cv2.imread("res/sprites/jp/common/button_close.png")])
-
+import os
+import json
+from typing import Sequence
+from cv2.typing import MatLike
+from dotenv import dotenv_values
+
+from .image_host import upload
+from .protocol import PushkitProtocol
+
+config = dotenv_values(".env")
+
+class Wxpusher(PushkitProtocol):
+    def __init__(self, app_token: str | None = None, uid: str | None = None):
+        self.app_token = app_token or config["WXPUSHER_APP_TOKEN"]
+        self.uid = uid or config["WXPUSHER_UID"]
+
+    def push(self, title: str, message: str, *, images: Sequence[str | MatLike] | None = None) -> None:
+        import requests
+        
+        summary = title
+        content = message
+
+        if images:
+            image_urls = upload(images)
+            img_md = "\n".join([f"![{img_url}]({img_url})" for img_url in image_urls])
+            content = content + "\n" + img_md
+
+        data = {
+            "appToken": self.app_token,
+            "uid": self.uid,
+            "summary": summary,
+            "content": content,
+            "contentType": 3, # 1: 文本 2: HTML 3: Markdown
+            "uids": [self.uid],
+            "verifyPay": False,
+            "verifyPayType": 0
+        }
+
+        response = requests.post(
+            "http://wxpusher.zjiecode.com/api/send/message",
+            json=data
+        )
+        result = response.json()
+        
+        if result["code"] != 1000 or not result["success"]:
+            raise RuntimeError(f"推送失败: {result['msg']}")
+
+# TODO: 极简推送 https://wxpusher.zjiecode.com/docs/#/?id=spt
+        
+if __name__ == "__main__":
+    import cv2
+    wxpusher = Wxpusher()
+    wxpusher.push("测试图片", "测试图片", images=[cv2.imread("res/sprites/jp/common/button_close.png")])
+

kotonebot/ui/user.py

@@ -1,149 +1,149 @@
-"""消息框、通知、推送等 UI 相关函数"""
-import os
-import time
-
-import cv2
-from cv2.typing import MatLike
-from kotonebot.util import is_windows
-if is_windows():
-    from win11toast import toast
-else:
-    def toast(title: str, message: str | None = None, buttons: list[str] | None = None):
-        raise ImportError('toast notification is only available on Windows')
-
-from .pushkit import Wxpusher
-from .. import logging
-
-logger = logging.getLogger(__name__)
-
-def retry(func):
-    """
-    装饰器：当函数发生 ConnectionResetError 时自动重试三次
-    """
-    def wrapper(*args, **kwargs):
-        for i in range(3):
-            try:
-                return func(*args, **kwargs)
-            except ConnectionResetError:
-                if i == 2:  # 最后一次重试失败
-                    raise
-                logger.warning(f'ConnectionResetError raised when calling {func}, retrying {i+1}/{3}')
-                continue
-    return wrapper
-
-def _save_local(
-    title: str,
-    message: str,
-    images: list[MatLike] | None = None
-):
-    """
-    保存消息到本地
-    """
-    if not os.path.exists('messages'):
-        os.makedirs('messages')
-    file_name = f'messages/{time.strftime("%Y-%m-%d_%H-%M-%S")}'
-    with open(file_name + '.txt', 'w', encoding='utf-8') as f:
-        logger.verbose('saving message to local: %s', file_name + '.txt')
-        f.write(message)
-    if images is not None:
-        for i, image in enumerate(images):
-            logger.verbose('saving image to local: %s', f'{file_name}_{i}.png')
-            cv2.imwrite(f'{file_name}_{i}.png', image)
-
-@retry
-def push(
-    title: str,
-    message: str | None = None,
-    *,
-    images: list[MatLike] | None = None
-):
-    """
-    推送消息
-    """
-    message = message or ''
-    try:
-        logger.verbose('pushing to wxpusher: %s', message)
-        wxpusher = Wxpusher()
-        wxpusher.push(title, message, images=images)
-    except Exception as e:
-        logger.warning('push remote message failed: %s', e)
-        _save_local(title, message, images)
-
-def _show_toast(title: str, message: str | None = None, buttons: list[str] | None = None):
-    """
-    统一的 Toast 通知函数
-
-    :param title: 通知标题
-    :param message: 通知消息内容
-    :param buttons: 按钮列表，如果提供则显示带按钮的通知
-    """
-    try:
-        if buttons:
-            logger.verbose('showing toast notification with buttons: %s - %s', title, message or '')
-            toast(title, message or '', buttons=buttons)
-        else:
-            # 如果没有 message，只显示 title
-            if message:
-                logger.verbose('showing toast notification: %s - %s', title, message)
-                toast(title, message)
-            else:
-                logger.verbose('showing toast notification: %s', title)
-                toast(title)
-    except Exception as e:
-        logger.warning('toast notification failed: %s', e)
-
-def ask(
-    question: str,
-    options: list[tuple[str, str]],
-    *,
-    timeout: float = -1,
-) -> str:
-    """
-    询问用户
-    """
-    # 将选项转换为按钮列表
-    buttons = [option[1] for option in options]
-    _show_toast("琴音小助手询问", question, buttons=buttons)
-    raise NotImplementedError
-
-def info(
-    title: str,
-    message: str | None = None,
-    images: list[MatLike] | None = None,
-    *,
-    once: bool = False
-):
-    logger.info('user.info: %s', message)
-    push('KAA：' + title, message, images=images)
-    _show_toast('KAA：' + title, message)
-
-def warning(
-    title: str,
-    message: str | None = None,
-    images: list[MatLike] | None = None,
-    *,
-    once: bool = False
-):
-    """
-    警告信息。
-
-    :param message: 消息内容
-    :param once: 每次运行是否只显示一次。
-    """
-    logger.warning('user.warning: %s', message)
-    push("琴音小助手警告：" + title, message, images=images)
-    _show_toast("琴音小助手警告：" + title, message)
-
-def error(
-    title: str,
-    message: str | None = None,
-    images: list[MatLike] | None = None,
-    *,
-    once: bool = False
-):
-    """
-    错误信息。
-    """
-    logger.error('user.error: %s', message)
-    push("琴音小助手错误：" + title, message, images=images)
+"""消息框、通知、推送等 UI 相关函数"""
+import os
+import time
+
+import cv2
+from cv2.typing import MatLike
+from kotonebot.util import is_windows
+if is_windows():
+    from win11toast import toast
+else:
+    def toast(title: str, message: str | None = None, buttons: list[str] | None = None):
+        raise ImportError('toast notification is only available on Windows')
+
+from .pushkit import Wxpusher
+from .. import logging
+
+logger = logging.getLogger(__name__)
+
+def retry(func):
+    """
+    装饰器：当函数发生 ConnectionResetError 时自动重试三次
+    """
+    def wrapper(*args, **kwargs):
+        for i in range(3):
+            try:
+                return func(*args, **kwargs)
+            except ConnectionResetError:
+                if i == 2:  # 最后一次重试失败
+                    raise
+                logger.warning(f'ConnectionResetError raised when calling {func}, retrying {i+1}/{3}')
+                continue
+    return wrapper
+
+def _save_local(
+    title: str,
+    message: str,
+    images: list[MatLike] | None = None
+):
+    """
+    保存消息到本地
+    """
+    if not os.path.exists('messages'):
+        os.makedirs('messages')
+    file_name = f'messages/{time.strftime("%Y-%m-%d_%H-%M-%S")}'
+    with open(file_name + '.txt', 'w', encoding='utf-8') as f:
+        logger.verbose('saving message to local: %s', file_name + '.txt')
+        f.write(message)
+    if images is not None:
+        for i, image in enumerate(images):
+            logger.verbose('saving image to local: %s', f'{file_name}_{i}.png')
+            cv2.imwrite(f'{file_name}_{i}.png', image)
+
+@retry
+def push(
+    title: str,
+    message: str | None = None,
+    *,
+    images: list[MatLike] | None = None
+):
+    """
+    推送消息
+    """
+    message = message or ''
+    try:
+        logger.verbose('pushing to wxpusher: %s', message)
+        wxpusher = Wxpusher()
+        wxpusher.push(title, message, images=images)
+    except Exception as e:
+        logger.warning('push remote message failed: %s', e)
+        _save_local(title, message, images)
+
+def _show_toast(title: str, message: str | None = None, buttons: list[str] | None = None):
+    """
+    统一的 Toast 通知函数
+
+    :param title: 通知标题
+    :param message: 通知消息内容
+    :param buttons: 按钮列表，如果提供则显示带按钮的通知
+    """
+    try:
+        if buttons:
+            logger.verbose('showing toast notification with buttons: %s - %s', title, message or '')
+            toast(title, message or '', buttons=buttons)
+        else:
+            # 如果没有 message，只显示 title
+            if message:
+                logger.verbose('showing toast notification: %s - %s', title, message)
+                toast(title, message)
+            else:
+                logger.verbose('showing toast notification: %s', title)
+                toast(title)
+    except Exception as e:
+        logger.warning('toast notification failed: %s', e)
+
+def ask(
+    question: str,
+    options: list[tuple[str, str]],
+    *,
+    timeout: float = -1,
+) -> str:
+    """
+    询问用户
+    """
+    # 将选项转换为按钮列表
+    buttons = [option[1] for option in options]
+    _show_toast("琴音小助手询问", question, buttons=buttons)
+    raise NotImplementedError
+
+def info(
+    title: str,
+    message: str | None = None,
+    images: list[MatLike] | None = None,
+    *,
+    once: bool = False
+):
+    logger.info('user.info: %s', message)
+    push('KAA：' + title, message, images=images)
+    _show_toast('KAA：' + title, message)
+
+def warning(
+    title: str,
+    message: str | None = None,
+    images: list[MatLike] | None = None,
+    *,
+    once: bool = False
+):
+    """
+    警告信息。
+
+    :param message: 消息内容
+    :param once: 每次运行是否只显示一次。
+    """
+    logger.warning('user.warning: %s', message)
+    push("琴音小助手警告：" + title, message, images=images)
+    _show_toast("琴音小助手警告：" + title, message)
+
+def error(
+    title: str,
+    message: str | None = None,
+    images: list[MatLike] | None = None,
+    *,
+    once: bool = False
+):
+    """
+    错误信息。
+    """
+    logger.error('user.error: %s', message)
+    push("琴音小助手错误：" + title, message, images=images)
     _show_toast("琴音小助手错误：" + title, message)