kotonebot 0.1.0__py3-none-any.whl

kotonebot/ui/file_host/sensio.py

@@ -0,0 +1,36 @@
+import requests
+import os
+
+def upload(file_path: str) -> str:
+    """
+    上传文件到 paste.sensio.no
+    
+    Args:
+        file_path: 要上传的文件路径
+    
+    Returns:
+        str: 上传后的 URL
+    """
+    url = 'https://paste.sensio.no/'
+    headers = {
+        'accept': 'text/plain',
+        'User-Agent': 'KAA',
+        'x-uuid': ''
+    }
+    
+    files = {
+        'file': (os.path.basename(file_path), open(file_path, 'rb'))
+    }
+        
+    response = requests.post(url, files=files, headers=headers, allow_redirects=False)
+    
+    if response.status_code != 200:
+        raise Exception(f"Upload failed with status code {response.status_code}")
+    
+    return response.text.strip()
+
+if __name__ == "__main__":
+    test_file = "version"
+    if os.path.exists(test_file):
+        result = upload(test_file)
+        print(f"Upload result: {result}")

kotonebot/ui/file_host/tmp_send.py

@@ -0,0 +1,54 @@
+import requests
+import os
+
+def upload(file_path: str) -> str:
+    url = 'https://tmpsend.com/upload'
+    headers = {
+        'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/133.0.0.0 Safari/537.36',
+        'Referer': 'https://tmpsend.com/',
+    }
+    
+    file_name = os.path.basename(file_path)
+    file_size = os.path.getsize(file_path)
+    
+    # 第一次请求：添加文件信息
+    files = {
+        'action': (None, 'add'),
+        'name': (None, file_name),
+        'size': (None, str(file_size)),
+        'file': (file_name, open(file_path, 'rb'))
+    }
+    
+    response = requests.post(url, headers=headers, files=files)
+    if response.status_code != 200:
+        raise Exception(f"Upload failed with status code {response.status_code}")
+    
+    result = response.json()
+    if result.get('hasError'):
+        raise Exception(result.get('error'))
+    
+    file_id = result.get('id')
+    if not file_id:
+        raise Exception("Failed to get file ID")
+    
+    # 第二次请求：上传实际文件
+    upload_files = {
+        'action': (None, 'upload'),
+        'id': (None, file_id),
+        'name': (None, file_name),
+        'size': (None, str(file_size)),
+        'start': (None, '0'),
+        'end': (None, str(file_size)),
+        'data': (file_name, open(file_path, 'rb'), 'application/octet-stream')
+    }
+    
+    upload_response = requests.post(url, headers=headers, files=upload_files)
+    if upload_response.status_code != 200:
+        raise Exception(f"File upload failed with status code {upload_response.status_code}")
+    
+    return 'https://tmpsend.com/' + file_id
+
+if __name__ == "__main__":
+    file_path = r"主题1.thmx"
+    print(upload(file_path))
+

kotonebot/ui/pushkit/__init__.py

@@ -0,0 +1,3 @@
+from .protocol import PushkitProtocol
+from .wxpusher import Wxpusher
+

kotonebot/ui/pushkit/image_host.py

@@ -0,0 +1,87 @@
+import os
+import tempfile
+from pathlib import Path
+from typing import Sequence
+
+import cv2
+import requests
+import numpy as np
+from cv2.typing import MatLike
+from dotenv import load_dotenv
+
+load_dotenv()
+
+def _save_temp_image(image: MatLike) -> Path:
+    """将OpenCV图片保存为临时文件"""
+    temp_file = Path(tempfile.mktemp(suffix='.jpg'))
+    cv2.imwrite(str(temp_file), image)
+    return temp_file
+
+def _upload_single(image: MatLike | str) -> str:
+    """
+    上传单张图片到freeimage.host
+    
+    :param image: OpenCV MatLike 或本地图片文件路径
+    """
+    api_url = 'https://freeimage.host/api/1/upload'
+    api_key = os.getenv('FREEIMAGEHOST_KEY')
+    
+    if not api_key:
+        raise RuntimeError('Environment variable FREEIMAGEHOST_KEY is not set')
+    
+    # 处理输入
+    temp_file = None
+    if isinstance(image, str):
+        # 本地文件路径
+        files = {'source': open(image, 'rb')}
+    else:
+        # 保存OpenCV图片为临时文件
+        temp_file = _save_temp_image(image)
+        files = {'source': open(temp_file, 'rb')}
+    
+    data = {
+        'key': api_key,
+        'action': 'upload',
+        'format': 'json'
+    }
+    
+    try:
+        # 发送POST请求
+        response = requests.post(api_url, data=data, files=files)
+        
+        if response.status_code != 200:
+            raise RuntimeError(f'Upload failed: HTTP {response.status_code}')
+            
+        result = response.json()
+        
+        if result['status_code'] != 200:
+            raise RuntimeError(f'Upload failed: API {result["status_txt"]}')
+            
+        return result['image']['url']
+        
+    finally:
+        # 清理临时文件
+        files['source'].close()
+        if temp_file and temp_file.exists():
+            temp_file.unlink()
+
+def upload(images: MatLike | str | Sequence[MatLike | str]) -> list[str]:
+    """上传一张或多张图片到freeimage.host
+    
+    Args:
+        images: 单张图片或图片列表。每个图片可以是OpenCV图片对象或本地图片文件路径
+        
+    Returns:
+        上传后的图片URL列表
+    """
+    if isinstance(images, (str, np.ndarray)):
+        _images = [images]
+    elif isinstance(images, Sequence):
+        _images = [img for img in images]
+    else:
+        raise ValueError("Invalid input type")
+        
+    return [_upload_single(img) for img in _images]
+
+if __name__ == "__main__":
+    print(upload(cv2.imread("res/sprites/jp/common/button_close.png")))

kotonebot/ui/pushkit/protocol.py

@@ -0,0 +1,13 @@
+from typing import Protocol
+
+from cv2.typing import MatLike
+
+class PushkitProtocol(Protocol):
+    def push(
+        self,
+        title: str,
+        message: str,
+        *,
+        images: list[str | MatLike] | None = None,
+    ) -> None:
+        ...

kotonebot/ui/pushkit/wxpusher.py

@@ -0,0 +1,53 @@
+import os
+import json
+from typing import Sequence
+import requests
+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:
+        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

@@ -0,0 +1,144 @@
+"""消息框、通知、推送等 UI 相关函数"""
+import os
+import time
+
+import cv2
+from cv2.typing import MatLike
+from win11toast import toast
+
+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)