diff --git a/.gitattributes b/.gitattributes index cf5cffa22..1c4521779 100644 --- a/.gitattributes +++ b/.gitattributes @@ -1,2 +1,3 @@ *.bat text eol=crlf -*.cmd text eol=crlf \ No newline at end of file +*.cmd text eol=crlf +MaiLauncher.bat text eol=crlf working-tree-encoding=GBK \ No newline at end of file diff --git a/.github/ISSUE_TEMPLATE/bug_report.yml b/.github/ISSUE_TEMPLATE/bug_report.yml index a4245d0a0..ce2623260 100644 --- a/.github/ISSUE_TEMPLATE/bug_report.yml +++ b/.github/ISSUE_TEMPLATE/bug_report.yml @@ -12,6 +12,23 @@ body: - label: "我确认在 Issues 列表中并无其他人已经提出过与此问题相同或相似的问题" required: true - label: "我使用了 Docker" +- type: dropdown + attributes: + label: "使用的分支" + description: "请选择您正在使用的版本分支" + options: + - main + - main-fix + - refactor + validations: + required: true +- type: input + attributes: + label: "具体版本号" + description: "请输入您使用的具体版本号" + placeholder: "例如：0.5.11、0.5.8" + validations: + required: true - type: textarea attributes: label: 遇到的问题 diff --git a/.github/pull_request_template.md b/.github/pull_request_template.md new file mode 100644 index 000000000..19a587960 --- /dev/null +++ b/.github/pull_request_template.md @@ -0,0 +1,17 @@ + +- 🔴**当前项目处于重构阶段（2025.3.14-）** +- ✅ 接受：与main直接相关的Bug修复：提交到main-fix分支 +- ⚠️ 冻结：所有新功能开发和非紧急重构 + +# 请填写以下内容 +（删除掉中括号内的空格，并替换为**小写的x**） +1. - [ ] `main` 分支 **禁止修改**，请确认本次提交的分支 **不是 `main` 分支** +2. - [ ] 本次更新 **包含破坏性变更**（如数据库结构变更、配置文件修改等） +3. - [ ] 本次更新是否经过测试 +4. - [ ] 请**不要**在数据库中添加group_id字段，这会影响本项目对其他平台的兼容 +5. 请填写破坏性更新的具体内容（如有）: +6. 请简要说明本次更新的内容和目的： +# 其他信息 +- **关联 Issue**：Close # +- **截图/GIF**： +- **附加信息**: diff --git a/.github/workflows/docker-image.yml b/.github/workflows/docker-image.yml index 5b09b8cda..c06d967ca 100644 --- a/.github/workflows/docker-image.yml +++ b/.github/workflows/docker-image.yml @@ -4,11 +4,10 @@ on: push: branches: - main - - debug # 新增 debug 分支触发 - - stable-dev + - main-fix tags: - - 'v*' - workflow_dispatch: + - 'v*' + workflow_dispatch: jobs: build-and-push: @@ -16,7 +15,7 @@ jobs: steps: - name: Checkout code uses: actions/checkout@v4 - + - name: Set up Docker Buildx uses: docker/setup-buildx-action@v3 @@ -33,10 +32,8 @@ jobs: echo "tags=${{ secrets.DOCKERHUB_USERNAME }}/maimbot:${{ github.ref_name }},${{ secrets.DOCKERHUB_USERNAME }}/maimbot:latest" >> $GITHUB_OUTPUT elif [ "${{ github.ref }}" == "refs/heads/main" ]; then echo "tags=${{ secrets.DOCKERHUB_USERNAME }}/maimbot:main,${{ secrets.DOCKERHUB_USERNAME }}/maimbot:latest" >> $GITHUB_OUTPUT - elif [ "${{ github.ref }}" == "refs/heads/debug" ]; then - echo "tags=${{ secrets.DOCKERHUB_USERNAME }}/maimbot:debug" >> $GITHUB_OUTPUT - elif [ "${{ github.ref }}" == "refs/heads/stable-dev" ]; then - echo "tags=${{ secrets.DOCKERHUB_USERNAME }}/maimbot:stable-dev" >> $GITHUB_OUTPUT + elif [ "${{ github.ref }}" == "refs/heads/main-fix" ]; then + echo "tags=${{ secrets.DOCKERHUB_USERNAME }}/maimbot:main-fix" >> $GITHUB_OUTPUT fi - name: Build and Push Docker Image @@ -48,4 +45,4 @@ jobs: tags: ${{ steps.tags.outputs.tags }} push: true cache-from: type=registry,ref=${{ secrets.DOCKERHUB_USERNAME }}/maimbot:buildcache - cache-to: type=registry,ref=${{ secrets.DOCKERHUB_USERNAME }}/maimbot:buildcache,mode=max \ No newline at end of file + cache-to: type=registry,ref=${{ secrets.DOCKERHUB_USERNAME }}/maimbot:buildcache,mode=max diff --git a/.github/workflows/precheck.yml b/.github/workflows/precheck.yml new file mode 100644 index 000000000..a7524ccb3 --- /dev/null +++ b/.github/workflows/precheck.yml @@ -0,0 +1,29 @@ +# .github/workflows/precheck.yml +name: PR Precheck +on: [pull_request] + +jobs: + conflict-check: + runs-on: ubuntu-latest + steps: + - uses: actions/checkout@v3 + - name: Check Conflicts + run: | + git fetch origin main + if git diff --name-only --diff-filter=U origin/main...HEAD | grep .; then + echo "CONFLICT=true" >> $GITHUB_ENV + fi + labeler: + runs-on: ubuntu-latest + needs: conflict-check + steps: + - uses: actions/github-script@v6 + if: env.CONFLICT == 'true' + with: + script: | + github.rest.issues.addLabels({ + owner: context.repo.owner, + repo: context.repo.repo, + issue_number: context.issue.number, + labels: ['🚫冲突需处理'] + }) diff --git a/.github/workflows/ruff.yml b/.github/workflows/ruff.yml index 0d1e50c5a..697c47759 100644 --- a/.github/workflows/ruff.yml +++ b/.github/workflows/ruff.yml @@ -5,4 +5,5 @@ jobs: runs-on: ubuntu-latest steps: - uses: actions/checkout@v4 - - uses: astral-sh/ruff-action@v3 \ No newline at end of file + - uses: astral-sh/ruff-action@v3 + diff --git a/.gitignore b/.gitignore index 2dd6de62e..2658ecc6f 100644 --- a/.gitignore +++ b/.gitignore @@ -16,6 +16,7 @@ memory_graph.gml .env.* config/bot_config_dev.toml config/bot_config.toml +config/bot_config.toml.bak src/plugins/remote/client_uuid.json # Byte-compiled / optimized / DLL files __pycache__/ @@ -25,7 +26,7 @@ llm_statistics.txt mongodb napcat run_dev.bat - +elua.confirmed # C extensions *.so @@ -205,3 +206,13 @@ jieba.cache .idea *.iml *.ipr + +# PyEnv +# If using PyEnv and configured to use a specific Python version locally +# a .local-version file will be created in the root of the project to specify the version. +.python-version + +OtherRes.txt + +/eula.confirmed +/privacy.confirmed \ No newline at end of file diff --git a/CLAUDE.md b/CLAUDE.md index d30b0e651..1b61f8ed4 100644 --- a/CLAUDE.md +++ b/CLAUDE.md @@ -1,6 +1,196 @@ -# MaiMBot 开发指南 +# MaiMBot 开发文档 -## 🛠️ 常用命令 +## 📊 系统架构图 + +```mermaid +graph TD + A[入口点] --> B[核心模块] + A --> C[插件系统] + B --> D[通用功能] + C --> E[聊天系统] + C --> F[记忆系统] + C --> G[情绪系统] + C --> H[意愿系统] + C --> I[其他插件] + + %% 入口点 + A1[bot.py] --> A + A2[run.py] --> A + A3[webui.py] --> A + + %% 核心模块 + B1[src/common/logger.py] --> B + B2[src/common/database.py] --> B + + %% 通用功能 + D1[日志系统] --> D + D2[数据库连接] --> D + D3[配置管理] --> D + + %% 聊天系统 + E1[消息处理] --> E + E2[提示构建] --> E + E3[LLM生成] --> E + E4[关系管理] --> E + + %% 记忆系统 + F1[记忆图] --> F + F2[记忆构建] --> F + F3[记忆检索] --> F + F4[记忆遗忘] --> F + + %% 情绪系统 + G1[情绪状态] --> G + G2[情绪更新] --> G + G3[情绪衰减] --> G + + %% 意愿系统 + H1[回复意愿] --> H + H2[意愿模式] --> H + H3[概率控制] --> H + + %% 其他插件 + I1[远程统计] --> I + I2[配置重载] --> I + I3[日程生成] --> I +``` + +## 📁 核心文件索引 + +| 功能 | 文件路径 | 描述 | +|------|----------|------| +| **入口点** | `/bot.py` | 主入口，初始化环境和启动服务 | +| | `/run.py` | 安装管理脚本，主要用于Windows | +| | `/webui.py` | Gradio基础的配置UI | +| **配置** | `/template.env` | 环境变量模板 | +| | `/template/bot_config_template.toml` | 机器人配置模板 | +| **核心基础** | `/src/common/database.py` | MongoDB连接管理 | +| | `/src/common/logger.py` | 基于loguru的日志系统 | +| **聊天系统** | `/src/plugins/chat/bot.py` | 消息处理核心逻辑 | +| | `/src/plugins/chat/config.py` | 配置管理与验证 | +| | `/src/plugins/chat/llm_generator.py` | LLM响应生成 | +| | `/src/plugins/chat/prompt_builder.py` | LLM提示构建 | +| **记忆系统** | `/src/plugins/memory_system/memory.py` | 图结构记忆实现 | +| | `/src/plugins/memory_system/draw_memory.py` | 记忆可视化 | +| **情绪系统** | `/src/plugins/moods/moods.py` | 情绪状态管理 | +| **意愿系统** | `/src/plugins/willing/willing_manager.py` | 回复意愿管理 | +| | `/src/plugins/willing/mode_classical.py` | 经典意愿模式 | +| | `/src/plugins/willing/mode_dynamic.py` | 动态意愿模式 | +| | `/src/plugins/willing/mode_custom.py` | 自定义意愿模式 | + +## 🔄 模块依赖关系 + +```mermaid +flowchart TD + A[bot.py] --> B[src/common/logger.py] + A --> C[src/plugins/chat/bot.py] + + C --> D[src/plugins/chat/config.py] + C --> E[src/plugins/chat/llm_generator.py] + C --> F[src/plugins/memory_system/memory.py] + C --> G[src/plugins/moods/moods.py] + C --> H[src/plugins/willing/willing_manager.py] + + E --> D + E --> I[src/plugins/chat/prompt_builder.py] + E --> J[src/plugins/models/utils_model.py] + + F --> B + F --> D + F --> J + + G --> D + + H --> B + H --> D + H --> K[src/plugins/willing/mode_classical.py] + H --> L[src/plugins/willing/mode_dynamic.py] + H --> M[src/plugins/willing/mode_custom.py] + + I --> B + I --> F + I --> G + + J --> B +``` + +## 🔄 消息处理流程 + +```mermaid +sequenceDiagram + participant User + participant ChatBot + participant WillingManager + participant Memory + participant PromptBuilder + participant LLMGenerator + participant MoodManager + + User->>ChatBot: 发送消息 + ChatBot->>ChatBot: 消息预处理 + ChatBot->>Memory: 记忆激活 + Memory-->>ChatBot: 激活度 + ChatBot->>WillingManager: 更新回复意愿 + WillingManager-->>ChatBot: 回复决策 + + alt 决定回复 + ChatBot->>PromptBuilder: 构建提示 + PromptBuilder->>Memory: 获取相关记忆 + Memory-->>PromptBuilder: 相关记忆 + PromptBuilder->>MoodManager: 获取情绪状态 + MoodManager-->>PromptBuilder: 情绪状态 + PromptBuilder-->>ChatBot: 完整提示 + ChatBot->>LLMGenerator: 生成回复 + LLMGenerator-->>ChatBot: AI回复 + ChatBot->>MoodManager: 更新情绪 + ChatBot->>User: 发送回复 + else 不回复 + ChatBot->>WillingManager: 更新未回复状态 + end +``` + +## 📋 类和功能清单 + +### 🤖 聊天系统 (`src/plugins/chat/`) + +| 类/功能 | 文件 | 描述 | +|--------|------|------| +| `ChatBot` | `bot.py` | 消息处理主类 | +| `ResponseGenerator` | `llm_generator.py` | 响应生成器 | +| `PromptBuilder` | `prompt_builder.py` | 提示构建器 | +| `Message`系列 | `message.py` | 消息表示类 | +| `RelationshipManager` | `relationship_manager.py` | 用户关系管理 | +| `EmojiManager` | `emoji_manager.py` | 表情符号管理 | + +### 🧠 记忆系统 (`src/plugins/memory_system/`) + +| 类/功能 | 文件 | 描述 | +|--------|------|------| +| `Memory_graph` | `memory.py` | 图结构记忆存储 | +| `Hippocampus` | `memory.py` | 记忆管理主类 | +| `memory_compress()` | `memory.py` | 记忆压缩函数 | +| `get_relevant_memories()` | `memory.py` | 记忆检索函数 | +| `operation_forget_topic()` | `memory.py` | 记忆遗忘函数 | + +### 😊 情绪系统 (`src/plugins/moods/`) + +| 类/功能 | 文件 | 描述 | +|--------|------|------| +| `MoodManager` | `moods.py` | 情绪管理器单例 | +| `MoodState` | `moods.py` | 情绪状态数据类 | +| `update_mood_from_emotion()` | `moods.py` | 情绪更新函数 | +| `_apply_decay()` | `moods.py` | 情绪衰减函数 | + +### 🤔 意愿系统 (`src/plugins/willing/`) + +| 类/功能 | 文件 | 描述 | +|--------|------|------| +| `WillingManager` | `willing_manager.py` | 意愿管理工厂类 | +| `ClassicalWillingManager` | `mode_classical.py` | 经典意愿模式 | +| `DynamicWillingManager` | `mode_dynamic.py` | 动态意愿模式 | +| `CustomWillingManager` | `mode_custom.py` | 自定义意愿模式 | + +## 🔧 常用命令 - **运行机器人**: `python run.py` 或 `python bot.py` - **安装依赖**: `pip install --upgrade -r requirements.txt` @@ -30,19 +220,25 @@ - **错误处理**: 使用带有具体异常的try/except - **文档**: 为类和公共函数编写docstrings -## 🧩 系统架构 +## 📋 常见修改点 -- **框架**: NoneBot2框架与插件架构 -- **数据库**: MongoDB持久化存储 -- **设计模式**: 工厂模式和单例管理器 -- **配置管理**: 使用环境变量和TOML文件 -- **内存系统**: 基于图的记忆结构，支持记忆构建、压缩、检索和遗忘 -- **情绪系统**: 情绪模拟与概率权重 -- **LLM集成**: 支持多个LLM服务提供商(ChatAnywhere, SiliconFlow, DeepSeek) +### 配置修改 +- **机器人配置**: `/template/bot_config_template.toml` +- **环境变量**: `/template.env` -## ⚙️ 环境配置 +### 行为定制 +- **个性调整**: `src/plugins/chat/config.py` 中的 BotConfig 类 +- **回复意愿算法**: `src/plugins/willing/mode_classical.py` +- **情绪反应模式**: `src/plugins/moods/moods.py` -- 使用`template.env`作为环境变量模板 -- 使用`template/bot_config_template.toml`作为机器人配置模板 -- MongoDB配置: 主机、端口、数据库名 -- API密钥配置: 各LLM提供商的API密钥 +### 消息处理 +- **消息管道**: `src/plugins/chat/message.py` +- **话题识别**: `src/plugins/chat/topic_identifier.py` + +### 记忆与学习 +- **记忆算法**: `src/plugins/memory_system/memory.py` +- **手动记忆构建**: `src/plugins/memory_system/memory_manual_build.py` + +### LLM集成 +- **LLM提供商**: `src/plugins/chat/llm_generator.py` +- **模型参数**: `template/bot_config_template.toml` 的 [model] 部分 \ No newline at end of file diff --git a/EULA.md b/EULA.md new file mode 100644 index 000000000..c7a734a27 --- /dev/null +++ b/EULA.md @@ -0,0 +1,97 @@ +# **MaiMBot最终用户许可协议** +**版本：V1.0** +**更新日期：2025年3月18日** +**生效日期：2025年3月18日** +**适用的MaiMBot版本号：<=v0.5.15** + +**2025© MaiMBot项目团队** + +--- + +## 一、一般条款 + +**1.1** MaiMBot项目（包括MaiMBot的源代码、可执行文件、文档，以及其它在本协议中所列出的文件）（以下简称“本项目”）是由开发者及贡献者（以下简称“项目团队”）共同维护，为用户提供自动回复功能的机器人代码项目。以下最终用户许可协议（EULA，以下简称“本协议”）是用户（以下简称“您”）与项目团队之间关于使用本项目所订立的合同条件。 + +**1.2** 在运行或使用本项目之前，您**必须阅读并同意本协议的所有条款**。未成年人或其它无/不完全民事行为能力责任人请**在监护人的陪同下**阅读并同意本协议。如果您不同意，则不得运行或使用本项目。在这种情况下，您应立即从您的设备上卸载或删除本项目及其所有副本。 + + +## 二、许可授权 + +### 源代码许可 +**2.1** 您**了解**本项目的源代码是基于GPLv3（GNU通用公共许可证第三版）开源协议发布的。您**可以自由使用、修改、分发**本项目的源代码，但**必须遵守**GPLv3许可证的要求。详细内容请参阅项目仓库中的LICENSE文件。 + +**2.2** 您**了解**本项目的源代码中可能包含第三方开源代码，这些代码的许可证可能与GPLv3许可证不同。您**同意**在使用这些代码时**遵守**相应的许可证要求。 + + +### 输入输出内容授权 + +**2.3** 您**了解**本项目是使用您的配置信息、提交的指令（以下简称“输入内容”）和生成的内容（以下简称“输出内容”）构建请求发送到第三方API生成回复的机器人项目。 + +**2.4** 您**授权**本项目使用您的输入和输出内容按照项目的隐私政策用于以下行为： + - 调用第三方API生成回复； + - 调用第三方API用于构建本项目专用的存储于您部署或使用的数据库中的知识库和记忆库； + - 收集并记录本项目专用的存储于您部署或使用的设备中的日志； + +**2.4** 您**了解**本项目的源代码中包含第三方API的调用代码，这些API的使用可能受到第三方的服务条款和隐私政策的约束。在使用这些API时，您**必须遵守**相应的服务条款。 + +**2.5** 项目团队**不对**第三方API的服务质量、稳定性、准确性、安全性负责，亦**不对**第三方API的服务变更、终止、限制等行为负责。 + + +## 三、用户行为 + +**3.1** 您**了解**本项目会将您的配置信息、输入指令和生成内容发送到第三方API，您**不应**在输入指令和生成内容中包含以下内容： + - 涉及任何国家或地区秘密、商业秘密或其他可能会对国家或地区安全或者公共利益造成不利影响的数据； + - 涉及个人隐私、个人信息或其他敏感信息的数据； + - 任何侵犯他人合法权益的内容； + - 任何违反国家或地区法律法规、政策规定的内容； + +**3.2** 您**不应**将本项目用于以下用途： + - 违反任何国家或地区法律法规、政策规定的行为； + +**3.3** 您**应当**自行确保您被存储在本项目的知识库、记忆库和日志中的输入和输出内容的合法性与合规性以及存储行为的合法性与合规性。您需**自行承担**由此产生的任何法律责任。 + + + +## 四、免责条款 + +**4.1** 本项目的输出内容依赖第三方API，**不受**项目团队控制，亦**不代表**项目团队的观点。 + +**4.2** 除本协议条目2.4提到的隐私政策之外，项目团队**不会**对您提供任何形式的担保，亦**不对**使用本项目的造成的任何后果负责。 + +## 五、其他条款 + +**5.1** 项目团队有权**随时修改本协议的条款**，但**没有**义务通知您。修改后的协议将在本项目的新版本中生效，您应定期检查本协议的最新版本。 + +**5.2** 项目团队**保留**本协议的最终解释权。 + + +## 附录：其他重要须知 + +### 一、过往版本使用条件追溯 + +**1.1** 对于本项目此前未配备 EULA 协议的版本，自本协议发布之日起，若用户希望继续使用本项目，应在本协议生效后的合理时间内，通过升级到最新版本并同意本协议全部条款。若在本版协议生效日（2025年3月18日）之后，用户仍使用此前无 EULA 协议的项目版本且未同意本协议，则用户无权继续使用，项目方有权采取措施阻止其使用行为，并保留追究相关法律责任的权利。 + + +### 二、风险提示 + +**2.1 隐私安全风险** + + - 本项目会将您的配置信息、输入指令和生成内容发送到第三方API，而这些API的服务质量、稳定性、准确性、安全性不受项目团队控制。 + - 本项目会收集您的输入和输出内容，用于构建本项目专用的知识库和记忆库，以提高回复的准确性和连贯性。 + + **因此，为了保障您的隐私信息安全，请注意以下事项：** + + - 避免在涉及个人隐私、个人信息或其他敏感信息的环境中使用本项目； + - 避免在不可信的环境中使用本项目； + +**2.2 精神健康风险** + +本项目仅为工具型机器人，不具备情感交互能力。建议用户： + - 避免过度依赖AI回复处理现实问题或情绪困扰； + - 如感到心理不适，请及时寻求专业心理咨询服务。 + - 如遇心理困扰，请寻求专业帮助（全国心理援助热线：12355）。 + +### 三、其他 +**3.1 争议解决** + - 本协议适用中国法律，争议提交相关地区法院管辖； + - 若因GPLv3许可产生纠纷，以许可证官方解释为准。 diff --git a/MaiLauncher.bat b/MaiLauncher.bat new file mode 100644 index 000000000..619f9c65d --- /dev/null +++ b/MaiLauncher.bat @@ -0,0 +1,636 @@ +@echo off +@setlocal enabledelayedexpansion +@chcp 936 + +@REM 设置版本号 +set "VERSION=1.0" + +title 麦麦Bot控制台 v%VERSION% + +@REM 设置Python和Git环境变量 +set "_root=%~dp0" +set "_root=%_root:~0,-1%" +cd "%_root%" + + +:search_python +cls +if exist "%_root%\python" ( + set "PYTHON_HOME=%_root%\python" +) else if exist "%_root%\venv" ( + call "%_root%\venv\Scripts\activate.bat" + set "PYTHON_HOME=%_root%\venv\Scripts" +) else ( + echo 正在自动查找Python解释器... + + where python >nul 2>&1 + if %errorlevel% equ 0 ( + for /f "delims=" %%i in ('where python') do ( + echo %%i | findstr /i /c:"!LocalAppData!\Microsoft\WindowsApps\python.exe" >nul + if errorlevel 1 ( + echo 找到Python解释器：%%i + set "py_path=%%i" + goto :validate_python + ) + ) + ) + set "search_paths=%ProgramFiles%\Git*;!LocalAppData!\Programs\Python\Python*" + for /d %%d in (!search_paths!) do ( + if exist "%%d\python.exe" ( + set "py_path=%%d\python.exe" + goto :validate_python + ) + ) + echo 没有找到Python解释器,要安装吗? + set /p pyinstall_confirm="继续？(Y/n): " + if /i "!pyinstall_confirm!"=="Y" ( + cls + echo 正在安装Python... + winget install --id Python.Python.3.13 -e --accept-package-agreements --accept-source-agreements + if %errorlevel% neq 0 ( + echo 安装失败，请手动安装Python + start https://www.python.org/downloads/ + exit /b + ) + echo 安装完成，正在验证Python... + goto search_python + + ) else ( + echo 取消安装Python，按任意键退出... + pause >nul + exit /b + ) + + echo 错误：未找到可用的Python解释器！ + exit /b 1 + + :validate_python + "!py_path!" --version >nul 2>&1 + if %errorlevel% neq 0 ( + echo 无效的Python解释器：%py_path% + exit /b 1 + ) + + :: 提取安装目录 + for %%i in ("%py_path%") do set "PYTHON_HOME=%%~dpi" + set "PYTHON_HOME=%PYTHON_HOME:~0,-1%" +) +if not exist "%PYTHON_HOME%\python.exe" ( + echo Python路径验证失败：%PYTHON_HOME% + echo 请检查Python安装路径中是否有python.exe文件 + exit /b 1 +) +echo 成功设置Python路径：%PYTHON_HOME% + + + +:search_git +cls +if exist "%_root%\tools\git\bin" ( + set "GIT_HOME=%_root%\tools\git\bin" +) else ( + echo 正在自动查找Git... + + where git >nul 2>&1 + if %errorlevel% equ 0 ( + for /f "delims=" %%i in ('where git') do ( + set "git_path=%%i" + goto :validate_git + ) + ) + echo 正在扫描常见安装路径... + set "search_paths=!ProgramFiles!\Git\cmd" + for /f "tokens=*" %%d in ("!search_paths!") do ( + if exist "%%d\git.exe" ( + set "git_path=%%d\git.exe" + goto :validate_git + ) + ) + echo 没有找到Git，要安装吗？ + set /p confirm="继续？(Y/N): " + if /i "!confirm!"=="Y" ( + cls + echo 正在安装Git... + set "custom_url=https://ghfast.top/https://github.com/git-for-windows/git/releases/download/v2.48.1.windows.1/Git-2.48.1-64-bit.exe" + + set "download_path=%TEMP%\Git-Installer.exe" + + echo 正在下载Git安装包... + curl -L -o "!download_path!" "!custom_url!" + + if exist "!download_path!" ( + echo 下载成功，开始安装Git... + start /wait "" "!download_path!" /SILENT /NORESTART + ) else ( + echo 下载失败，请手动安装Git + start https://git-scm.com/download/win + exit /b + ) + + del "!download_path!" + echo 临时文件已清理。 + + echo 安装完成，正在验证Git... + where git >nul 2>&1 + if %errorlevel% equ 0 ( + for /f "delims=" %%i in ('where git') do ( + set "git_path=%%i" + goto :validate_git + ) + goto :search_git + + ) else ( + echo 安装完成，但未找到Git，请手动安装Git + start https://git-scm.com/download/win + exit /b + ) + + ) else ( + echo 取消安装Git，按任意键退出... + pause >nul + exit /b + ) + + echo 错误：未找到可用的Git！ + exit /b 1 + + :validate_git + "%git_path%" --version >nul 2>&1 + if %errorlevel% neq 0 ( + echo 无效的Git：%git_path% + exit /b 1 + ) + + :: 提取安装目录 + for %%i in ("%git_path%") do set "GIT_HOME=%%~dpi" + set "GIT_HOME=%GIT_HOME:~0,-1%" +) + +:search_mongodb +cls +sc query | findstr /i "MongoDB" >nul +if !errorlevel! neq 0 ( + echo MongoDB服务未运行，是否尝试运行服务？ + set /p confirm="是否启动？(Y/N): " + if /i "!confirm!"=="Y" ( + echo 正在尝试启动MongoDB服务... + powershell -Command "Start-Process -Verb RunAs cmd -ArgumentList '/c net start MongoDB'" + echo 正在等待MongoDB服务启动... + echo 按下任意键跳过等待... + timeout /t 30 >nul + sc query | findstr /i "MongoDB" >nul + if !errorlevel! neq 0 ( + echo MongoDB服务启动失败，可能是没有安装，要安装吗？ + set /p install_confirm="继续安装？(Y/N): " + if /i "!install_confirm!"=="Y" ( + echo 正在安装MongoDB... + winget install --id MongoDB.Server -e --accept-package-agreements --accept-source-agreements + echo 安装完成，正在启动MongoDB服务... + net start MongoDB + if !errorlevel! neq 0 ( + echo 启动MongoDB服务失败，请手动启动 + exit /b + ) else ( + echo MongoDB服务已成功启动 + ) + ) else ( + echo 取消安装MongoDB，按任意键退出... + pause >nul + exit /b + ) + ) + ) else ( + echo "警告：MongoDB服务未运行，将导致MaiMBot无法访问数据库！" + ) +) else ( + echo MongoDB服务已运行 +) + +@REM set "GIT_HOME=%_root%\tools\git\bin" +set "PATH=%PYTHON_HOME%;%GIT_HOME%;%PATH%" + +:install_maim +if not exist "!_root!\bot.py" ( + cls + echo 你似乎没有安装麦麦Bot，要安装在当前目录吗？ + set /p confirm="继续？(Y/N): " + if /i "!confirm!"=="Y" ( + echo 要使用Git代理下载吗？ + set /p proxy_confirm="继续？(Y/N): " + if /i "!proxy_confirm!"=="Y" ( + echo 正在安装麦麦Bot... + git clone https://ghfast.top/https://github.com/SengokuCola/MaiMBot + ) else ( + echo 正在安装麦麦Bot... + git clone https://github.com/SengokuCola/MaiMBot + ) + xcopy /E /H /I MaiMBot . >nul 2>&1 + rmdir /s /q MaiMBot + git checkout main-fix + + echo 安装完成，正在安装依赖... + python -m pip config set global.index-url https://mirrors.aliyun.com/pypi/simple + python -m pip install virtualenv + python -m virtualenv venv + call venv\Scripts\activate.bat + python -m pip install -r requirements.txt + + echo 安装完成，要编辑配置文件吗？ + set /p edit_confirm="继续？(Y/N): " + if /i "!edit_confirm!"=="Y" ( + goto config_menu + ) else ( + echo 取消编辑配置文件，按任意键返回主菜单... + ) + ) +) + + +@REM git获取当前分支名并保存在变量里 +for /f "delims=" %%b in ('git symbolic-ref --short HEAD 2^>nul') do ( + set "BRANCH=%%b" +) + +@REM 根据不同分支名给分支名字符串使用不同颜色 +echo 分支名: %BRANCH% +if "!BRANCH!"=="main" ( + set "BRANCH_COLOR=" +) else if "!BRANCH!"=="main-fix" ( + set "BRANCH_COLOR=" +@REM ) else if "%BRANCH%"=="stable-dev" ( +@REM set "BRANCH_COLOR=" +) else ( + set "BRANCH_COLOR=" +) + +@REM endlocal & set "BRANCH_COLOR=%BRANCH_COLOR%" + +:check_is_venv +echo 正在检查虚拟环境状态... +if exist "%_root%\config\no_venv" ( + echo 检测到no_venv,跳过虚拟环境检查 + goto menu +) + +:: 环境检测 +if defined VIRTUAL_ENV ( + goto menu +) + +echo ===================================== +echo 虚拟环境检测警告： +echo 当前使用系统Python路径：!PYTHON_HOME! +echo 未检测到激活的虚拟环境！ + +:env_interaction +echo ===================================== +echo 请选择操作： +echo 1 - 创建并激活Venv虚拟环境 +echo 2 - 创建/激活Conda虚拟环境 +echo 3 - 临时跳过本次检查 +echo 4 - 永久跳过虚拟环境检查 +set /p choice="请输入选项(1-4): " + +if "!choice!"=="4" ( + echo 要永久跳过虚拟环境检查吗？ + set /p no_venv_confirm="继续？(Y/N): ....." + if /i "!no_venv_confirm!"=="Y" ( + echo 1 > "%_root%\config\no_venv" + echo 已创建no_venv文件 + pause >nul + goto menu + ) else ( + echo 取消跳过虚拟环境检查，按任意键返回... + pause >nul + goto env_interaction + ) +) + +if "!choice!"=="3" ( + echo 警告：使用系统环境可能导致依赖冲突！ + timeout /t 2 >nul + goto menu +) + +if "!choice!"=="2" goto handle_conda +if "!choice!"=="1" goto handle_venv + +echo 无效的输入，请输入1-4之间的数字 +timeout /t 2 >nul +goto env_interaction + +:handle_venv +python -m pip config set global.index-url https://mirrors.aliyun.com/pypi/simple +echo 正在初始化Venv环境... +python -m pip install virtualenv || ( + echo 安装环境失败，错误码：!errorlevel! + pause + goto env_interaction +) +echo 创建虚拟环境到：venv + python -m virtualenv venv || ( + echo 环境创建失败，错误码：!errorlevel! + pause + goto env_interaction +) + +call venv\Scripts\activate.bat +echo 已激活Venv环境 +echo 要安装依赖吗？ +set /p install_confirm="继续？(Y/N): " +if /i "!install_confirm!"=="Y" ( + goto update_dependencies +) +goto menu + +:handle_conda +where conda >nul 2>&1 || ( + echo 未检测到conda，可能原因： + echo 1. 未安装Miniconda + echo 2. conda配置异常 + timeout /t 10 >nul + goto env_interaction +) + +:conda_menu +echo 请选择Conda操作： +echo 1 - 创建新环境 +echo 2 - 激活已有环境 +echo 3 - 返回上级菜单 +set /p choice="请输入选项(1-3): " + +if "!choice!"=="3" goto env_interaction +if "!choice!"=="2" goto activate_conda +if "!choice!"=="1" goto create_conda + +echo 无效的输入，请输入1-3之间的数字 +timeout /t 2 >nul +goto conda_menu + +:create_conda +set /p "CONDA_ENV=请输入新环境名称：" +if "!CONDA_ENV!"=="" ( + echo 环境名称不能为空！ + goto create_conda +) +conda create -n !CONDA_ENV! python=3.13 -y || ( + echo 环境创建失败，错误码：!errorlevel! + timeout /t 10 >nul + goto conda_menu +) +goto activate_conda + +:activate_conda +set /p "CONDA_ENV=请输入要激活的环境名称：" +call conda activate !CONDA_ENV! || ( + echo 激活失败，可能原因： + echo 1. 环境不存在 + echo 2. conda配置异常 + pause + goto conda_menu +) +echo 成功激活conda环境：!CONDA_ENV! +echo 要安装依赖吗？ +set /p install_confirm="继续？(Y/N): " +if /i "!install_confirm!"=="Y" ( + goto update_dependencies +) +:menu +@chcp 936 +cls +echo 麦麦Bot控制台 v%VERSION% 当前分支: %BRANCH_COLOR%%BRANCH% +echo 当前Python环境: !PYTHON_HOME! +echo ====================== +echo 1. 更新并启动麦麦Bot (默认) +echo 2. 直接启动麦麦Bot +echo 3. 启动麦麦配置界面 +echo 4. 打开麦麦神奇工具箱 +echo 5. 退出 +echo ====================== + +set /p choice="请输入选项数字 (1-5)并按下回车以选择: " + +if "!choice!"=="" set choice=1 + +if "!choice!"=="1" goto update_and_start +if "!choice!"=="2" goto start_bot +if "!choice!"=="3" goto config_menu +if "!choice!"=="4" goto tools_menu +if "!choice!"=="5" exit /b + +echo 无效的输入，请输入1-5之间的数字 +timeout /t 2 >nul +goto menu + +:config_menu +@chcp 936 +cls +if not exist config/bot_config.toml ( + copy /Y "template\bot_config_template.toml" "config\bot_config.toml" + +) +if not exist .env.prod ( + copy /Y "template.env" ".env.prod" +) + +start python webui.py + +goto menu + + +:tools_menu +@chcp 936 +cls +echo 麦麦时尚工具箱 当前分支: %BRANCH_COLOR%%BRANCH% +echo ====================== +echo 1. 更新依赖 +echo 2. 切换分支 +echo 3. 重置当前分支 +echo 4. 更新配置文件 +echo 5. 学习新的知识库 +echo 6. 打开知识库文件夹 +echo 7. 返回主菜单 +echo ====================== + +set /p choice="请输入选项数字: " +if "!choice!"=="1" goto update_dependencies +if "!choice!"=="2" goto switch_branch +if "!choice!"=="3" goto reset_branch +if "!choice!"=="4" goto update_config +if "!choice!"=="5" goto learn_new_knowledge +if "!choice!"=="6" goto open_knowledge_folder +if "!choice!"=="7" goto menu + +echo 无效的输入，请输入1-6之间的数字 +timeout /t 2 >nul +goto tools_menu + +:update_dependencies +cls +echo 正在更新依赖... +python -m pip config set global.index-url https://mirrors.aliyun.com/pypi/simple +python.exe -m pip install -r requirements.txt + +echo 依赖更新完成，按任意键返回工具箱菜单... +pause +goto tools_menu + +:switch_branch +cls +echo 正在切换分支... +echo 当前分支: %BRANCH% +@REM echo 可用分支: main, debug, stable-dev +echo 1. 切换到main +echo 2. 切换到main-fix +echo 请输入要切换到的分支: +set /p branch_name="分支名: " +if "%branch_name%"=="" set branch_name=main +if "%branch_name%"=="main" ( + set "BRANCH_COLOR=" +) else if "%branch_name%"=="main-fix" ( + set "BRANCH_COLOR=" +@REM ) else if "%branch_name%"=="stable-dev" ( +@REM set "BRANCH_COLOR=" +) else if "%branch_name%"=="1" ( + set "BRANCH_COLOR=" + set "branch_name=main" +) else if "%branch_name%"=="2" ( + set "BRANCH_COLOR=" + set "branch_name=main-fix" +) else ( + echo 无效的分支名, 请重新输入 + timeout /t 2 >nul + goto switch_branch +) + +echo 正在切换到分支 %branch_name%... +git checkout %branch_name% +echo 分支切换完成，当前分支: %BRANCH_COLOR%%branch_name% +set "BRANCH=%branch_name%" +echo 按任意键返回工具箱菜单... +pause >nul +goto tools_menu + + +:reset_branch +cls +echo 正在重置当前分支... +echo 当前分支: !BRANCH! +echo 确认要重置当前分支吗？ +set /p confirm="继续？(Y/N): " +if /i "!confirm!"=="Y" ( + echo 正在重置当前分支... + git reset --hard !BRANCH! + echo 分支重置完成，按任意键返回工具箱菜单... +) else ( + echo 取消重置当前分支，按任意键返回工具箱菜单... +) +pause >nul +goto tools_menu + + +:update_config +cls +echo 正在更新配置文件... +echo 请确保已备份重要数据，继续将修改当前配置文件。 +echo 继续请按Y，取消请按任意键... +set /p confirm="继续？(Y/N): " +if /i "!confirm!"=="Y" ( + echo 正在更新配置文件... + python.exe config\auto_update.py + echo 配置文件更新完成，按任意键返回工具箱菜单... +) else ( + echo 取消更新配置文件，按任意键返回工具箱菜单... +) +pause >nul +goto tools_menu + +:learn_new_knowledge +cls +echo 正在学习新的知识库... +echo 请确保已备份重要数据，继续将修改当前知识库。 +echo 继续请按Y，取消请按任意键... +set /p confirm="继续？(Y/N): " +if /i "!confirm!"=="Y" ( + echo 正在学习新的知识库... + python.exe src\plugins\zhishi\knowledge_library.py + echo 学习完成，按任意键返回工具箱菜单... +) else ( + echo 取消学习新的知识库，按任意键返回工具箱菜单... +) +pause >nul +goto tools_menu + +:open_knowledge_folder +cls +echo 正在打开知识库文件夹... +if exist data\raw_info ( + start explorer data\raw_info +) else ( + echo 知识库文件夹不存在！ + echo 正在创建文件夹... + mkdir data\raw_info + timeout /t 2 >nul +) +goto tools_menu + + +:update_and_start +cls +:retry_git_pull +git pull > temp.log 2>&1 +findstr /C:"detected dubious ownership" temp.log >nul +if %errorlevel% equ 0 ( + echo 检测到仓库权限问题，正在自动修复... + git config --global --add safe.directory "%cd%" + echo 已添加例外，正在重试git pull... + del temp.log + goto retry_git_pull +) +del temp.log +echo 正在更新依赖... +python -m pip config set global.index-url https://mirrors.aliyun.com/pypi/simple +python -m pip install -r requirements.txt && cls + +echo 当前代理设置: +echo HTTP_PROXY=%HTTP_PROXY% +echo HTTPS_PROXY=%HTTPS_PROXY% + +echo Disable Proxy... +set HTTP_PROXY= +set HTTPS_PROXY= +set no_proxy=0.0.0.0/32 + +REM chcp 65001 +python bot.py +echo. +echo Bot已停止运行，按任意键返回主菜单... +pause >nul +goto menu + +:start_bot +cls +echo 正在更新依赖... +python -m pip config set global.index-url https://mirrors.aliyun.com/pypi/simple +python -m pip install -r requirements.txt && cls + +echo 当前代理设置: +echo HTTP_PROXY=%HTTP_PROXY% +echo HTTPS_PROXY=%HTTPS_PROXY% + +echo Disable Proxy... +set HTTP_PROXY= +set HTTPS_PROXY= +set no_proxy=0.0.0.0/32 + +REM chcp 65001 +python bot.py +echo. +echo Bot已停止运行，按任意键返回主菜单... +pause >nul +goto menu + + +:open_dir +start explorer "%cd%" +goto menu diff --git a/PRIVACY.md b/PRIVACY.md new file mode 100644 index 000000000..ba85f617f --- /dev/null +++ b/PRIVACY.md @@ -0,0 +1,21 @@ +### MaiMBot用户隐私条款 +**版本：V1.0** +**更新日期：2025年3月18日** +**生效日期：2025年3月18日** +**适用的MaiMBot版本号：<=v0.5.15** + +**2025© MaiMBot项目团队** + +MaiMBot项目团队（以下简称项目团队）**尊重并保护**用户（以下简称您）的隐私。若您选择使用MaiMBot项目（以下简称本项目），则您需同意本项目按照以下隐私条款处理您的输入和输出内容： + +**1.1** 本项目**会**收集您的输入和输出内容并发送到第三方API，用于生成新的输出内容。因此您的输入和输出内容**会**同时受到本项目和第三方API的隐私政策约束。 + +**1.2** 本项目**会**收集您的输入和输出内容，用于构建本项目专用的仅存储在您使用的数据库中的知识库和记忆库，以提高回复的准确性和连贯性。 + +**1.3** 本项目**会**收集您的输入和输出内容，用于生成仅存储于您部署或使用的设备中的不会上传至互联网的日志。但当您向项目团队反馈问题时，项目团队可能需要您提供日志文件以帮助解决问题。 + +**1.4** 本项目可能**会**收集部分统计信息（如使用频率、基础指令类型）以改进服务，您可在[bot_config.toml]中随时关闭此功能**。 + +**1.5** 由于您的自身行为或不可抗力等情形，导致上述可能涉及您隐私或您认为是私人信息的内容发生被泄露、批漏，或被第三方获取、使用、转让等情形的，均由您**自行承担**不利后果，我们对此**不承担**任何责任。 + +**1.6** 项目团队保留在未来更新隐私条款的权利，但没有义务通知您。若您不同意更新后的隐私条款，您应立即停止使用本项目。 \ No newline at end of file diff --git a/README.md b/README.md index 183d59fee..5f8f75627 100644 --- a/README.md +++ b/README.md @@ -1,3 +1,81 @@ +# 关于项目分支调整与贡献指南的重要通知 +
+ + - 📂 致所有为麦麦提交过贡献，以及想要为麦麦提交贡献的朋友们！ + +--- + +**📢 关于项目分支调整与贡献指南的重要通知** +**致所有关注MaiMBot的开发者与贡献者：** + +首先，我们由衷感谢大家近期的热情参与！感谢大家对MaiMBot的喜欢，项目突然受到广泛关注让我们倍感惊喜，也深深感受到开源社区的温暖力量。为了保障项目长期健康发展，我们不得不对开发流程做出重要调整，恳请理解与支持。 + +--- + +### **📌 本次调整的核心原因** + +1. **维护团队精力有限** + 核心成员（包括我本人）均为在校学生/在职开发者，近期涌入的大量PR和意见已远超我们的处理能力。为确保本职工作与项目质量，我们必须优化协作流程。 + +2. **重构核心架构的紧迫性** + 当前我们正与核心团队全力重构项目底层逻辑，这是为未来扩展性、性能提升打下的必要基础，需要高度专注。 + +3. **保障现有用户的稳定性** + 我们深知许多用户已依赖当前版本，因此必须划分清晰的维护边界，确保生产环境可用性。 + +--- + +### **🌿 全新分支策略与贡献指南** + +为平衡上述目标，即日起启用以下分支结构： + +| 分支 | 定位 | 接受PR类型 | 提交对象 | +| ---------- | ---------------------------- | --------------------------------------------- | ---------------- | +| `main` | **稳定版**（供下载使用） | 仅接受来自`main-fix`的合并 | 维护团队直接管理 | +| `main-fix` | 生产环境紧急修复 | 明确的功能缺陷修复（需附带复现步骤/测试用例） | 所有开发者 | +| `refactor` | 重构版（**不兼容当前main**） | 仅重构与相关Bug修复 | 重构小组维护 | + +--- + +### **⚠️ 对现有PR的处理说明** + +由于分支结构调整，**GitHub已自动关闭所有未合并的PR**，这并非否定您的贡献价值！如果您认为自己的PR符合以下条件： + +- 属于`main-fix`明确的**功能性缺陷修复**（非功能增强） ，包括非预期行为和严重报错，需要发布issue讨论确定。 +- 属于`refactor`分支的**重构适配性修复** + +**欢迎您重新提交到对应分支**，并在PR描述中标注`[Re-submit from closed PR]`，我们将优先审查。其他类型PR暂缓受理，但您的创意我们已记录在案，未来重构完成后将重新评估。 + +--- + +### **🙏 致谢与协作倡议** + +- 感谢每一位提交Issue、PR、参与讨论的开发者！您的每一行代码都是maim吃的 +- 特别致敬在交流群中积极答疑的社区成员，你们自发维护的氛围令人感动❤️ ，maim哭了 +- **重构期间的非代码贡献同样珍贵**：文档改进、测试用例补充、用户反馈整理等，欢迎通过Issue认领任务！ + +--- + +### **📬 高效协作小贴士** + +1. **提交前请先讨论**：创建Issue描述问题，确认是否符合`main-fix`修复范围 +2. **对重构提出您的想法**：如果您对重构版有自己的想法，欢迎提交讨论issue亟需测试伙伴，欢迎邮件联系`team@xxx.org`报名 +3. **部分main-fix的功能在issue讨论后，经过严格讨论，一致决定可以添加功能改动或修复的，可以提交pr** + +--- + +**谢谢大家谢谢大家谢谢大家谢谢大家谢谢大家谢谢大家！** +虽然此刻不得不放缓脚步，但这一切都是为了跳得更高。期待在重构完成后与各位共建更强大的版本！ + +千石可乐 敬上 +2025年3月14日 + +
+ + + + + # 麦麦！MaiMBot (编辑中)
@@ -21,8 +99,6 @@ > [!WARNING] > 注意，3月12日的v0.5.13, 该版本更新较大，建议单独开文件夹部署，然后转移/data文件 和数据库，数据库可能需要删除messages下的内容（不需要删除记忆） - -
麦麦演示视频 @@ -39,7 +115,19 @@ > - 由于持续迭代，可能存在一些已知或未知的bug > - 由于开发中，可能消耗较多token -## 💬交流群 +**📚 有热心网友创作的wiki:** https://maimbot.pages.dev/ + +**📚 由SLAPQ制作的B站教程:** https://www.bilibili.com/opus/1041609335464001545 + +**😊 其他平台版本** + +- (由 [CabLate](https://github.com/cablate) 贡献) [Telegram 与其他平台(未来可能会有)的版本](https://github.com/cablate/MaiMBot/tree/telegram) - [集中讨论串](https://github.com/SengokuCola/MaiMBot/discussions/149) + +## ✍️如何给本项目报告BUG/提交建议/做贡献 + +MaiMBot是一个开源项目，我们非常欢迎你的参与。你的贡献，无论是提交bug报告、功能需求还是代码pr，都对项目非常宝贵。我们非常感谢你的支持！🎉 但无序的讨论会降低沟通效率，进而影响问题的解决速度，因此在提交任何贡献前，请务必先阅读本项目的[贡献指南](CONTRIBUTE.md) + +### 💬交流群 - [一群](https://qm.qq.com/q/VQ3XZrWgMs) 766798517 ，建议加下面的（开发和建议相关讨论）不一定有空回复，会优先写文档和代码 - [二群](https://qm.qq.com/q/RzmCiRtHEW) 571780722 （开发和建议相关讨论）不一定有空回复，会优先写文档和代码 - [三群](https://qm.qq.com/q/wlH5eT8OmQ) 1035228475（开发和建议相关讨论）不一定有空回复，会优先写文档和代码 @@ -47,19 +135,6 @@ -**📚 有热心网友创作的wiki:** https://maimbot.pages.dev/ - - -**😊 其他平台版本** - -- (由 [CabLate](https://github.com/cablate) 贡献) [Telegram 与其他平台(未来可能会有)的版本](https://github.com/cablate/MaiMBot/tree/telegram) - [集中讨论串](https://github.com/SengokuCola/MaiMBot/discussions/149) - - - -## 📝 注意注意注意注意注意注意注意注意注意注意注意注意注意注意注意注意注意 -**如果你有想法想要提交pr** -- 由于本项目在快速迭代和功能调整，并且有重构计划，目前不接受任何未经过核心开发组讨论的pr合并，谢谢！如您仍旧希望提交pr，可以详情请看置顶issue -

📚 文档 ⬇️ 快速开始使用麦麦 ⬇️

@@ -78,8 +153,6 @@ - [🐳 Docker部署指南](docs/docker_deploy.md) - - ### 配置说明 - [🎀 新手配置指南](docs/installation_cute.md) - 通俗易懂的配置教程，适合初次使用的猫娘 @@ -155,23 +228,24 @@ ## 设计理念 -- **千石可乐说：** -- 这个项目最初只是为了给牛牛bot添加一点额外的功能，但是功能越写越多，最后决定重写。其目的是为了创造一个活跃在QQ群聊的"生命体"。可以目的并不是为了写一个功能齐全的机器人，而是一个尽可能让人感知到真实的类人存在. -- 程序的功能设计理念基于一个核心的原则："最像而不是好" -- 主打一个陪伴 -- 如果人类真的需要一个AI来陪伴自己，并不是所有人都需要一个完美的，能解决所有问题的helpful assistant，而是一个会犯错的，拥有自己感知和想法的"生命形式"。 -- 代码会保持开源和开放，但个人希望MaiMbot的运行时数据保持封闭，尽量避免以显式命令来对其进行控制和调试.我认为一个你无法完全掌控的个体才更能让你感觉到它的自主性，而视其成为一个对话机器. +> **千石可乐说：** +> - 这个项目最初只是为了给牛牛bot添加一点额外的功能，但是功能越写越多，最后决定重写。其目的是为了创造一个活跃在QQ群聊的"生命体"。可以目的并不是为了写一个功能齐全的机器人，而是一个尽可能让人感知到真实的类人存在. +> - 程序的功能设计理念基于一个核心的原则："最像而不是好" +> - 主打一个陪伴 +> - 如果人类真的需要一个AI来陪伴自己，并不是所有人都需要一个完美的，能解决所有问题的helpful assistant，而是一个会犯错的，拥有自己感知和想法的"生命形式"。 +> - 代码会保持开源和开放，但个人希望MaiMbot的运行时数据保持封闭，尽量避免以显式命令来对其进行控制和调试.我认为一个你无法完全掌控的个体才更能让你感觉到它的自主性，而视其成为一个对话机器. ## 📌 注意事项 SengokuCola~~纯编程外行，面向cursor编程，很多代码写得不好多多包涵~~已得到大脑升级 + > [!WARNING] > 本应用生成内容来自人工智能模型，由 AI 生成，请仔细甄别，请勿用于违反法律的用途，AI生成内容不代表本人观点和立场。 ## 致谢 -[nonebot2](https://github.com/nonebot/nonebot2): 跨平台 Python 异步聊天机器人框架 -[NapCat](https://github.com/NapNeko/NapCatQQ): 现代化的基于 NTQQ 的 Bot 协议端实现 +- [nonebot2](https://github.com/nonebot/nonebot2): 跨平台 Python 异步聊天机器人框架 +- [NapCat](https://github.com/NapNeko/NapCatQQ): 现代化的基于 NTQQ 的 Bot 协议端实现 ### 贡献者 diff --git a/bot.py b/bot.py index 7a97f485e..e8f3ae806 100644 --- a/bot.py +++ b/bot.py @@ -1,7 +1,9 @@ import asyncio +import hashlib import os import shutil import sys +from pathlib import Path import nonebot import time @@ -10,10 +12,11 @@ import uvicorn from dotenv import load_dotenv from nonebot.adapters.onebot.v11 import Adapter import platform -from src.plugins.utils.logger_config import LogModule, LogClassification +from src.common.logger import get_module_logger -# 配置日志格式 +# 配置主程序日志格式 +logger = get_module_logger("main_bot") # 获取没有加载env时的环境变量 env_mask = {key: os.getenv(key) for key in os.environ} @@ -76,11 +79,11 @@ def init_env(): def load_env(): # 使用闭包实现对加载器的横向扩展，避免大量重复判断 def prod(): - logger.success("加载生产环境变量配置") + logger.success("成功加载生产环境变量配置") load_dotenv(".env.prod", override=True) # override=True 允许覆盖已存在的环境变量 def dev(): - logger.success("加载开发环境变量配置") + logger.success("成功加载开发环境变量配置") load_dotenv(".env.dev", override=True) # override=True 允许覆盖已存在的环境变量 fn_map = {"prod": prod, "dev": dev} @@ -100,11 +103,6 @@ def load_env(): RuntimeError(f"ENVIRONMENT 配置错误，请检查 .env 文件中的 ENVIRONMENT 变量及对应 .env.{env} 是否存在") -def load_logger(): - global logger # 使得bot.py中其他函数也能调用 - log_module = LogModule() - logger = log_module.setup_logger(LogClassification.BASE) - def scan_provider(env_config: dict): provider = {} @@ -168,13 +166,84 @@ async def uvicorn_main(): uvicorn_server = server await server.serve() +def check_eula(): + eula_confirm_file = Path("eula.confirmed") + privacy_confirm_file = Path("privacy.confirmed") + eula_file = Path("EULA.md") + privacy_file = Path("PRIVACY.md") + + eula_updated = True + eula_new_hash = None + privacy_updated = True + privacy_new_hash = None + + eula_confirmed = False + privacy_confirmed = False + + # 首先计算当前EULA文件的哈希值 + if eula_file.exists(): + with open(eula_file, "r", encoding="utf-8") as f: + eula_content = f.read() + eula_new_hash = hashlib.md5(eula_content.encode("utf-8")).hexdigest() + else: + logger.error("EULA.md 文件不存在") + raise FileNotFoundError("EULA.md 文件不存在") + + # 首先计算当前隐私条款文件的哈希值 + if privacy_file.exists(): + with open(privacy_file, "r", encoding="utf-8") as f: + privacy_content = f.read() + privacy_new_hash = hashlib.md5(privacy_content.encode("utf-8")).hexdigest() + else: + logger.error("PRIVACY.md 文件不存在") + raise FileNotFoundError("PRIVACY.md 文件不存在") + + # 检查EULA确认文件是否存在 + if eula_confirm_file.exists(): + with open(eula_confirm_file, "r", encoding="utf-8") as f: + confirmed_content = f.read() + if eula_new_hash == confirmed_content: + eula_confirmed = True + eula_updated = False + + # 检查隐私条款确认文件是否存在 + if privacy_confirm_file.exists(): + with open(privacy_confirm_file, "r", encoding="utf-8") as f: + confirmed_content = f.read() + if privacy_new_hash == confirmed_content: + privacy_confirmed = True + privacy_updated = False + + # 如果EULA或隐私条款有更新，提示用户重新确认 + if eula_updated or privacy_updated: + print("EULA或隐私条款内容已更新，请在阅读后重新确认，继续运行视为同意更新后的以上两款协议") + print('输入"同意"或"confirmed"继续运行') + while True: + user_input = input().strip().lower() + if user_input in ['同意', 'confirmed']: + # print("确认成功，继续运行") + # print(f"确认成功，继续运行{eula_updated} {privacy_updated}") + if eula_updated: + print(f"更新EULA确认文件{eula_new_hash}") + eula_confirm_file.write_text(eula_new_hash,encoding="utf-8") + if privacy_updated: + print(f"更新隐私条款确认文件{privacy_new_hash}") + privacy_confirm_file.write_text(privacy_new_hash,encoding="utf-8") + break + else: + print('请输入"同意"或"confirmed"以继续运行') + return + elif eula_confirmed and privacy_confirmed: + return def raw_main(): # 利用 TZ 环境变量设定程序工作的时区 # 仅保证行为一致，不依赖 localtime()，实际对生产环境几乎没有作用 if platform.system().lower() != "windows": time.tzset() - + + check_eula() + print("检查EULA和隐私条款完成") easter_egg() init_config() init_env() @@ -206,8 +275,6 @@ def raw_main(): if __name__ == "__main__": try: - # 配置日志，使得主程序直接退出时候也能访问logger - load_logger() raw_main() app = nonebot.get_asgi() diff --git a/changelog.md b/changelog.md index 73803d714..193d81303 100644 --- a/changelog.md +++ b/changelog.md @@ -1,6 +1,64 @@ # Changelog AI总结 +## [0.5.15] - 2025-3-17 +### 🌟 核心功能增强 +#### 关系系统升级 +- 新增关系系统构建与启用功能 +- 优化关系管理系统 +- 改进prompt构建器结构 + +#### 启动器优化 +- 新增MaiLauncher.bat 1.0版本 +- 优化Python和Git环境检测逻辑 +- 添加虚拟环境检查功能 +- 改进工具箱菜单选项 +- 新增分支重置功能 +- 添加MongoDB支持 +- 优化脚本逻辑 + +#### 日志系统改进 +- 新增GUI日志查看器 +- 重构日志工厂处理机制 +- 优化日志级别配置 +- 支持环境变量配置日志级别 +- 改进控制台日志输出 + +### 💻 系统架构优化 +#### 配置系统升级 +- 更新配置文件到0.0.10版本 +- 优化配置文件可视化编辑 +- 新增配置文件版本检测功能 +- 改进配置文件保存机制 +- 修复重复保存可能清空list内容的bug + +#### 部署支持扩展 +- 优化Docker构建流程 +- 改进MongoDB服务启动逻辑 +- 完善Windows脚本支持 + +### 🐛 问题修复 +#### 功能稳定性 +- 修复bot无法识别at对象和reply对象的问题 +- 修复每次从数据库读取额外加0.5的问题 +- 修复新版本由于版本判断不能启动的问题 +- 修复配置文件更新和学习知识库的确认逻辑 +- 优化token统计功能 + +### 📚 文档更新 +- 更新CLAUDE.md为高信息密度项目文档 +- 添加mermaid系统架构图和模块依赖图 +- 添加核心文件索引和类功能表格 +- 添加消息处理流程图 +- 优化文档结构 + +### 主要改进方向 +1. 完善关系系统功能 +2. 优化启动器和部署流程 +3. 改进日志系统 +4. 提升配置系统稳定性 +5. 加强文档完整性 + ## [0.5.14] - 2025-3-14 ### 🌟 核心功能增强 #### 记忆系统优化 @@ -48,8 +106,6 @@ AI总结 4. 改进日志和错误处理 5. 加强部署文档的完整性 - - ## [0.5.13] - 2025-3-12 ### 🌟 核心功能增强 #### 记忆系统升级 diff --git a/config/auto_update.py b/config/auto_update.py index 28ab108da..d87b7c129 100644 --- a/config/auto_update.py +++ b/config/auto_update.py @@ -42,8 +42,16 @@ def update_config(): update_dict(target[key], value) else: try: - # 直接使用tomlkit的item方法创建新值 - target[key] = tomlkit.item(value) + # 对数组类型进行特殊处理 + if isinstance(value, list): + # 如果是空数组，确保它保持为空数组 + if not value: + target[key] = tomlkit.array() + else: + target[key] = tomlkit.array(value) + else: + # 其他类型使用item方法创建新值 + target[key] = tomlkit.item(value) except (TypeError, ValueError): # 如果转换失败，直接赋值 target[key] = value diff --git a/run.sh b/run.sh deleted file mode 100644 index 663fc8a67..000000000 --- a/run.sh +++ /dev/null @@ -1,280 +0,0 @@ -#!/bin/bash - -# Maimbot 一键安装脚本 by Cookie987 -# 适用于Debian系 -# 请小心使用任何一键脚本！ - -# 如无法访问GitHub请修改此处镜像地址 - -LANG=C.UTF-8 - -GITHUB_REPO="https://ghfast.top/https://github.com/SengokuCola/MaiMBot.git" - -# 颜色输出 -GREEN="\e[32m" -RED="\e[31m" -RESET="\e[0m" - -# 需要的基本软件包 -REQUIRED_PACKAGES=("git" "sudo" "python3" "python3-venv" "curl" "gnupg" "python3-pip") - -# 默认项目目录 -DEFAULT_INSTALL_DIR="/opt/maimbot" - -# 服务名称 -SERVICE_NAME="maimbot" - -IS_INSTALL_MONGODB=false -IS_INSTALL_NAPCAT=false - -# 1/6: 检测是否安装 whiptail -if ! command -v whiptail &>/dev/null; then - echo -e "${RED}[1/6] whiptail 未安装，正在安装...${RESET}" - apt update && apt install -y whiptail -fi - -get_os_info() { - if command -v lsb_release &>/dev/null; then - OS_INFO=$(lsb_release -d | cut -f2) - elif [[ -f /etc/os-release ]]; then - OS_INFO=$(grep "^PRETTY_NAME=" /etc/os-release | cut -d '"' -f2) - else - OS_INFO="Unknown OS" - fi - echo "$OS_INFO" -} - -# 检查系统 -check_system() { - # 检查是否为 root 用户 - if [[ "$(id -u)" -ne 0 ]]; then - whiptail --title "🚫 权限不足" --msgbox "请使用 root 用户运行此脚本！\n执行方式: sudo bash $0" 10 60 - exit 1 - fi - - if [[ -f /etc/os-release ]]; then - source /etc/os-release - if [[ "$ID" != "debian" || "$VERSION_ID" != "12" ]]; then - whiptail --title "🚫 不支持的系统" --msgbox "此脚本仅支持 Debian 12 (Bookworm)！\n当前系统: $PRETTY_NAME\n安装已终止。" 10 60 - exit 1 - fi - else - whiptail --title "⚠️ 无法检测系统" --msgbox "无法识别系统版本，安装已终止。" 10 60 - exit 1 - fi -} - -# 3/6: 询问用户是否安装缺失的软件包 -install_packages() { - missing_packages=() - for package in "${REQUIRED_PACKAGES[@]}"; do - if ! dpkg -s "$package" &>/dev/null; then - missing_packages+=("$package") - fi - done - - if [[ ${#missing_packages[@]} -gt 0 ]]; then - whiptail --title "📦 [3/6] 软件包检查" --yesno "检测到以下必须的依赖项目缺失:\n${missing_packages[*]}\n\n是否要自动安装？" 12 60 - if [[ $? -eq 0 ]]; then - return 0 - else - whiptail --title "⚠️ 注意" --yesno "某些必要的依赖项未安装，可能会影响运行！\n是否继续？" 10 60 || exit 1 - fi - fi -} - -# 4/6: Python 版本检查 -check_python() { - PYTHON_VERSION=$(python3 -c 'import sys; print(f"{sys.version_info.major}.{sys.version_info.minor}")') - - python3 -c "import sys; exit(0) if sys.version_info >= (3,9) else exit(1)" - if [[ $? -ne 0 ]]; then - whiptail --title "⚠️ [4/6] Python 版本过低" --msgbox "检测到 Python 版本为 $PYTHON_VERSION，需要 3.9 或以上！\n请升级 Python 后重新运行本脚本。" 10 60 - exit 1 - fi -} - -# 5/6: 选择分支 -choose_branch() { - BRANCH=$(whiptail --title "🔀 [5/6] 选择 Maimbot 分支" --menu "请选择要安装的 Maimbot 分支：" 15 60 2 \ - "main" "稳定版本（推荐，供下载使用）" \ - "main-fix" "生产环境紧急修复" 3>&1 1>&2 2>&3) - - if [[ -z "$BRANCH" ]]; then - BRANCH="main" - whiptail --title "🔀 默认选择" --msgbox "未选择分支，默认安装稳定版本（main）" 10 60 - fi -} - -# 6/6: 选择安装路径 -choose_install_dir() { - INSTALL_DIR=$(whiptail --title "📂 [6/6] 选择安装路径" --inputbox "请输入 Maimbot 的安装目录：" 10 60 "$DEFAULT_INSTALL_DIR" 3>&1 1>&2 2>&3) - - if [[ -z "$INSTALL_DIR" ]]; then - whiptail --title "⚠️ 取消输入" --yesno "未输入安装路径，是否退出安装？" 10 60 - if [[ $? -ne 0 ]]; then - INSTALL_DIR="$DEFAULT_INSTALL_DIR" - else - exit 1 - fi - fi -} - -# 显示确认界面 -confirm_install() { - local confirm_message="请确认以下更改:\n\n" - - if [[ ${#missing_packages[@]} -gt 0 ]]; then - confirm_message+="📦 安装缺失的依赖项: ${missing_packages[*]}\n" - else - confirm_message+="✅ 所有依赖项已安装\n" - fi - - confirm_message+="📂 安装麦麦Bot到: $INSTALL_DIR\n" - confirm_message+="🔀 分支: $BRANCH\n" - - if [[ "$MONGODB_INSTALLED" == "true" ]]; then - confirm_message+="✅ MongoDB 已安装\n" - else - if [[ "$IS_INSTALL_MONGODB" == "true" ]]; then - confirm_message+="📦 安装 MongoDB\n" - fi - fi - - if [[ "$NAPCAT_INSTALLED" == "true" ]]; then - confirm_message+="✅ NapCat 已安装\n" - else - if [[ "$IS_INSTALL_NAPCAT" == "true" ]]; then - confirm_message+="📦 安装 NapCat\n" - fi - fi - - confirm_message+="🛠️ 添加麦麦Bot作为系统服务 ($SERVICE_NAME.service)\n" - - confitm_message+="\n\n注意：本脚本默认使用ghfast.top为GitHub进行加速，如不想使用请手动修改脚本开头的GITHUB_REPO变量。" - whiptail --title "🔧 安装确认" --yesno "$confirm_message\n\n是否继续安装？" 15 60 - if [[ $? -ne 0 ]]; then - whiptail --title "🚫 取消安装" --msgbox "安装已取消。" 10 60 - exit 1 - fi -} - -check_mongodb() { - if command -v mongod &>/dev/null; then - MONGO_INSTALLED=true - else - MONGO_INSTALLED=false - fi -} - -# 安装 MongoDB -install_mongodb() { - if [[ "$MONGO_INSTALLED" == "true" ]]; then - return 0 - fi - - whiptail --title "📦 [3/6] 软件包检查" --yesno "检测到未安装MongoDB，是否安装？\n如果您想使用远程数据库，请跳过此步。" 10 60 - if [[ $? -ne 0 ]]; then - return 1 - fi - IS_INSTALL_MONGODB=true -} - -check_napcat() { - if command -v napcat &>/dev/null; then - NAPCAT_INSTALLED=true - else - NAPCAT_INSTALLED=false - fi -} - -install_napcat() { - if [[ "$NAPCAT_INSTALLED" == "true" ]]; then - return 0 - fi - - whiptail --title "📦 [3/6] 软件包检查" --yesno "检测到未安装NapCat，是否安装？\n如果您想使用远程NapCat，请跳过此步。" 10 60 - if [[ $? -ne 0 ]]; then - return 1 - fi - IS_INSTALL_NAPCAT=true -} - -# 运行安装步骤 -whiptail --title "⚠️ 警告：安装前详阅" --msgbox "项目处于活跃开发阶段，代码可能随时更改\n文档未完善，有问题可以提交 Issue 或者 Discussion\nQQ机器人存在被限制风险，请自行了解，谨慎使用\n由于持续迭代，可能存在一些已知或未知的bug\n由于开发中，可能消耗较多token\n\n本脚本可能更新不及时，如遇到bug请优先尝试手动部署以确定是否为脚本问题" 14 60 - -check_system -check_mongodb -check_napcat -install_packages -install_mongodb -install_napcat -check_python -choose_branch -choose_install_dir -confirm_install - -# 开始安装 -whiptail --title "🚀 开始安装" --msgbox "所有环境检查完毕，即将开始安装麦麦Bot！" 10 60 - -echo -e "${GREEN}安装依赖项...${RESET}" - -apt update && apt install -y "${missing_packages[@]}" - - -if [[ "$IS_INSTALL_MONGODB" == "true" ]]; then - echo -e "${GREEN}安装 MongoDB...${RESET}" - curl -fsSL https://www.mongodb.org/static/pgp/server-8.0.asc | gpg -o /usr/share/keyrings/mongodb-server-8.0.gpg --dearmor - echo "deb [ signed-by=/usr/share/keyrings/mongodb-server-8.0.gpg ] http://repo.mongodb.org/apt/debian bookworm/mongodb-org/8.0 main" | sudo tee /etc/apt/sources.list.d/mongodb-org-8.0.list - apt-get update - apt-get install -y mongodb-org - - systemctl enable mongod - systemctl start mongod -fi - -if [[ "$IS_INSTALL_NAPCAT" == "true" ]]; then - echo -e "${GREEN}安装 NapCat...${RESET}" - curl -o napcat.sh https://nclatest.znin.net/NapNeko/NapCat-Installer/main/script/install.sh && bash napcat.sh --cli y --docker n -fi - -echo -e "${GREEN}创建 Python 虚拟环境...${RESET}" -mkdir -p "$INSTALL_DIR" -cd "$INSTALL_DIR" || exit -python3 -m venv venv -source venv/bin/activate - -echo -e "${GREEN}克隆仓库...${RESET}" -# 安装 Maimbot -mkdir -p "$INSTALL_DIR/repo" -cd "$INSTALL_DIR/repo" || exit 1 -git clone -b "$BRANCH" $GITHUB_REPO . - -echo -e "${GREEN}安装 Python 依赖...${RESET}" -pip install -r requirements.txt - -echo -e "${GREEN}设置服务...${RESET}" - -# 设置 Maimbot 服务 -cat <&1 1>&2 2>&3) + + [[ $? -ne 0 ]] && exit 0 + + case "$choice" in + 1) + systemctl start ${SERVICE_NAME} + whiptail --msgbox "✅麦麦Bot已启动" 10 60 + ;; + 2) + systemctl stop ${SERVICE_NAME} + whiptail --msgbox "🛑麦麦Bot已停止" 10 60 + ;; + 3) + systemctl restart ${SERVICE_NAME} + whiptail --msgbox "🔄麦麦Bot已重启" 10 60 + ;; + 4) + systemctl start ${SERVICE_NAME_WEB} + whiptail --msgbox "✅WebUI已启动" 10 60 + ;; + 5) + systemctl stop ${SERVICE_NAME_WEB} + whiptail --msgbox "🛑WebUI已停止" 10 60 + ;; + 6) + systemctl restart ${SERVICE_NAME_WEB} + whiptail --msgbox "🔄WebUI已重启" 10 60 + ;; + 7) + update_dependencies + ;; + 8) + switch_branch + ;; + 9) + update_config + ;; + 10) + exit 0 + ;; + *) + whiptail --msgbox "无效选项！" 10 60 + ;; + esac + done +} + +# 更新依赖 +update_dependencies() { + cd "${INSTALL_DIR}/repo" || { + whiptail --msgbox "🚫 无法进入安装目录！" 10 60 + return 1 + } + if ! git pull origin "${BRANCH}"; then + whiptail --msgbox "🚫 代码更新失败！" 10 60 + return 1 + fi + source "${INSTALL_DIR}/venv/bin/activate" + if ! pip install -r requirements.txt; then + whiptail --msgbox "🚫 依赖安装失败！" 10 60 + deactivate + return 1 + fi + deactivate + systemctl restart ${SERVICE_NAME} + whiptail --msgbox "✅ 依赖已更新并重启服务！" 10 60 +} + +# 切换分支 +switch_branch() { + new_branch=$(whiptail --inputbox "请输入要切换的分支名称：" 10 60 "${BRANCH}" 3>&1 1>&2 2>&3) + [[ -z "$new_branch" ]] && { + whiptail --msgbox "🚫 分支名称不能为空！" 10 60 + return 1 + } + + cd "${INSTALL_DIR}/repo" || { + whiptail --msgbox "🚫 无法进入安装目录！" 10 60 + return 1 + } + + if ! git ls-remote --exit-code --heads origin "${new_branch}" >/dev/null 2>&1; then + whiptail --msgbox "🚫 分支 ${new_branch} 不存在！" 10 60 + return 1 + fi + + if ! git checkout "${new_branch}"; then + whiptail --msgbox "🚫 分支切换失败！" 10 60 + return 1 + fi + + if ! git pull origin "${new_branch}"; then + whiptail --msgbox "🚫 代码拉取失败！" 10 60 + return 1 + fi + + source "${INSTALL_DIR}/venv/bin/activate" + pip install -r requirements.txt + deactivate + + sed -i "s/^BRANCH=.*/BRANCH=${new_branch}/" /etc/maimbot_install.conf + BRANCH="${new_branch}" + systemctl restart ${SERVICE_NAME} + touch "${INSTALL_DIR}/repo/elua.confirmed" + whiptail --msgbox "✅ 已切换到分支 ${new_branch} 并重启服务！" 10 60 +} + +# 更新配置文件 +update_config() { + cd "${INSTALL_DIR}/repo" || { + whiptail --msgbox "🚫 无法进入安装目录！" 10 60 + return 1 + } + if [[ -f config/bot_config.toml ]]; then + cp config/bot_config.toml config/bot_config.toml.bak + whiptail --msgbox "📁 原配置文件已备份为 bot_config.toml.bak" 10 60 + source "${INSTALL_DIR}/venv/bin/activate" + python3 config/auto_update.py + deactivate + whiptail --msgbox "🆕 已更新配置文件，请重启麦麦Bot！" 10 60 + return 0 + else + whiptail --msgbox "🚫 未找到配置文件 bot_config.toml\n 请先运行一次麦麦Bot" 10 60 + return 1 + fi +} + +# ----------- 主安装流程 ----------- +run_installation() { + # 1/6: 检测是否安装 whiptail + if ! command -v whiptail &>/dev/null; then + echo -e "${RED}[1/6] whiptail 未安装，正在安装...${RESET}" + apt update && apt install -y whiptail + fi + + # 协议确认 + if ! (whiptail --title "ℹ️ [1/6] 使用协议" --yes-button "我同意" --no-button "我拒绝" --yesno "使用麦麦Bot及此脚本前请先阅读ELUA协议\nhttps://github.com/SengokuCola/MaiMBot/blob/main/EULA.md\n\n您是否同意此协议？" 12 70); then + exit 1 + fi + + # 欢迎信息 + whiptail --title "[2/6] 欢迎使用麦麦Bot一键安装脚本 by Cookie987" --msgbox "检测到您未安装麦麦Bot，将自动进入安装流程，安装完成后再次运行此脚本即可进入管理菜单。\n\n项目处于活跃开发阶段，代码可能随时更改\n文档未完善，有问题可以提交 Issue 或者 Discussion\nQQ机器人存在被限制风险，请自行了解，谨慎使用\n由于持续迭代，可能存在一些已知或未知的bug\n由于开发中，可能消耗较多token\n\n本脚本可能更新不及时，如遇到bug请优先尝试手动部署以确定是否为脚本问题" 17 60 + + # 系统检查 + check_system() { + if [[ "$(id -u)" -ne 0 ]]; then + whiptail --title "🚫 权限不足" --msgbox "请使用 root 用户运行此脚本！\n执行方式: sudo bash $0" 10 60 + exit 1 + fi + + if [[ -f /etc/os-release ]]; then + source /etc/os-release + if [[ "$ID" != "debian" || "$VERSION_ID" != "12" ]]; then + whiptail --title "🚫 不支持的系统" --msgbox "此脚本仅支持 Debian 12 (Bookworm)！\n当前系统: $PRETTY_NAME\n安装已终止。" 10 60 + exit 1 + fi + else + whiptail --title "⚠️ 无法检测系统" --msgbox "无法识别系统版本，安装已终止。" 10 60 + exit 1 + fi + } + check_system + + # 检查MongoDB + check_mongodb() { + if command -v mongod &>/dev/null; then + MONGO_INSTALLED=true + else + MONGO_INSTALLED=false + fi + } + check_mongodb + + # 检查NapCat + check_napcat() { + if command -v napcat &>/dev/null; then + NAPCAT_INSTALLED=true + else + NAPCAT_INSTALLED=false + fi + } + check_napcat + + # 安装必要软件包 + install_packages() { + missing_packages=() + for package in "${REQUIRED_PACKAGES[@]}"; do + if ! dpkg -s "$package" &>/dev/null; then + missing_packages+=("$package") + fi + done + + if [[ ${#missing_packages[@]} -gt 0 ]]; then + whiptail --title "📦 [3/6] 软件包检查" --yesno "检测到以下必须的依赖项目缺失:\n${missing_packages[*]}\n\n是否要自动安装？" 12 60 + if [[ $? -eq 0 ]]; then + IS_INSTALL_DEPENDENCIES=true + else + whiptail --title "⚠️ 注意" --yesno "某些必要的依赖项未安装，可能会影响运行！\n是否继续？" 10 60 || exit 1 + fi + fi + } + install_packages + + # 安装MongoDB + install_mongodb() { + [[ $MONGO_INSTALLED == true ]] && return + whiptail --title "📦 [3/6] 软件包检查" --yesno "检测到未安装MongoDB，是否安装？\n如果您想使用远程数据库，请跳过此步。" 10 60 && { + echo -e "${GREEN}安装 MongoDB...${RESET}" + curl -fsSL https://www.mongodb.org/static/pgp/server-8.0.asc | gpg -o /usr/share/keyrings/mongodb-server-8.0.gpg --dearmor + echo "deb [ signed-by=/usr/share/keyrings/mongodb-server-8.0.gpg ] http://repo.mongodb.org/apt/debian bookworm/mongodb-org/8.0 main" | tee /etc/apt/sources.list.d/mongodb-org-8.0.list + apt update + apt install -y mongodb-org + systemctl enable --now mongod + IS_INSTALL_MONGODB=true + } + } + install_mongodb + + # 安装NapCat + install_napcat() { + [[ $NAPCAT_INSTALLED == true ]] && return + whiptail --title "📦 [3/6] 软件包检查" --yesno "检测到未安装NapCat，是否安装？\n如果您想使用远程NapCat，请跳过此步。" 10 60 && { + echo -e "${GREEN}安装 NapCat...${RESET}" + curl -o napcat.sh https://nclatest.znin.net/NapNeko/NapCat-Installer/main/script/install.sh && bash napcat.sh --cli y --docker n + IS_INSTALL_NAPCAT=true + } + } + install_napcat + + # Python版本检查 + check_python() { + PYTHON_VERSION=$(python3 -c 'import sys; print(f"{sys.version_info.major}.{sys.version_info.minor}")') + if ! python3 -c "import sys; exit(0) if sys.version_info >= (3,9) else exit(1)"; then + whiptail --title "⚠️ [4/6] Python 版本过低" --msgbox "检测到 Python 版本为 $PYTHON_VERSION，需要 3.9 或以上！\n请升级 Python 后重新运行本脚本。" 10 60 + exit 1 + fi + } + check_python + + # 选择分支 + choose_branch() { + BRANCH=$(whiptail --title "🔀 [5/6] 选择麦麦Bot分支" --menu "请选择要安装的麦麦Bot分支：" 15 60 2 \ + "main" "稳定版本（推荐，供下载使用）" \ + "main-fix" "生产环境紧急修复" 3>&1 1>&2 2>&3) + [[ -z "$BRANCH" ]] && BRANCH="main" + } + choose_branch + + # 选择安装路径 + choose_install_dir() { + INSTALL_DIR=$(whiptail --title "📂 [6/6] 选择安装路径" --inputbox "请输入麦麦Bot的安装目录：" 10 60 "$DEFAULT_INSTALL_DIR" 3>&1 1>&2 2>&3) + [[ -z "$INSTALL_DIR" ]] && { + whiptail --title "⚠️ 取消输入" --yesno "未输入安装路径，是否退出安装？" 10 60 && exit 1 + INSTALL_DIR="$DEFAULT_INSTALL_DIR" + } + } + choose_install_dir + + # 确认安装 + confirm_install() { + local confirm_msg="请确认以下信息：\n\n" + confirm_msg+="📂 安装麦麦Bot到: $INSTALL_DIR\n" + confirm_msg+="🔀 分支: $BRANCH\n" + [[ $IS_INSTALL_DEPENDENCIES == true ]] && confirm_msg+="📦 安装依赖：${missing_packages}\n" + [[ $IS_INSTALL_MONGODB == true || $IS_INSTALL_NAPCAT == true ]] && confirm_msg+="📦 安装额外组件：\n" + + [[ $IS_INSTALL_MONGODB == true ]] && confirm_msg+=" - MongoDB\n" + [[ $IS_INSTALL_NAPCAT == true ]] && confirm_msg+=" - NapCat\n" + confirm_msg+="\n注意：本脚本默认使用ghfast.top为GitHub进行加速，如不想使用请手动修改脚本开头的GITHUB_REPO变量。" + + whiptail --title "🔧 安装确认" --yesno "$confirm_msg" 16 60 || exit 1 + } + confirm_install + + # 开始安装 + echo -e "${GREEN}安装依赖...${RESET}" + [[ $IS_INSTALL_DEPENDENCIES == true ]] && apt update && apt install -y "${missing_packages[@]}" + + echo -e "${GREEN}创建安装目录...${RESET}" + mkdir -p "$INSTALL_DIR" + cd "$INSTALL_DIR" || exit 1 + + echo -e "${GREEN}设置Python虚拟环境...${RESET}" + python3 -m venv venv + source venv/bin/activate + + echo -e "${GREEN}克隆仓库...${RESET}" + git clone -b "$BRANCH" "$GITHUB_REPO" repo || { + echo -e "${RED}克隆仓库失败！${RESET}" + exit 1 + } + + echo -e "${GREEN}安装Python依赖...${RESET}" + pip install -r repo/requirements.txt + + echo -e "${GREEN}同意协议...${RESET}" + touch repo/elua.confirmed + + echo -e "${GREEN}创建系统服务...${RESET}" + cat > /etc/systemd/system/${SERVICE_NAME}.service < /etc/systemd/system/${SERVICE_NAME_WEB}.service < /etc/maimbot_install.conf + echo "INSTALL_DIR=${INSTALL_DIR}" >> /etc/maimbot_install.conf + echo "BRANCH=${BRANCH}" >> /etc/maimbot_install.conf + + whiptail --title "🎉 安装完成" --msgbox "麦麦Bot安装完成！\n已创建系统服务：${SERVICE_NAME}，${SERVICE_NAME_WEB}\n\n使用以下命令管理服务：\n启动服务：systemctl start ${SERVICE_NAME}\n查看状态：systemctl status ${SERVICE_NAME}" 14 60 +} + +# ----------- 主执行流程 ----------- +# 检查root权限 +[[ $(id -u) -ne 0 ]] && { + echo -e "${RED}请使用root用户运行此脚本！${RESET}" + exit 1 +} + +# 如果已安装显示菜单 +if check_installed; then + load_install_info + show_menu +else + run_installation + # 安装完成后询问是否启动 + if whiptail --title "安装完成" --yesno "是否立即启动麦麦Bot服务？" 10 60; then + systemctl start ${SERVICE_NAME} + whiptail --msgbox "✅ 服务已启动！\n使用 systemctl status ${SERVICE_NAME} 查看状态" 10 60 + fi +fi diff --git a/src/common/logger.py b/src/common/logger.py new file mode 100644 index 000000000..143fe9f95 --- /dev/null +++ b/src/common/logger.py @@ -0,0 +1,372 @@ +from loguru import logger +from typing import Dict, Optional, Union, List +import sys +import os +from types import ModuleType +from pathlib import Path +from dotenv import load_dotenv +# from ..plugins.chat.config import global_config + +load_dotenv() + +# 保存原生处理器ID +default_handler_id = None +for handler_id in logger._core.handlers: + default_handler_id = handler_id + break + +# 移除默认处理器 +if default_handler_id is not None: + logger.remove(default_handler_id) + +# 类型别名 +LoguruLogger = logger.__class__ + +# 全局注册表：记录模块与处理器ID的映射 +_handler_registry: Dict[str, List[int]] = {} + +# 获取日志存储根地址 +current_file_path = Path(__file__).resolve() +LOG_ROOT = "logs" + +# 从环境变量获取是否启用高级输出 +# ENABLE_ADVANCE_OUTPUT = True +ENABLE_ADVANCE_OUTPUT = False + +if ENABLE_ADVANCE_OUTPUT: + # 默认全局配置 + DEFAULT_CONFIG = { + # 日志级别配置 + "console_level": "INFO", + "file_level": "DEBUG", + + # 格式配置 + "console_format": ( + "{time:YYYY-MM-DD HH:mm:ss} | " + "{level: <8} | " + "{extra[module]: <12} | " + "{message}" + ), + "file_format": ( + "{time:YYYY-MM-DD HH:mm:ss} | " + "{level: <8} | " + "{extra[module]: <15} | " + "{message}" + ), + "log_dir": LOG_ROOT, + "rotation": "00:00", + "retention": "3 days", + "compression": "zip", + } +else: + DEFAULT_CONFIG = { + # 日志级别配置 + "console_level": "INFO", + "file_level": "DEBUG", + + # 格式配置 + "console_format": ( + "{time:MM-DD HH:mm} | " + "{extra[module]} | " + "{message}" + ), + "file_format": ( + "{time:YYYY-MM-DD HH:mm:ss} | " + "{level: <8} | " + "{extra[module]: <15} | " + "{message}" + ), + "log_dir": LOG_ROOT, + "rotation": "00:00", + "retention": "3 days", + "compression": "zip", + } + +# 控制nonebot日志输出的环境变量 +NONEBOT_LOG_ENABLED = False + +# 海马体日志样式配置 +MEMORY_STYLE_CONFIG = { + "advanced": { + "console_format": ( + "{time:YYYY-MM-DD HH:mm:ss} | " + "{level: <8} | " + "{extra[module]: <12} | " + "海马体 | " + "{message}" + ), + "file_format": ( + "{time:YYYY-MM-DD HH:mm:ss} | " + "{level: <8} | " + "{extra[module]: <15} | " + "海马体 | " + "{message}" + ) + }, + "simple": { + "console_format": ( + "{time:MM-DD HH:mm} | " + "海马体 | " + "{message}" + ), + "file_format": ( + "{time:YYYY-MM-DD HH:mm:ss} | " + "{level: <8} | " + "{extra[module]: <15} | " + "海马体 | " + "{message}" + ) + } +} + +# 海马体日志样式配置 +SENDER_STYLE_CONFIG = { + "advanced": { + "console_format": ( + "{time:YYYY-MM-DD HH:mm:ss} | " + "{level: <8} | " + "{extra[module]: <12} | " + "消息发送 | " + "{message}" + ), + "file_format": ( + "{time:YYYY-MM-DD HH:mm:ss} | " + "{level: <8} | " + "{extra[module]: <15} | " + "消息发送 | " + "{message}" + ) + }, + "simple": { + "console_format": ( + "{time:MM-DD HH:mm} | " + "消息发送 | " + "{message}" + ), + "file_format": ( + "{time:YYYY-MM-DD HH:mm:ss} | " + "{level: <8} | " + "{extra[module]: <15} | " + "消息发送 | " + "{message}" + ) + } +} + +LLM_STYLE_CONFIG = { + "advanced": { + "console_format": ( + "{time:YYYY-MM-DD HH:mm:ss} | " + "{level: <8} | " + "{extra[module]: <12} | " + "麦麦组织语言 | " + "{message}" + ), + "file_format": ( + "{time:YYYY-MM-DD HH:mm:ss} | " + "{level: <8} | " + "{extra[module]: <15} | " + "麦麦组织语言 | " + "{message}" + ) + }, + "simple": { + "console_format": ( + "{time:MM-DD HH:mm} | " + "麦麦组织语言 | " + "{message}" + ), + "file_format": ( + "{time:YYYY-MM-DD HH:mm:ss} | " + "{level: <8} | " + "{extra[module]: <15} | " + "麦麦组织语言 | " + "{message}" + ) + } +} + + + +# Topic日志样式配置 +TOPIC_STYLE_CONFIG = { + "advanced": { + "console_format": ( + "{time:YYYY-MM-DD HH:mm:ss} | " + "{level: <8} | " + "{extra[module]: <12} | " + "话题 | " + "{message}" + ), + "file_format": ( + "{time:YYYY-MM-DD HH:mm:ss} | " + "{level: <8} | " + "{extra[module]: <15} | " + "话题 | " + "{message}" + ) + }, + "simple": { + "console_format": ( + "{time:MM-DD HH:mm} | " + "主题 | " + "{message}" + ), + "file_format": ( + "{time:YYYY-MM-DD HH:mm:ss} | " + "{level: <8} | " + "{extra[module]: <15} | " + "话题 | " + "{message}" + ) + } +} + +# 根据ENABLE_ADVANCE_OUTPUT选择配置 +MEMORY_STYLE_CONFIG = MEMORY_STYLE_CONFIG["advanced"] if ENABLE_ADVANCE_OUTPUT else MEMORY_STYLE_CONFIG["simple"] +TOPIC_STYLE_CONFIG = TOPIC_STYLE_CONFIG["advanced"] if ENABLE_ADVANCE_OUTPUT else TOPIC_STYLE_CONFIG["simple"] +SENDER_STYLE_CONFIG = SENDER_STYLE_CONFIG["advanced"] if ENABLE_ADVANCE_OUTPUT else SENDER_STYLE_CONFIG["simple"] +LLM_STYLE_CONFIG = LLM_STYLE_CONFIG["advanced"] if ENABLE_ADVANCE_OUTPUT else LLM_STYLE_CONFIG["simple"] + +def filter_nonebot(record: dict) -> bool: + """过滤nonebot的日志""" + return record["extra"].get("module") != "nonebot" + +def is_registered_module(record: dict) -> bool: + """检查是否为已注册的模块""" + return record["extra"].get("module") in _handler_registry + +def is_unregistered_module(record: dict) -> bool: + """检查是否为未注册的模块""" + return not is_registered_module(record) + +def log_patcher(record: dict) -> None: + """自动填充未设置模块名的日志记录，保留原生模块名称""" + if "module" not in record["extra"]: + # 尝试从name中提取模块名 + module_name = record.get("name", "") + if module_name == "": + module_name = "root" + record["extra"]["module"] = module_name + +# 应用全局修补器 +logger.configure(patcher=log_patcher) + +class LogConfig: + """日志配置类""" + + def __init__(self, **kwargs): + self.config = DEFAULT_CONFIG.copy() + self.config.update(kwargs) + + def to_dict(self) -> dict: + return self.config.copy() + + def update(self, **kwargs): + self.config.update(kwargs) + + +def get_module_logger( + module: Union[str, ModuleType], + *, + console_level: Optional[str] = None, + file_level: Optional[str] = None, + extra_handlers: Optional[List[dict]] = None, + config: Optional[LogConfig] = None +) -> LoguruLogger: + module_name = module if isinstance(module, str) else module.__name__ + current_config = config.config if config else DEFAULT_CONFIG + + # 清理旧处理器 + if module_name in _handler_registry: + for handler_id in _handler_registry[module_name]: + logger.remove(handler_id) + del _handler_registry[module_name] + + handler_ids = [] + + # 控制台处理器 + console_id = logger.add( + sink=sys.stderr, + level=os.getenv("CONSOLE_LOG_LEVEL", console_level or current_config["console_level"]), + format=current_config["console_format"], + filter=lambda record: record["extra"].get("module") == module_name, + enqueue=True, + ) + handler_ids.append(console_id) + + # 文件处理器 + log_dir = Path(current_config["log_dir"]) + log_dir.mkdir(parents=True, exist_ok=True) + log_file = log_dir / module_name / f"{{time:YYYY-MM-DD}}.log" + log_file.parent.mkdir(parents=True, exist_ok=True) + + file_id = logger.add( + sink=str(log_file), + level=os.getenv("FILE_LOG_LEVEL", file_level or current_config["file_level"]), + format=current_config["file_format"], + rotation=current_config["rotation"], + retention=current_config["retention"], + compression=current_config["compression"], + encoding="utf-8", + filter=lambda record: record["extra"].get("module") == module_name, + enqueue=True, + ) + handler_ids.append(file_id) + + # 额外处理器 + if extra_handlers: + for handler in extra_handlers: + handler_id = logger.add(**handler) + handler_ids.append(handler_id) + + # 更新注册表 + _handler_registry[module_name] = handler_ids + + return logger.bind(module=module_name) + + +def remove_module_logger(module_name: str) -> None: + """清理指定模块的日志处理器""" + if module_name in _handler_registry: + for handler_id in _handler_registry[module_name]: + logger.remove(handler_id) + del _handler_registry[module_name] + + +# 添加全局默认处理器（只处理未注册模块的日志--->控制台） +DEFAULT_GLOBAL_HANDLER = logger.add( + sink=sys.stderr, + level=os.getenv("DEFAULT_CONSOLE_LOG_LEVEL", "SUCCESS"), + format=( + "{time:YYYY-MM-DD HH:mm:ss} | " + "{level: <8} | " + "{name: <12} | " + "{message}" + ), + filter=lambda record: is_unregistered_module(record) and filter_nonebot(record), # 只处理未注册模块的日志，并过滤nonebot + enqueue=True, +) + +# 添加全局默认文件处理器（只处理未注册模块的日志--->logs文件夹） +log_dir = Path(DEFAULT_CONFIG["log_dir"]) +log_dir.mkdir(parents=True, exist_ok=True) +other_log_dir = log_dir / "other" +other_log_dir.mkdir(parents=True, exist_ok=True) + +DEFAULT_FILE_HANDLER = logger.add( + sink=str(other_log_dir / f"{{time:YYYY-MM-DD}}.log"), + level=os.getenv("DEFAULT_FILE_LOG_LEVEL", "DEBUG"), + format=( + "{time:YYYY-MM-DD HH:mm:ss} | " + "{level: <8} | " + "{name: <15} | " + "{message}" + ), + rotation=DEFAULT_CONFIG["rotation"], + retention=DEFAULT_CONFIG["retention"], + compression=DEFAULT_CONFIG["compression"], + encoding="utf-8", + filter=lambda record: is_unregistered_module(record) and filter_nonebot(record), # 只处理未注册模块的日志，并过滤nonebot + enqueue=True, +) diff --git a/src/gui/logger_gui.py b/src/gui/logger_gui.py new file mode 100644 index 000000000..f2dd698cd --- /dev/null +++ b/src/gui/logger_gui.py @@ -0,0 +1,347 @@ +import customtkinter as ctk +import subprocess +import threading +import queue +import re +import os +import signal +from collections import deque + +# 设置应用的外观模式和默认颜色主题 +ctk.set_appearance_mode("dark") +ctk.set_default_color_theme("blue") + + +class LogViewerApp(ctk.CTk): + """日志查看器应用的主类，继承自customtkinter的CTk类""" + + def __init__(self): + """初始化日志查看器应用的界面和状态""" + super().__init__() + self.title("日志查看器") + self.geometry("1200x800") + + # 初始化进程、日志队列、日志数据等变量 + self.process = None + self.log_queue = queue.Queue() + self.log_data = deque(maxlen=10000) # 使用固定长度队列 + self.available_levels = set() + self.available_modules = set() + self.sorted_modules = [] + self.module_checkboxes = {} # 存储模块复选框的字典 + + # 日志颜色配置 + self.color_config = { + "time": "#888888", + "DEBUG": "#2196F3", + "INFO": "#4CAF50", + "WARNING": "#FF9800", + "ERROR": "#F44336", + "module": "#D4D0AB", + "default": "#FFFFFF", + } + + # 列可见性配置 + self.column_visibility = {"show_time": True, "show_level": True, "show_module": True} + + # 选中的日志等级和模块 + self.selected_levels = set() + self.selected_modules = set() + + # 创建界面组件并启动日志队列处理 + self.create_widgets() + self.after(100, self.process_log_queue) + + def create_widgets(self): + """创建应用界面的各个组件""" + self.grid_columnconfigure(0, weight=1) + self.grid_rowconfigure(1, weight=1) + + # 控制面板 + control_frame = ctk.CTkFrame(self) + control_frame.grid(row=0, column=0, sticky="ew", padx=10, pady=5) + + self.start_btn = ctk.CTkButton(control_frame, text="启动", command=self.start_process) + self.start_btn.pack(side="left", padx=5) + + self.stop_btn = ctk.CTkButton(control_frame, text="停止", command=self.stop_process, state="disabled") + self.stop_btn.pack(side="left", padx=5) + + self.clear_btn = ctk.CTkButton(control_frame, text="清屏", command=self.clear_logs) + self.clear_btn.pack(side="left", padx=5) + + column_filter_frame = ctk.CTkFrame(control_frame) + column_filter_frame.pack(side="left", padx=20) + + self.time_check = ctk.CTkCheckBox(column_filter_frame, text="显示时间", command=self.refresh_logs) + self.time_check.pack(side="left", padx=5) + self.time_check.select() + + self.level_check = ctk.CTkCheckBox(column_filter_frame, text="显示等级", command=self.refresh_logs) + self.level_check.pack(side="left", padx=5) + self.level_check.select() + + self.module_check = ctk.CTkCheckBox(column_filter_frame, text="显示模块", command=self.refresh_logs) + self.module_check.pack(side="left", padx=5) + self.module_check.select() + + # 筛选面板 + filter_frame = ctk.CTkFrame(self) + filter_frame.grid(row=0, column=1, rowspan=2, sticky="ns", padx=5) + + ctk.CTkLabel(filter_frame, text="日志等级筛选").pack(pady=5) + self.level_scroll = ctk.CTkScrollableFrame(filter_frame, width=150, height=200) + self.level_scroll.pack(fill="both", expand=True, padx=5) + + ctk.CTkLabel(filter_frame, text="模块筛选").pack(pady=5) + self.module_filter_entry = ctk.CTkEntry(filter_frame, placeholder_text="输入模块过滤词") + self.module_filter_entry.pack(pady=5) + self.module_filter_entry.bind("", self.update_module_filter) + + self.module_scroll = ctk.CTkScrollableFrame(filter_frame, width=300, height=200) + self.module_scroll.pack(fill="both", expand=True, padx=5) + + self.log_text = ctk.CTkTextbox(self, wrap="word") + self.log_text.grid(row=1, column=0, sticky="nsew", padx=10, pady=5) + + self.init_text_tags() + + def update_module_filter(self, event): + """根据模块过滤词更新模块复选框的显示""" + filter_text = self.module_filter_entry.get().strip().lower() + for module, checkbox in self.module_checkboxes.items(): + if filter_text in module.lower(): + checkbox.pack(anchor="w", padx=5, pady=2) + else: + checkbox.pack_forget() + + def update_filters(self, level, module): + """更新日志等级和模块的筛选器""" + if level not in self.available_levels: + self.available_levels.add(level) + self.add_checkbox(self.level_scroll, level, "level") + + module_key = self.get_module_key(module) + if module_key not in self.available_modules: + self.available_modules.add(module_key) + self.sorted_modules = sorted(self.available_modules, key=lambda x: x.lower()) + self.rebuild_module_checkboxes() + + def rebuild_module_checkboxes(self): + """重新构建模块复选框""" + # 清空现有复选框 + for widget in self.module_scroll.winfo_children(): + widget.destroy() + self.module_checkboxes.clear() + + # 重建排序后的复选框 + for module in self.sorted_modules: + self.add_checkbox(self.module_scroll, module, "module") + + def add_checkbox(self, parent, text, type_): + """在指定父组件中添加复选框""" + + def update_filter(): + current = cb.get() + if type_ == "level": + (self.selected_levels.add if current else self.selected_levels.discard)(text) + else: + (self.selected_modules.add if current else self.selected_modules.discard)(text) + self.refresh_logs() + + cb = ctk.CTkCheckBox(parent, text=text, command=update_filter) + cb.select() # 初始选中 + + # 手动同步初始状态到集合（关键修复） + if type_ == "level": + self.selected_levels.add(text) + else: + self.selected_modules.add(text) + + if type_ == "module": + self.module_checkboxes[text] = cb + cb.pack(anchor="w", padx=5, pady=2) + return cb + + def check_filter(self, entry): + """检查日志条目是否符合当前筛选条件""" + level_ok = not self.selected_levels or entry["level"] in self.selected_levels + module_key = self.get_module_key(entry["module"]) + module_ok = not self.selected_modules or module_key in self.selected_modules + return level_ok and module_ok + + def init_text_tags(self): + """初始化日志文本的颜色标签""" + for tag, color in self.color_config.items(): + self.log_text.tag_config(tag, foreground=color) + self.log_text.tag_config("default", foreground=self.color_config["default"]) + + def start_process(self): + """启动日志进程并开始读取输出""" + self.process = subprocess.Popen( + ["nb", "run"], + stdout=subprocess.PIPE, + stderr=subprocess.STDOUT, + text=True, + bufsize=1, + encoding="utf-8", + errors="ignore", + ) + self.start_btn.configure(state="disabled") + self.stop_btn.configure(state="normal") + threading.Thread(target=self.read_output, daemon=True).start() + + def stop_process(self): + """停止日志进程并清理相关资源""" + if self.process: + try: + if hasattr(self.process, "pid"): + if os.name == "nt": + subprocess.run( + ["taskkill", "/F", "/T", "/PID", str(self.process.pid)], check=True, capture_output=True + ) + else: + os.killpg(os.getpgid(self.process.pid), signal.SIGTERM) + except (subprocess.CalledProcessError, ProcessLookupError, OSError) as e: + print(f"终止进程失败: {e}") + finally: + self.process = None + self.log_queue.queue.clear() + self.start_btn.configure(state="normal") + self.stop_btn.configure(state="disabled") + self.refresh_logs() + + def read_output(self): + """读取日志进程的输出并放入队列""" + try: + while self.process and self.process.poll() is None: + line = self.process.stdout.readline() + if line: + self.log_queue.put(line) + else: + break # 避免空循环 + self.process.stdout.close() # 确保关闭文件描述符 + except ValueError: # 处理可能的I/O操作异常 + pass + + def process_log_queue(self): + """处理日志队列中的日志条目""" + while not self.log_queue.empty(): + line = self.log_queue.get() + self.process_log_line(line) + self.after(100, self.process_log_queue) + + def process_log_line(self, line): + """解析单行日志并更新日志数据和筛选器""" + match = re.match( + r"""^ + (?:(?P