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

kotonebot/ui/file_host/sensio.py

@@ -1,36 +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}")
+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

@@ -1,54 +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))
-
+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

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

kotonebot/ui/pushkit/image_host.py

@@ -1,87 +1,88 @@
-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")))
+import os
+import tempfile
+from pathlib import Path
+from typing import Sequence
+
+import cv2
+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 或本地图片文件路径
+    """
+    import requests
+    
+    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

@@ -1,13 +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:
-        ...
+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

@@ -1,53 +1,54 @@
-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")])
-
+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")])
+