onebot-protocol 0.0.2__tar.gz

onebot_protocol-0.0.2/.gitignore

@@ -0,0 +1 @@
+dist/

onebot_protocol-0.0.2/PKG-INFO

@@ -0,0 +1,11 @@
+Metadata-Version: 2.4
+Name: onebot-protocol
+Version: 0.0.2
+Summary: OneBot 通信协议，定义消息收发的公共数据结构
+Project-URL: Repository, https://github.com/XiaoHui2023/onebot-protocol
+License-Expression: MIT
+Requires-Python: >=3.10
+Requires-Dist: pydantic>=2.0
+Description-Content-Type: text/markdown
+
+# onebot-protocol

onebot_protocol-0.0.2/README.md

@@ -0,0 +1 @@
+# onebot-protocol

onebot_protocol-0.0.2/onebot_protocol/__init__.py

@@ -0,0 +1,27 @@
+from .models import (
+    MessagePayload,
+    MessageSegment,
+    TextMessageSegment,
+    MentionMessageSegment,
+    MentionAllMessageSegment,
+    ImageMessageSegment,
+    VoiceMessageSegment,
+    AudioMessageSegment,
+    VideoMessageSegment,
+    FileMessageSegment,
+    LocationMessageSegment,
+)
+
+__all__ = [
+    "MessagePayload",
+    "MessageSegment",
+    "TextMessageSegment",
+    "MentionMessageSegment",
+    "MentionAllMessageSegment",
+    "ImageMessageSegment",
+    "VoiceMessageSegment",
+    "AudioMessageSegment",
+    "VideoMessageSegment",
+    "FileMessageSegment",
+    "LocationMessageSegment",
+]

onebot_protocol-0.0.2/onebot_protocol/models.py

@@ -0,0 +1,110 @@
+from pydantic import BaseModel, Field
+from typing import Literal, List, Annotated, Union
+
+class TextSegmentData(BaseModel):
+    """纯文本"""
+    text: str
+
+class MentionSegmentData(BaseModel):
+    """@某人"""
+    user_id: str
+
+class MentionAllSegmentData(BaseModel):
+    """@所有人"""
+    pass
+
+class ImageSegmentData(BaseModel):
+    """图片"""
+    file_id: str
+
+class VoiceSegmentData(BaseModel):
+    """语音"""
+    file_id: str
+
+class AudioSegmentData(BaseModel):
+    """音频"""
+    file_id: str
+
+class VideoSegmentData(BaseModel):
+    """视频"""
+    file_id: str
+
+class FileSegmentData(BaseModel):
+    """文件"""
+    file_id: str
+
+class LocationSegmentData(BaseModel):
+    """位置"""
+    latitude: float
+    longitude: float
+    title: str
+    content: str
+
+class ReplySegmentData(BaseModel):
+    """回复"""
+    message_id: str
+    user_id: str | None = None  # 可选
+
+class TextMessageSegment(BaseModel):
+    type: Literal["text"] = "text"
+    data: TextSegmentData
+
+class MentionMessageSegment(BaseModel):
+    type: Literal["mention"] = "mention"
+    data: MentionSegmentData
+
+class MentionAllMessageSegment(BaseModel):
+    type: Literal["mention_all"] = "mention_all"
+    data: MentionAllSegmentData = Field(default_factory=MentionAllSegmentData)
+
+class ImageMessageSegment(BaseModel):
+    type: Literal["image"] = "image"
+    data: ImageSegmentData
+
+class VoiceMessageSegment(BaseModel):
+    type: Literal["voice"] = "voice"
+    data: VoiceSegmentData
+
+class AudioMessageSegment(BaseModel):
+    type: Literal["audio"] = "audio"
+    data: AudioSegmentData
+
+class VideoMessageSegment(BaseModel):
+    type: Literal["video"] = "video"
+    data: VideoSegmentData
+
+class FileMessageSegment(BaseModel):
+    type: Literal["file"] = "file"
+    data: FileSegmentData
+
+class LocationMessageSegment(BaseModel):
+    type: Literal["location"] = "location"
+    data: LocationSegmentData
+
+class ReplyMessageSegment(BaseModel):
+    type: Literal["reply"] = "reply"
+    data: ReplySegmentData
+
+MessageSegment = Annotated[
+    Union[
+        TextMessageSegment,
+        MentionMessageSegment,
+        MentionAllMessageSegment,
+        ImageMessageSegment,
+        VoiceMessageSegment,
+        AudioMessageSegment,
+        VideoMessageSegment,
+        FileMessageSegment,
+        LocationMessageSegment,
+        ReplyMessageSegment,
+    ],
+    Field(discriminator="type"),
+]
+
+class MessagePayload(BaseModel):
+    """消息载体"""
+    source_type: Literal['group', 'private'] = Field(..., description="消息来源类型")
+    session_id: str = Field(..., description="会话 ID")
+    bot_id: str = Field(..., description="机器人 ID")
+    user_id: str = Field(..., description="用户 ID")
+    messages: List[MessageSegment] = Field(..., description="消息列表")

onebot_protocol-0.0.2/publish.py

@@ -0,0 +1,60 @@
+"""
+PyPI 一键发布脚本
+自动递增补丁版本号并发布到 PyPI
+用法: python publish.py
+"""
+
+import re
+import subprocess
+import sys
+import shutil
+from pathlib import Path
+
+SCRIPT_DIR = Path(__file__).resolve().parent
+PYPROJECT = SCRIPT_DIR / "pyproject.toml"
+DIST_DIR = SCRIPT_DIR / "dist"
+
+
+def run(cmd: str):
+    print(f"\n>>> {cmd}")
+    result = subprocess.run(cmd, shell=True, cwd=SCRIPT_DIR)
+    if result.returncode != 0:
+        print(f"\n命令失败 (exit code: {result.returncode})")
+        sys.exit(result.returncode)
+
+
+def bump_version() -> str:
+    content = PYPROJECT.read_text(encoding="utf-8")
+    match = re.search(r'^version\s*=\s*"(\d+)\.(\d+)\.(\d+)"', content, re.MULTILINE)
+    if not match:
+        print("无法在 pyproject.toml 中找到版本号")
+        sys.exit(1)
+
+    major, minor, patch = int(match.group(1)), int(match.group(2)), int(match.group(3))
+    old_version = f"{major}.{minor}.{patch}"
+    new_version = f"{major}.{minor}.{patch + 1}"
+
+    new_content = content.replace(f'version = "{old_version}"', f'version = "{new_version}"')
+    PYPROJECT.write_text(new_content, encoding="utf-8")
+    return new_version
+
+
+def main():
+    new_version = bump_version()
+    print(f"版本号已更新为: {new_version}")
+
+    if DIST_DIR.exists():
+        print(f"清理 {DIST_DIR} ...")
+        shutil.rmtree(DIST_DIR)
+
+    print("开始构建...")
+    run("python -m build")
+
+    print("上传到 PyPI...")
+    run("python -m twine upload dist/*")
+
+    print("\n发布完成！")
+
+
+if __name__ == "__main__":
+    main()

onebot_protocol-0.0.2/pyproject.toml

@@ -0,0 +1,20 @@
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[project]
+name = "onebot-protocol"
+version = "0.0.2"
+description = "OneBot 通信协议，定义消息收发的公共数据结构"
+readme = "README.md"
+requires-python = ">=3.10"
+license = "MIT"
+dependencies = [
+    "pydantic>=2.0",
+]
+
+[project.urls]
+Repository = "https://github.com/XiaoHui2023/onebot-protocol"
+
+[tool.hatch.build.targets.wheel]
+packages = ["onebot_protocol"]