初始化多平台机器人部署项目 - Telegram/Discord/QQ资源搜索机器人

2026-09-11 11:19:56 +08:00
commit 2253a2d774
21 changed files with 4004 additions and 0 deletions
@@ -0,0 +1,143 @@
+# tg-search-bot
+
+[English](README.md)｜[中文](README.zh.md)
+
+一个基于 Python3 的 Telegram 资源搜索机器人：搜索各类视频磁力链接，支持收藏、导出记录、自动保存到网盘，可配置 NSFW 开关与代理上网；同时集成 AI 自然语言理解能力，可直接根据用户自然语言意图自动识别搜索目标、类型与数据源。
+
+- 数据来源：TorrentKitty（中文）+ apibay（英文）+ 影视磁力索引 API（Jvav）
+- 网盘：Pikpak 官方 OpenAPI
+- AI：任意 OpenAI 兼容接口，支持意图识别与自动搜索
+
+## 功能
+
+- 按编号 / 演职人员 / 关键词搜索影片，返回封面、评分、日期、标签、演职人员及磁力链接（支持 HD / 字幕等过滤）
+- 获取预览视频、完整视频与影片截图
+- 收藏与导出记录（`record.json`）
+- 排行榜、随机高分 / 最新影片
+- 自动把最优磁力链接保存到 Pikpak（登录 token 直传离线任务）
+- AI 自然语言理解搜索：识别编号 / 演职人员 / 片名 / 剧情 / 类型并自动选择数据源
+- BT 种子搜索（中文 / 英文双引擎），结果分页
+- 多轮追问：`换一批` / `下一页` / `上一页` / `保存第N个`
+- 可配置 NSFW 开关与代理上网
+
+## 核心流程（AI 自然语言搜索）
+
+搜索流程：
+
+1. 用户输入想要的资源 / 资源相关信息（电影、电视剧名称、影片编号、剧情 / 对话、演职人员、影片类型）
+2. AI 根据输入理解用户目标（根据剧情推测片名、根据类型推荐等），输出 `intent / target / source / explain`
+3. 机器人回复：`🔍 搜索：{目标}`
+4. AI 选择最合适的爬虫 / API 进行搜索（编号 → 编号索引站，演职人员 → 演职人员搜索，其余 → 关键词或 BT）
+5. 机器人回复：`⏳ 搜索中 ...`
+6. 爬虫 / API 搜索得到磁力链接（没有磁力时返回 BT 种子等其他格式内容）
+7. 机器人回复：`📄 搜索结果`
+8. 程序将最优磁力链接保存到已配置的网盘（PikPak）
+9. 机器人回复：`✅ 保存结果：已提交离线任务到 PikPak`
+
+> 未配置 AI 接口时自动回退到原来的编号识别 + BT 关键词搜索逻辑。
+
+```mermaid
+flowchart TD
+    A["① 用户输入<br/>电影 / 电视剧 / 编号 / 剧情 / 演职人员 / 类型"] --> B["② AI 理解意图<br/>intent · target · source · explain"]
+    B --> C["③ 回复 🔍 搜索：目标"]
+    C --> D["④ 选择最合适的爬虫 / API"]
+    D --> D1{意图 intent}
+    D1 -->|编号| E["按编号搜索<br/>编号索引站"]
+    D1 -->|演职人员| F["演职人员搜索"]
+    D1 -->|关键词 / 类型 / 剧情| G["关键词 → BT 种子"]
+    E --> H["⑤ 回复 ⏳ 搜索中"]
+    F --> H
+    G --> H
+    H --> I["⑥ 搜索得到磁力链接<br/>（无磁力则返回 BT 等其他格式）"]
+    I --> J["⑦ 回复 📄 搜索结果"]
+    J --> K["⑧ 保存最优磁力链接到 PikPak"]
+    K --> L["⑨ 回复 ✅ 保存结果"]
+```
+
+## 目录结构
+
+```
+bot.py              主程序：配置、日志、消息 / 回调处理、AI 搜索流程
+ai.py               AI 意图理解客户端（OpenAI 兼容 /chat/completions）
+pikpak.py           Pikpak 官方 OpenAPI 客户端（登录 / 刷新 token、离线下载）
+database.py         数据层：BotFileDb（JSON 收藏）+ BotCacheDb（SQLite 缓存）
+requirements.txt    依赖
+docker-compose.yml  一键部署
+```
+
+配置与数据文件存放于 `~/.tg_search_bot/`：`config.yaml`（主配置）、`record.json`（收藏）、`cache.db`（缓存）、`pikpak_token.json`（Pikpak 登录态）、`log.txt`（日志）。
+
+## 使用
+
+### 1. 配置
+
+编辑 `~/.tg_search_bot/config.yaml`：
+
+```yaml
+# 必填，你的 Telegram chat id
+tg_chat_id: 
+# 必填，你的 Telegram bot token
+tg_bot_token: 
+# 必填，是否启用全局代理，1 是 / 0 否
+use_proxy: 
+# 可选，代理地址（use_proxy 为 1 时必填），如 http://127.0.0.1:7890
+proxy_addr: 
+# 必填，是否开启 NSFW 内容，1 是 / 0 否
+enable_nsfw: 0
+# 可选，Pikpak 账号（用于自动保存磁力到网盘，官方 API）
+pikpak_username: 
+pikpak_password: 
+# 可选，AI 自然语言搜索（任意 OpenAI 兼容接口）
+ai_base_url: 
+ai_api_key: 
+ai_model: 
+```
+
+### 2. 运行
+
+```sh
+# 方式一：Docker 一键部署
+docker-compose up -d
+
+# 方式二：直接运行（Python >= 3.9）
+pip install -r requirements.txt
+python3 bot.py
+```
+
+### 3. 命令
+
+| 命令 | 说明 |
+| --- | --- |
+| `/help` | 查看帮助 |
+| `/stars` | 查看已收藏演职人员 |
+| `/ids` | 查看已收藏编号 |
+| `/record` | 导出收藏记录文件 |
+
+直接发送电影名 / 关键词 / 编号 / 剧情 / 演员名 / 类型即可，AI 会自动识别并搜索。
+
+### 4. 多轮追问
+
+BT 搜索结果每页 5 条，可继续回复：
+
+- `换一批` / `下一页` / `还有吗` — 下一页
+- `上一页` — 上一页
+- `保存第N个`（或 `save N`）— 把第 N 条磁力保存到 Pikpak
+
+## 开发
+
+推荐使用 Python 3.9，并建议使用虚拟环境：
+
+```sh
+git clone https://github.com/akynazh/tg-search-bot.git
+cd tg-search-bot
+python3 -m venv .venv
+source ./.venv/bin/activate
+pip3 install -r requirements.txt
+```
+
+## 致谢
+
+<a href="https://www.jetbrains.com/">
+<img src="https://resources.jetbrains.com/storage/products/company/brand/logos/jb_beam.png" alt="JetBrains Logo (Main) logo." style="width: 200px;"></a>
+
+感谢 JetBrains 对这个项目的支持！