From 85ca0e758954dd91e2a2c20a3b0848836fcbe259 Mon Sep 17 00:00:00 2001
From: JianFeeeee <109188060+JianFeeeee@users.noreply.github.com>
Date: Thu, 14 Aug 2025 16:01:37 +0800
Subject: [PATCH] new file:   dependence.py new file:   package.py new file:  
 packup.bat new file:   packup.sh new file:   requirements.txt new file:  
 scripts/__init__.py new file:   scripts/file_store_api.py new file:  
 src/__init__.py new file:   src/config.toml new file:  
 src/modules/__init__.py new file:   src/modules/plugin_modules.py new file:  
 src/modules/user_module.py new file:   src/process.py new file:   test.py

---
 dependence.py                 |  87 +++++++++++++++++++++
 package.py                    |  62 +++++++++++++++
 packup.bat                    |  21 +++++
 packup.sh                     |  18 +++++
 requirements.txt              |   1 +
 scripts/__init__.py           |   0
 scripts/file_store_api.py     |  65 ++++++++++++++++
 src/__init__.py               |   0
 src/config.toml               |   4 +
 src/modules/__init__.py       |   0
 src/modules/plugin_modules.py |  55 +++++++++++++
 src/modules/user_module.py    |  42 ++++++++++
 src/process.py                |  63 +++++++++++++++
 test.py                       | 141 ++++++++++++++++++++++++++++++++++
 14 files changed, 559 insertions(+)
 create mode 100644 dependence.py
 create mode 100644 package.py
 create mode 100644 packup.bat
 create mode 100644 packup.sh
 create mode 100644 requirements.txt
 create mode 100644 scripts/__init__.py
 create mode 100644 scripts/file_store_api.py
 create mode 100644 src/__init__.py
 create mode 100644 src/config.toml
 create mode 100644 src/modules/__init__.py
 create mode 100644 src/modules/plugin_modules.py
 create mode 100644 src/modules/user_module.py
 create mode 100644 src/process.py
 create mode 100644 test.py

diff --git a/dependence.py b/dependence.py
new file mode 100644
index 0000000..1f02c32
--- /dev/null
+++ b/dependence.py
@@ -0,0 +1,87 @@
+#!/usr/bin/env python3
+import os
+import sys
+import shutil
+import subprocess
+from pathlib import Path
+
+def check_requirements_file(requirements_file="requirements.txt"):
+    """检查 requirements.txt 是否存在"""
+    if not os.path.exists(requirements_file):
+        print(f"❌ 错误：未找到 {requirements_file}！请确保它在当前目录。")
+        print(f"👉 生成 requirements.txt 的方法：`pip freeze > requirements.txt`")
+        sys.exit(1)
+
+def install_deps_to_folder(
+    requirements_file="requirements.txt",
+    target_dir="package",
+    upgrade=True,
+    no_deps=False
+):
+    """
+    安装依赖到目标文件夹
+    :param requirements_file: 依赖文件路径
+    :param target_dir: 目标目录（默认 package）
+    :param upgrade: 是否更新已安装的包
+    :param no_deps: 是否跳过依赖包（仅安装直接依赖）
+    """
+    target_path = Path(target_dir).resolve()
+    
+    # 1. 如果目标文件夹已存在，提醒用户确认覆盖
+    if target_path.exists():
+        print(f"⚠️ 警告：目标文件夹 {target_path} 已存在！")
+        choice = input("是否删除并重建？(y/N) ").strip().lower()
+        if choice != "y":
+            print("⏹ 用户取消操作")
+            return False
+        shutil.rmtree(target_path)
+    
+    target_path.mkdir(parents=True)
+
+    # 2. 构造 pip 安装命令
+    pip_cmd = [
+        sys.executable, "-m", "pip", "install",
+        "-r", requirements_file,
+        "--target", str(target_path),
+    ]
+    
+    if upgrade:
+        pip_cmd.append("--upgrade")
+    if no_deps:
+        pip_cmd.append("--no-deps")
+    
+    # 3. 执行安装
+    print(f"\n📦 正在安装依赖到 {target_path} ...")
+    result = subprocess.run(pip_cmd, capture_output=True, text=True)
+    
+    if result.returncode != 0:
+        print("\n❌ 安装失败！错误信息：")
+        print(result.stderr)
+        return False
+    
+    # 4. 成功时打印提示
+    print("\n✅ 安装成功！依赖已保存到：")
+    print(f"   → {target_path}\n")
+    # 5. 打印后续使用说明
+    return True
+
+def main():
+    check_requirements_file()
+    dir  = "src"
+    dir = os.path.join(dir,"packages")
+    install_success = install_deps_to_folder(
+        requirements_file="requirements.txt",
+        target_dir=dir,
+        upgrade=True,
+        no_deps=False,  # 设为 True 仅安装直接依赖
+    )
+    
+    if install_success:
+        # 检查是否安装了 pip 依赖
+        package_size = sum(f.stat().size for f in Path("package").glob("**/*") if f.is_file())
+        print(f"📂 安装大小: {package_size / 1024 / 1024:.2f} MB")
+    else:
+        print("\n❌ 安装失败！请检查报错信息。")
+
+if __name__ == "__main__":
+    main()
diff --git a/package.py b/package.py
new file mode 100644
index 0000000..d1a32e2
--- /dev/null
+++ b/package.py
@@ -0,0 +1,62 @@
+import os
+import re
+import shutil
+import ast
+from pathlib import Path
+
+def find_plugin_class(process_file: str) -> str:
+    """从process.py中找到继承BasePlugin的类名"""
+    with open(process_file, "r", encoding="utf-8") as f:
+        content = f.read()
+    
+    # 使用AST解析Python代码，比正则更可靠
+    tree = ast.parse(content)
+    
+    for node in ast.walk(tree):
+        if isinstance(node, ast.ClassDef):
+            # 检查是否继承BasePlugin
+            for base in node.bases:
+                if isinstance(base, ast.Name) and base.id == "BasePlugin":
+                    return node.name
+    
+    raise ValueError("❌ 未找到继承自BasePlugin的类！请检查process.py")
+
+def create_zip_package(class_name: str):
+    """打包成ZIP文件，类名全小写"""
+    src_path = Path("src") 
+    zip_name = class_name.lower()  # 类名转全小写作为ZIP名
+    temp_dir = Path(f"temp_{zip_name}")  # 临时打包目录
+    
+    # 1. 创建临时目录
+    temp_dir.mkdir(exist_ok=True)
+    
+    # 2. 复制需要的文件
+    shutil.copytree(src_path / "packages", temp_dir / "packages")
+    shutil.copy(src_path / "process.py", temp_dir)
+    shutil.copy(src_path / "config.toml", temp_dir)
+    
+    # 3. 生成ZIP
+    shutil.make_archive(f"dist/{zip_name}", "zip", temp_dir)
+    
+    # 4. 清理临时目录
+    shutil.rmtree(temp_dir)
+    print(f"✅ 打包完成：dist/{zip_name}.zip")
+
+def main():
+    process_file = "src/process.py"
+    if not os.path.exists(process_file):
+        print(f"❌ 错误：{process_file} 文件不存在！")
+        return
+    
+    try:
+        plugin_class = find_plugin_class(process_file)
+        print(f"找到插件类: {plugin_class} → 包名: {plugin_class.lower()}.zip")
+        
+        Path("dist").mkdir(exist_ok=True)  # 创建dist目录
+        create_zip_package(plugin_class)
+        
+    except Exception as e:
+        print(f"❌ 打包失败：{e}")
+
+if __name__ == "__main__":
+    main()
diff --git a/packup.bat b/packup.bat
new file mode 100644
index 0000000..d0a04b4
--- /dev/null
+++ b/packup.bat
@@ -0,0 +1,21 @@
+@echo off
+REM 依次运行dependence.py和package.py的批处理脚本
+
+echo 正在运行依赖安装脚本...
+python dependence.py
+if %ERRORLEVEL% neq 0 (
+    echo 运行dependence.py失败! 错误码: %ERRORLEVEL%
+    pause
+    exit /b %ERRORLEVEL%
+)
+
+echo 正在运行打包脚本...
+python package.py
+if %ERRORLEVEL% neq 0 (
+    echo 运行package.py失败! 错误码: %ERRORLEVEL%
+    pause
+    exit /b %ERRORLEVEL%
+)
+
+echo 所有脚本执行完毕!
+pause
diff --git a/packup.sh b/packup.sh
new file mode 100644
index 0000000..7d5ed14
--- /dev/null
+++ b/packup.sh
@@ -0,0 +1,18 @@
+#!/bin/bash
+# 依次运行dependence.py和package.py的shell脚本
+
+echo "正在运行依赖安装脚本..."
+python3 dependence.py
+if [ $? -ne 0 ]; then
+    echo "运行dependence.py失败! 错误码: $?"
+    exit $?
+fi
+
+echo "正在运行打包脚本..."
+python3 package.py
+if [ $? -ne 0 ]; then
+    echo "运行package.py失败! 错误码: $?"
+    exit $?
+fi
+
+echo "所有脚本执行完毕!"
diff --git a/requirements.txt b/requirements.txt
new file mode 100644
index 0000000..f0dd0ae
--- /dev/null
+++ b/requirements.txt
@@ -0,0 +1 @@
+openai
\ No newline at end of file
diff --git a/scripts/__init__.py b/scripts/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/scripts/file_store_api.py b/scripts/file_store_api.py
new file mode 100644
index 0000000..7ad7284
--- /dev/null
+++ b/scripts/file_store_api.py
@@ -0,0 +1,65 @@
+import json
+import os
+import logging
+import sqlite3
+import os
+import time
+import toml
+from pathlib import Path
+from http import HTTPStatus
+from datetime import datetime
+
+
+class ConfigManager:
+    """配置管理类，处理应用配置"""
+    def __init__(self, config_path="src"):
+        self.config = {}
+        self.config_path = config_path
+        self.build_config_dict()
+        
+
+    def build_config_dict(self) -> dict[str, str]:
+        config_dict = {}
+        for config_file in Path(self.config_path).rglob("*.toml"):
+            if not config_file.is_file():
+                continue
+            
+            # 获取相对路径的父目录名
+            rel_path = config_file.relative_to(self.config_path)
+            parent_name = rel_path.parent.name if rel_path.parent.name else None
+    
+            if parent_name:
+                key = parent_name
+            else:
+                key = config_file.stem  # 去掉扩展名
+                
+            config_dict[key] = str(config_file.absolute())
+            self.config = config_dict
+    
+    def load_config(self,name):
+        """加载配置文件"""
+        if not os.path.exists(self.config[name]):
+            return {}
+        with open(self.config[name], 'r', encoding='utf-8') as f:
+            try:
+                return toml.load(f)
+            except toml.TomlDecodeError:
+                return {}
+    
+    def save_config(self, key=None, value=None):
+        """保存配置项"""
+        if key is not None and value is not None:
+            # 如果提供了 key 和 value，则更新单个值
+            self.config[key] = value
+        with open(self.config_path, 'w', encoding='utf-8') as f:
+            toml.dump(self.config, f)
+    
+    def update_config(self, config_dict):
+        """更新配置字典"""
+        self.config.update(config_dict)
+        self.save_config()
+
+
+class ChatManager:
+    def __init__(self):
+        return None
\ No newline at end of file
diff --git a/src/__init__.py b/src/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/src/config.toml b/src/config.toml
new file mode 100644
index 0000000..669a31c
--- /dev/null
+++ b/src/config.toml
@@ -0,0 +1,4 @@
+[siloconflow]
+api_key = ""
+modules = ["Qwen/Qwen2.5-7B-Instruct", "Qwen/Qwen2.5-Coder-7B-Instruct"]
+base_url = "https://api.siliconflow.cn/v1"
\ No newline at end of file
diff --git a/src/modules/__init__.py b/src/modules/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/src/modules/plugin_modules.py b/src/modules/plugin_modules.py
new file mode 100644
index 0000000..c205635
--- /dev/null
+++ b/src/modules/plugin_modules.py
@@ -0,0 +1,55 @@
+from abc import ABC, abstractmethod
+from typing import Optional
+from dataclasses import dataclass
+from src.modules import user_module as usermod
+import scripts.file_store_api as file_M
+
+class MessageContext:
+    """封装消息处理的上下文数据"""
+    def __init__(self, uid: str, gid: Optional[str], raw_message: str,id:str):
+        self.raw_message = raw_message
+        self._processed = False
+        self.response: Optional[str] = None
+        # 核心服务实例化
+        self.chat_manager = file_M.ChatManager()
+        self.user = usermod.User(user_id=uid)
+        self.rebot_id = id
+        
+        # 动态加载数据
+        if gid:
+            self.group = usermod.Group(group_id=gid)
+            self.group.current_user = self.user
+        else:
+            self.group = None
+
+
+@dataclass
+class PluginPermission:
+    access_private: bool = False  # 允许处理私聊消息
+    access_group: bool = True     # 允许处理群消息
+    read_history: bool = False    # 允许读取历史记录
+ 
+from pathlib import Path
+import os
+ 
+class BasePlugin(ABC):
+    def __init__(self, ctx: MessageContext):
+        self.ctx = ctx
+        self._config_manager = file_M.ConfigManager(self._get_plugin_config_path())
+        
+    @property 
+    def config(self) -> dict:
+        """直接访问插件配置的快捷方式"""
+        return self._config_manager.load_config(self.name)
+    
+    def _get_plugin_config_path(self) -> str:
+        """获取插件配置目录路径（config/插件名）"""
+        plugin_name = self.__class__.__name__.lower()
+        
+        return str(Path("config") / plugin_name)
+    
+    def save_config(self, config_dict: dict = None) -> bool:
+        """快捷保存配置"""
+        if config_dict:
+            self._config_manager.update_config({"config": config_dict})
+        return self._config_manager.save_config()
\ No newline at end of file
diff --git a/src/modules/user_module.py b/src/modules/user_module.py
new file mode 100644
index 0000000..1dc71aa
--- /dev/null
+++ b/src/modules/user_module.py
@@ -0,0 +1,42 @@
+import json
+import time
+
+
+
+class User:
+    def __init__(self, user_id):
+        self.user_id = user_id
+        self.nickname = "test"
+        self.messages = []
+        self.signal = True
+
+    def set_input_status(self, status):
+        payload = json.dumps({
+                "user_id": self.user_id,
+                "event_type": status
+                })
+        headers = {
+                'Content-Type': 'application/json'
+            }
+        while self.signal:
+            print(f"刷新 {self.nickname} 的输入状态为: {status}")
+            time.sleep(0.5)
+
+    def send_message(self, message):
+        print(f"send message{0}".format(message))
+
+class Group:
+    def __init__(self, group_id,user=None,users=None):
+        self.group_id = group_id
+        self.current_user = user
+        self.nickname = "test"
+        self.users = users
+        self.get_group_users()
+        self.messages =[]
+
+    
+    def get_group_users(self):
+        return(["user1","user2"])
+
+    def send_message(self,message):
+        print(f"send message{0}".format(message))
diff --git a/src/process.py b/src/process.py
new file mode 100644
index 0000000..51f7830
--- /dev/null
+++ b/src/process.py
@@ -0,0 +1,63 @@
+import threading
+from src.modules.plugin_modules import BasePlugin, MessageContext
+from openai import OpenAI
+import time
+
+
+class SiloconFlowAPI:
+    def __init__(self, api_key,modules,base_url):
+        self.client = OpenAI(api_key=api_key, base_url=base_url)
+        self.api_key = api_key
+        self.module = modules
+
+    def get_ai_message(self, message,i=0):
+        try:
+            response = self.client.chat.completions.create(  
+                model=self.module[i],  
+                messages=message,
+                temperature=0.7,  
+                max_tokens=4096
+            )
+            print("token usage {0}".format(response.usage.total_tokens))
+            return response.choices[0].message.content
+        except:
+            if i<len(self.module)-1:
+                i+=1
+                self.get_ai_message(message,i)
+            else:
+                print("server busy,waiting")
+                time.sleep(10)
+                i = 0
+                self.get_ai_message(message,i)
+
+
+
+class Ai_reply(BasePlugin):
+    def after_save(self):
+        config = self.config.get("siloconflow")
+        ai_server = SiloconFlowAPI(base_url= config.get("base_url"), api_key= config.get("api_key"), modules= config.get("modules"))
+        if self.ctx.group is None:
+            self.ctx.user.messages.append({'role': 'user', 
+            'content': f"当前用户为:{self.ctx.user.nickname}，其uid为:{self.ctx.user.user_id}，以上为背景信息，根据背景信息回复用户消息。用户消息:{self.ctx.raw_message}"})
+            message = ai_server.get_ai_message(self.ctx.user.messages)
+            typing_thread = threading.Thread(target=self.ctx.user.set_input_status, args=(1,))
+            typing_thread.start()
+            #获取ai返回值
+            message = ai_server.get_ai_message(self.ctx.user.messages)
+            #结束刷新状态
+            self.ctx.user.signal = False
+            #回收线程
+            typing_thread.join()
+            #指示器归零
+            self.ctx.user.signal = True
+            #发送消息
+            self.ctx.user.send_message(message)
+        else:
+             if "CQ:at,qq={0}".format(self.ctx.rebot_id) in self.ctx.raw_message:
+                self.ctx.group.messages.append({'role': 'user', 
+                                                'content': 
+                                                        f"群聊名称为:{self.ctx.group.nickname},用户在群中的近十条消息为:{self.ctx.group.current_user.messages},用户名叫:{self.ctx.group.current_user.nickname},以上为背景信息，根据背景信息回复消息。消息为：{self.ctx.raw_message}"})
+                message = ai_server.get_ai_message(self.ctx.group.messages)
+           
+                self.ctx.group.send_message(message)
+        return "ok"
\ No newline at end of file
diff --git a/test.py b/test.py
new file mode 100644
index 0000000..1d920e3
--- /dev/null
+++ b/test.py
@@ -0,0 +1,141 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+import sys
+import os
+import importlib
+from pathlib import Path
+from typing import Type, List
+from unittest.mock import Mock
+
+# 添加src目录到系统路径
+
+# 导入基础类
+from src.modules.plugin_modules import MessageContext, BasePlugin
+
+class PluginTester:
+    """
+    插件测试助手类
+    创建模拟上下文和测试配置环境
+    """
+    def __init__(self, plugin_class: Type[BasePlugin], config_dir: str = "test_configs"):
+        self.plugin_class = plugin_class
+        self.config_dir = config_dir
+        os.makedirs(self.config_dir, exist_ok=True)
+        
+        # 创建模拟上下文
+        self.mock_ctx = MessageContext(
+            uid="test_user",
+            gid="test_group",
+            raw_message="测试消息",
+            id="test_bot"
+        )
+        
+        # 创建插件实例
+        self.plugin = self.plugin_class(self.mock_ctx)
+
+    def test_before_load(self):
+        """测试before_load方法（如果存在）"""
+        if hasattr(self.plugin, 'before_load'):
+            print(f"\n正在测试 {self.plugin_class.__name__} 的 before_load")
+            try:
+                result = self.plugin.before_load()
+                print(f"before_load 执行成功，返回值: {result}")
+                return True
+            except Exception as e:
+                print(f"before_load 执行出错: {str(e)}")
+                return False
+        else:
+            print(f"\n{self.plugin_class.__name__} 中没有找到 before_load 方法")
+            return None
+
+    def test_after_load(self):
+        """测试after_load方法（如果存在）"""
+        if hasattr(self.plugin, 'after_load'):
+            print(f"\n正在测试 {self.plugin_class.__name__} 的 after_load")
+            try:
+                result = self.plugin.after_load()
+                print(f"after_load 执行成功，返回值: {result}")
+                return True
+            except Exception as e:
+                print(f"after_load 执行出错: {str(e)}")
+                return False
+        else:
+            print(f"\n{self.plugin_class.__name__} 中没有找到 after_load 方法")
+            return None
+
+    def test_after_save(self):
+        """测试after_save方法（如果存在）"""
+        if hasattr(self.plugin, 'after_save'):
+            print(f"\n正在测试 {self.plugin_class.__name__} 的 after_save")
+            try:
+                result = self.plugin.after_save()
+                print(f"after_save 执行成功，返回值: {result}")
+                return True
+            except Exception as e:
+                print(f"after_save 执行出错: {str(e)}")
+                return False
+        else:
+            print(f"\n{self.plugin_class.__name__} 中没有找到 after_save 方法")
+            return None
+
+def find_plugin_classes(module_path: str) -> List[Type[BasePlugin]]:
+    """
+    在process.py中查找所有继承自BasePlugin的类
+    返回类类型列表
+    """
+    plugin_classes = []
+    
+    # 动态导入模块
+    module_name = os.path.splitext(os.path.basename(module_path))[0]
+    spec = importlib.util.spec_from_file_location(module_name, module_path)
+    module = importlib.util.module_from_spec(spec)
+    spec.loader.exec_module(module)
+    
+    # 查找所有继承自BasePlugin的类
+    for name, obj in vars(module).items():
+        try:
+            if isinstance(obj, type) and issubclass(obj, BasePlugin) and obj != BasePlugin:
+                plugin_classes.append(obj)
+        except TypeError:
+            continue
+            
+    return plugin_classes
+
+def main():
+    # process.py文件路径
+    process_path = Path("src/process.py")
+    
+    if not process_path.exists():
+        print(f"错误: 在 {process_path} 没有找到 process.py")
+        return
+    
+    # 查找process.py中的所有插件类
+    plugin_classes = find_plugin_classes(str(process_path))
+    
+    if not plugin_classes:
+        print("在 process.py 中没有找到插件类")
+        return
+    
+    print(f"发现了 {len(plugin_classes)} 个需要测试的插件类:")
+    for i, plugin_class in enumerate(plugin_classes, 1):
+        print(f"{i}. {plugin_class.__name__}")
+    
+    # 测试每个插件类
+    for plugin_class in plugin_classes:
+        print(f"\n{'='*50}")
+        print(f"正在测试插件: {plugin_class.__name__}")
+        
+        tester = PluginTester(plugin_class)
+        
+        # 按照自然顺序测试生命周期方法
+        tester.test_before_load()
+        tester.test_after_load()
+        
+        # 如果需要测试配置保存
+        if hasattr(plugin_class, 'after_save'):
+            tester.test_after_save()
+        
+        print(f"{'='*50}\n")
+
+if __name__ == "__main__":
+    main()