Compare commits
15 Commits
perf/provi
...
feat/vpo-t
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
c56d2aeced | ||
|
|
fb56ac9f47 | ||
|
|
5719dbd8b2 | ||
|
|
5217450c8a | ||
|
|
15aafa2043 | ||
|
|
36a3b4d318 | ||
|
|
7518c4a057 | ||
|
|
8e3bd92c09 | ||
|
|
9e4fec8488 | ||
|
|
983264dc1a | ||
|
|
b3f23b23da | ||
|
|
212d8cc101 | ||
|
|
36de3c541a | ||
|
|
1151677f13 | ||
|
|
2ccb85d802 |
2
.env.example
Normal file
2
.env.example
Normal file
@@ -0,0 +1,2 @@
|
||||
# ASTRBOT 数据目录
|
||||
# ASTRBOT_ROOT = ./data
|
||||
98
.github/copilot-instructions.md
vendored
98
.github/copilot-instructions.md
vendored
@@ -1,63 +1,63 @@
|
||||
# AstrBot Development Instructions
|
||||
# AstrBot 开发指南
|
||||
|
||||
AstrBot is a multi-platform LLM chatbot and development framework written in Python with a Vue.js dashboard. It supports multiple messaging platforms (QQ, Telegram, Discord, etc.) and various LLM providers (OpenAI, Anthropic, Google Gemini, etc.).
|
||||
AstrBot 是一个使用 Python 编写、配备 Vue.js 仪表盘的多平台 LLM 聊天机器人开发框架。它支持多个消息平台(QQ、Telegram、Discord 等)和多种 LLM 提供商(OpenAI、Anthropic、Google Gemini 等)。
|
||||
|
||||
Always reference these instructions first and fallback to search or bash commands only when you encounter unexpected information that does not match the info here.
|
||||
始终优先参考这些指南,仅在遇到与此处信息不符的意外情况时才回退到搜索或 bash 命令。
|
||||
|
||||
## Working Effectively
|
||||
## 高效工作
|
||||
|
||||
### Bootstrap and Install Dependencies
|
||||
- **Python 3.10+ required** - Check `.python-version` file
|
||||
- Install UV package manager: `pip install uv`
|
||||
- Install project dependencies: `uv sync` -- takes 6-7 minutes. NEVER CANCEL. Set timeout to 10+ minutes.
|
||||
- Create required directories: `mkdir -p data/plugins data/config data/temp`
|
||||
### 引导和安装依赖
|
||||
- **需要 Python 3.10+** - 检查 `.python-version` 文件
|
||||
- 安装 UV 包管理器:`pip install uv`
|
||||
- 安装项目依赖:`uv sync` -- 很快几分钟。绝不要取消。设置超时时间为 10+ 分钟。
|
||||
- 创建必需的目录:`mkdir -p data/plugins data/config data/temp`
|
||||
|
||||
### Running the Application
|
||||
- Run main application: `uv run main.py` -- starts in ~3 seconds
|
||||
- Application creates WebUI on http://localhost:6185 (default credentials: `astrbot`/`astrbot`)
|
||||
- Application loads plugins automatically from `packages/` and `data/plugins/` directories
|
||||
### 运行应用程序
|
||||
- 运行主应用程序:`uv run main.py` -- 约 3 秒启动
|
||||
- 应用程序在 http://localhost:6185 创建 WebUI(默认凭据:`astrbot`/`astrbot`)
|
||||
- 应用程序自动从 `packages/` 和 `data/plugins/` 目录加载插件
|
||||
|
||||
### Dashboard Build (Vue.js/Node.js)
|
||||
- **Prerequisites**: Node.js 20+ and npm 10+ required
|
||||
- Navigate to dashboard: `cd dashboard`
|
||||
- Install dashboard dependencies: `npm install` -- takes 2-3 minutes. NEVER CANCEL. Set timeout to 5+ minutes.
|
||||
- Build dashboard: `npm run build` -- takes 25-30 seconds. NEVER CANCEL.
|
||||
- Dashboard creates optimized production build in `dashboard/dist/`
|
||||
### 仪表盘构建(Vue.js/Node.js)
|
||||
- **前置要求**:需要 Node.js 20+ 和 npm 10+
|
||||
- 导航到仪表盘:`cd dashboard`
|
||||
- 安装仪表盘依赖:`npm install` -- 需要 2-3 分钟。绝不要取消。设置超时时间为 5+ 分钟。
|
||||
- 构建仪表盘:`npm run build` -- 需要 25-30 秒。绝不要取消。
|
||||
- 仪表盘在 `dashboard/dist/` 创建优化的生产构建
|
||||
|
||||
### Testing
|
||||
- Do not generate test files for now.
|
||||
### 测试
|
||||
- 暂时不要生成测试文件。
|
||||
|
||||
### Code Quality and Linting
|
||||
- Install ruff linter: `uv add --dev ruff`
|
||||
- Check code style: `uv run ruff check .` -- takes <1 second
|
||||
- Check formatting: `uv run ruff format --check .` -- takes <1 second
|
||||
- Fix formatting: `uv run ruff format .`
|
||||
- **ALWAYS** run `uv run ruff check .` and `uv run ruff format .` before committing changes
|
||||
### 代码质量和检查
|
||||
- 安装 ruff 检查器:`uv add --dev ruff`
|
||||
- 检查代码风格:`uv run ruff check .` -- 耗时 <1 秒
|
||||
- 检查格式:`uv run ruff format --check .` -- 耗时 <1 秒
|
||||
- 修复格式:`uv run ruff format .`
|
||||
- **始终**在提交更改前运行 `uv run ruff check .` 和 `uv run ruff format .`
|
||||
|
||||
### Plugin Development
|
||||
- Plugins load from `packages/` (built-in) and `data/plugins/` (user-installed)
|
||||
- Plugin system supports function tools and message handlers
|
||||
- Key plugins: python_interpreter, web_searcher, astrbot, reminder, session_controller
|
||||
### 插件开发
|
||||
- 插件从 `packages/`(内置)和 `data/plugins/`(用户安装)加载
|
||||
- 插件系统支持函数工具和消息处理器
|
||||
- 关键插件:python_interpreter、web_searcher、astrbot、reminder、session_controller
|
||||
|
||||
### Common Issues and Workarounds
|
||||
- **Dashboard download fails**: Known issue with "division by zero" error - application still works
|
||||
- **Import errors in tests**: Ensure `uv run` is used to run tests in proper environment
|
||||
=- **Build timeouts**: Always set appropriate timeouts (10+ minutes for uv sync, 5+ minutes for npm install)
|
||||
### 常见问题和解决方法
|
||||
- **仪表盘下载失败**:已知的"除以零"错误问题 - 应用程序仍可正常工作
|
||||
- **测试中的导入错误**:确保使用 `uv run` 在适当的环境中运行测试
|
||||
- **构建超时**:始终设置适当的超时时间(uv sync 为 10+ 分钟,npm install 为 5+ 分钟)
|
||||
|
||||
## CI/CD Integration
|
||||
- GitHub Actions workflows in `.github/workflows/`
|
||||
- Docker builds supported via `Dockerfile`
|
||||
- Pre-commit hooks enforce ruff formatting and linting
|
||||
## CI/CD 集成
|
||||
- GitHub Actions 工作流在 `.github/workflows/` 中
|
||||
- 通过 `Dockerfile` 支持 Docker 构建
|
||||
- Pre-commit 钩子强制执行 ruff 格式化和检查
|
||||
|
||||
## Docker Support
|
||||
- Primary deployment method: `docker run soulter/astrbot:latest`
|
||||
- Compose file available: `compose.yml`
|
||||
- Exposes ports: 6185 (WebUI), 6195 (WeChat), 6199 (QQ), etc.
|
||||
- Volume mount required: `./data:/AstrBot/data`
|
||||
## Docker 支持
|
||||
- 主要部署方法:`docker run soulter/astrbot:latest`
|
||||
- 可用的 Compose 文件:`compose.yml`
|
||||
- 暴露端口:6185(WebUI)、6195(WeChat)、6199(QQ)等
|
||||
- 需要挂载卷:`./data:/AstrBot/data`
|
||||
|
||||
## Multi-language Support
|
||||
- Documentation in Chinese (README.md), English (README_en.md), Japanese (README_ja.md)
|
||||
- UI supports internationalization
|
||||
- Default language is Chinese
|
||||
## 多语言支持
|
||||
- 文档包括中文(README.md)、英文(README_en.md)、日文(README_ja.md)
|
||||
- UI 支持国际化
|
||||
- 默认语言为中文
|
||||
|
||||
Remember: This is a production chatbot framework with real users. Always test thoroughly and ensure changes don't break existing functionality.
|
||||
请记住:这是一个有真实用户的生产聊天机器人框架。始终进行彻底测试,确保更改不会破坏现有功能。
|
||||
|
||||
97
astrbot/__main__.py
Normal file
97
astrbot/__main__.py
Normal file
@@ -0,0 +1,97 @@
|
||||
import argparse
|
||||
import asyncio
|
||||
import mimetypes
|
||||
import os
|
||||
import sys
|
||||
|
||||
from astrbot.core import LogBroker, LogManager, db_helper, logger
|
||||
from astrbot.core.config.default import VERSION
|
||||
from astrbot.core.initial_loader import InitialLoader
|
||||
from astrbot.core.utils.io import download_dashboard, get_dashboard_version
|
||||
from astrbot_api import LOGO, IAstrbotPaths
|
||||
from astrbot_sdk import sync_base_container
|
||||
|
||||
AstrbotPaths: type[IAstrbotPaths] = sync_base_container.get(type[IAstrbotPaths])
|
||||
|
||||
def check_env() -> None:
|
||||
if not (sys.version_info.major == 3 and sys.version_info.minor >= 10):
|
||||
logger.error("请使用 Python3.10+ 运行本项目。")
|
||||
exit()
|
||||
|
||||
# os.makedirs("data/config", exist_ok=True)
|
||||
# os.makedirs("data/plugins", exist_ok=True)
|
||||
# os.makedirs("data/temp", exist_ok=True)
|
||||
|
||||
# 针对问题 #181 的临时解决方案
|
||||
mimetypes.add_type("text/javascript", ".js")
|
||||
mimetypes.add_type("text/javascript", ".mjs")
|
||||
mimetypes.add_type("application/json", ".json")
|
||||
|
||||
|
||||
async def check_dashboard_files(webui_dir: str | None = None):
|
||||
"""下载管理面板文件"""
|
||||
# 指定webui目录
|
||||
if webui_dir:
|
||||
if os.path.exists(webui_dir):
|
||||
logger.info(f"使用指定的 WebUI 目录: {webui_dir}")
|
||||
return webui_dir
|
||||
logger.warning(f"指定的 WebUI 目录 {webui_dir} 不存在,将使用默认逻辑。")
|
||||
|
||||
data_dist_path = str(AstrbotPaths.astrbot_root / "dist")
|
||||
if os.path.exists(data_dist_path):
|
||||
v = await get_dashboard_version()
|
||||
if v is not None:
|
||||
# 存在文件
|
||||
if v == f"v{VERSION}":
|
||||
logger.info("WebUI 版本已是最新。")
|
||||
else:
|
||||
logger.warning(
|
||||
f"检测到 WebUI 版本 ({v}) 与当前 AstrBot 版本 (v{VERSION}) 不符。",
|
||||
)
|
||||
return data_dist_path
|
||||
|
||||
logger.info(
|
||||
"开始下载管理面板文件...高峰期(晚上)可能导致较慢的速度。如多次下载失败,请前往 https://github.com/AstrBotDevs/AstrBot/releases/latest 下载 dist.zip,并将其中的 dist 文件夹解压至 data 目录下。",
|
||||
)
|
||||
|
||||
try:
|
||||
await download_dashboard(version=f"v{VERSION}", latest=False)
|
||||
except Exception as e:
|
||||
logger.critical(f"下载管理面板文件失败: {e}。")
|
||||
return None
|
||||
|
||||
logger.info("管理面板下载完成。")
|
||||
return data_dist_path
|
||||
|
||||
|
||||
def main():
|
||||
parser = argparse.ArgumentParser(description="AstrBot")
|
||||
parser.add_argument(
|
||||
"--webui-dir",
|
||||
type=str,
|
||||
help="指定 WebUI 静态文件目录路径",
|
||||
default=None,
|
||||
)
|
||||
args = parser.parse_args()
|
||||
|
||||
check_env()
|
||||
|
||||
# 启动日志代理
|
||||
log_broker = LogBroker()
|
||||
LogManager.set_queue_handler(logger, log_broker)
|
||||
|
||||
# 检查仪表板文件
|
||||
webui_dir = asyncio.run(check_dashboard_files(args.webui_dir))
|
||||
|
||||
db = db_helper
|
||||
|
||||
# 打印 logo
|
||||
logger.info(LOGO)
|
||||
|
||||
core_lifecycle = InitialLoader(db, log_broker)
|
||||
core_lifecycle.webui_dir = webui_dir
|
||||
asyncio.run(core_lifecycle.start())
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
@@ -1 +1 @@
|
||||
__version__ = "3.5.23"
|
||||
"""AstrBot CLI入口"""
|
||||
|
||||
@@ -1,27 +1,22 @@
|
||||
"""AstrBot CLI入口"""
|
||||
|
||||
import sys
|
||||
from importlib.metadata import version
|
||||
|
||||
import click
|
||||
|
||||
from . import __version__
|
||||
from astrbot_api import LOGO
|
||||
|
||||
from .commands import conf, init, plug, run
|
||||
|
||||
logo_tmpl = r"""
|
||||
___ _______.___________..______ .______ ______ .___________.
|
||||
/ \ / | || _ \ | _ \ / __ \ | |
|
||||
/ ^ \ | (----`---| |----`| |_) | | |_) | | | | | `---| |----`
|
||||
/ /_\ \ \ \ | | | / | _ < | | | | | |
|
||||
/ _____ \ .----) | | | | |\ \----.| |_) | | `--' | | |
|
||||
/__/ \__\ |_______/ |__| | _| `._____||______/ \______/ |__|
|
||||
"""
|
||||
__version__ = version("astrbot")
|
||||
|
||||
|
||||
@click.group()
|
||||
@click.version_option(__version__, prog_name="AstrBot")
|
||||
def cli() -> None:
|
||||
"""The AstrBot CLI"""
|
||||
click.echo(logo_tmpl)
|
||||
click.echo(LOGO)
|
||||
click.echo("Welcome to AstrBot CLI!")
|
||||
click.echo(f"AstrBot CLI version: {__version__}")
|
||||
|
||||
|
||||
@@ -3,6 +3,11 @@ import shutil
|
||||
from pathlib import Path
|
||||
|
||||
import click
|
||||
from astrbot_api.abc import IAstrbotPaths
|
||||
|
||||
from astrbot_sdk import sync_base_container
|
||||
|
||||
AstrbotPaths: type[IAstrbotPaths] = sync_base_container.get(type[IAstrbotPaths])
|
||||
|
||||
from ..utils import (
|
||||
PluginStatus,
|
||||
@@ -47,8 +52,7 @@ def display_plugins(plugins, title=None, color=None):
|
||||
@click.argument("name")
|
||||
def new(name: str):
|
||||
"""创建新插件"""
|
||||
base_path = _get_data_path()
|
||||
plug_path = base_path / "plugins" / name
|
||||
plug_path = AstrbotPaths.getPaths(name).plugins
|
||||
|
||||
if plug_path.exists():
|
||||
raise click.ClickException(f"插件 {name} 已存在")
|
||||
|
||||
@@ -1,22 +1,31 @@
|
||||
import warnings
|
||||
from pathlib import Path
|
||||
|
||||
import click
|
||||
from astrbot_api.abc import IAstrbotPaths
|
||||
|
||||
from astrbot_sdk import sync_base_container
|
||||
|
||||
AstrbotPaths = sync_base_container.get(type[IAstrbotPaths])
|
||||
|
||||
def check_astrbot_root(path: str | Path) -> bool:
|
||||
"""检查路径是否为 AstrBot 根目录"""
|
||||
if not isinstance(path, Path):
|
||||
path = Path(path)
|
||||
if not path.exists() or not path.is_dir():
|
||||
return False
|
||||
if not (path / ".astrbot").exists():
|
||||
return False
|
||||
return True
|
||||
warnings.warn(
|
||||
"请使用 AstrbotPaths 类代替本模块中的函数",
|
||||
DeprecationWarning,
|
||||
stacklevel=2,
|
||||
)
|
||||
return AstrbotPaths.is_root(Path(path))
|
||||
|
||||
|
||||
def get_astrbot_root() -> Path:
|
||||
"""获取Astrbot根目录路径"""
|
||||
return Path.cwd()
|
||||
warnings.warn(
|
||||
"请使用 AstrbotPaths 类代替本模块中的函数",
|
||||
DeprecationWarning,
|
||||
stacklevel=2,
|
||||
)
|
||||
return AstrbotPaths.astrbot_root
|
||||
|
||||
|
||||
async def check_dashboard(astrbot_root: Path) -> None:
|
||||
|
||||
@@ -9,10 +9,6 @@ from astrbot.core.utils.shared_preferences import SharedPreferences
|
||||
from astrbot.core.utils.t2i.renderer import HtmlRenderer
|
||||
|
||||
from .log import LogBroker, LogManager # noqa
|
||||
from .utils.astrbot_path import get_astrbot_data_path
|
||||
|
||||
# 初始化数据存储文件夹
|
||||
os.makedirs(get_astrbot_data_path(), exist_ok=True)
|
||||
|
||||
DEMO_MODE = os.getenv("DEMO_MODE", False)
|
||||
|
||||
|
||||
@@ -3,11 +3,15 @@ import json
|
||||
import logging
|
||||
import os
|
||||
|
||||
from astrbot.core.utils.astrbot_path import get_astrbot_data_path
|
||||
from astrbot_api.abc import IAstrbotPaths
|
||||
|
||||
from astrbot_sdk import sync_base_container
|
||||
|
||||
from .default import DEFAULT_CONFIG, DEFAULT_VALUE_MAP
|
||||
|
||||
ASTRBOT_CONFIG_PATH = os.path.join(get_astrbot_data_path(), "cmd_config.json")
|
||||
AstrbotPaths: type[IAstrbotPaths] = sync_base_container.get(type[IAstrbotPaths])
|
||||
|
||||
ASTRBOT_CONFIG_PATH = str(AstrbotPaths.astrbot_root / "cmd_config.json")
|
||||
logger = logging.getLogger("astrbot")
|
||||
|
||||
|
||||
|
||||
@@ -1,11 +1,17 @@
|
||||
"""如需修改配置,请在 `data/cmd_config.json` 中修改或者在管理面板中可视化修改。"""
|
||||
|
||||
import os
|
||||
from importlib.metadata import version
|
||||
|
||||
from astrbot.core.utils.astrbot_path import get_astrbot_data_path
|
||||
from astrbot_api.abc import IAstrbotPaths
|
||||
|
||||
VERSION = "4.5.1"
|
||||
DB_PATH = os.path.join(get_astrbot_data_path(), "data_v4.db")
|
||||
from astrbot_sdk import sync_base_container
|
||||
|
||||
AstrbotPaths: type[IAstrbotPaths] = sync_base_container.get(type[IAstrbotPaths])
|
||||
|
||||
# 警告,请使用version函数获取版本,此变量兼容保留
|
||||
VERSION = version("astrbot")
|
||||
|
||||
DB_PATH = str(AstrbotPaths.astrbot_root / "data_v4.db")
|
||||
|
||||
# 默认配置
|
||||
DEFAULT_CONFIG = {
|
||||
|
||||
@@ -21,20 +21,20 @@ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
||||
SOFTWARE.
|
||||
"""
|
||||
|
||||
import asyncio
|
||||
import base64
|
||||
import json
|
||||
import os
|
||||
import uuid
|
||||
from enum import Enum
|
||||
|
||||
from astrbot_api.abc import IAstrbotPaths
|
||||
from pydantic.v1 import BaseModel
|
||||
|
||||
from astrbot.core import astrbot_config, file_token_service, logger
|
||||
from astrbot.core.utils.astrbot_path import get_astrbot_data_path
|
||||
from astrbot.core.utils.io import download_file, download_image_by_url, file_to_base64
|
||||
from astrbot_sdk import sync_base_container
|
||||
|
||||
|
||||
AstrbotPaths: type[IAstrbotPaths] = sync_base_container.get(type[IAstrbotPaths])
|
||||
class ComponentType(str, Enum):
|
||||
# Basic Segment Types
|
||||
Plain = "Plain" # plain text message
|
||||
@@ -153,8 +153,7 @@ class Record(BaseMessageComponent):
|
||||
if self.file.startswith("base64://"):
|
||||
bs64_data = self.file.removeprefix("base64://")
|
||||
image_bytes = base64.b64decode(bs64_data)
|
||||
temp_dir = os.path.join(get_astrbot_data_path(), "temp")
|
||||
file_path = os.path.join(temp_dir, f"{uuid.uuid4()}.jpg")
|
||||
file_path = str(AstrbotPaths.astrbot_root / "temp" / f"{uuid.uuid4()}.jpg")
|
||||
with open(file_path, "wb") as f:
|
||||
f.write(image_bytes)
|
||||
return os.path.abspath(file_path)
|
||||
@@ -242,8 +241,9 @@ class Video(BaseMessageComponent):
|
||||
if url and url.startswith("file:///"):
|
||||
return url[8:]
|
||||
if url and url.startswith("http"):
|
||||
download_dir = os.path.join(get_astrbot_data_path(), "temp")
|
||||
video_file_path = os.path.join(download_dir, f"{uuid.uuid4().hex}")
|
||||
video_file_path = str(
|
||||
AstrbotPaths.astrbot_root / "temp" / f"{uuid.uuid4().hex}"
|
||||
)
|
||||
await download_file(url, video_file_path)
|
||||
if os.path.exists(video_file_path):
|
||||
return os.path.abspath(video_file_path)
|
||||
@@ -442,8 +442,9 @@ class Image(BaseMessageComponent):
|
||||
if url.startswith("base64://"):
|
||||
bs64_data = url.removeprefix("base64://")
|
||||
image_bytes = base64.b64decode(bs64_data)
|
||||
temp_dir = os.path.join(get_astrbot_data_path(), "temp")
|
||||
image_file_path = os.path.join(temp_dir, f"{uuid.uuid4()}.jpg")
|
||||
image_file_path = str(
|
||||
AstrbotPaths.astrbot_root / "temp" / f"{uuid.uuid4()}.jpg"
|
||||
)
|
||||
with open(image_file_path, "wb") as f:
|
||||
f.write(image_bytes)
|
||||
return os.path.abspath(image_file_path)
|
||||
@@ -527,7 +528,7 @@ class Reply(BaseMessageComponent):
|
||||
|
||||
|
||||
class Poke(BaseMessageComponent):
|
||||
type: str = ComponentType.Poke
|
||||
type = ComponentType.Poke
|
||||
id: int | None = 0
|
||||
qq: int | None = 0
|
||||
|
||||
@@ -654,33 +655,19 @@ class File(BaseMessageComponent):
|
||||
|
||||
@property
|
||||
def file(self) -> str:
|
||||
"""获取文件路径,如果文件不存在但有URL,则同步下载文件
|
||||
"""获取本地文件路径(仅返回已存在的文件)
|
||||
|
||||
⚠️ 警告:此属性不会自动下载文件!
|
||||
- 如果文件已存在,返回绝对路径
|
||||
- 如果只有 URL 没有本地文件,返回空字符串
|
||||
- 需要下载文件请使用 `await get_file()` 方法
|
||||
|
||||
Returns:
|
||||
str: 文件路径
|
||||
str: 文件的绝对路径,如果文件不存在则返回空字符串
|
||||
|
||||
"""
|
||||
if self.file_ and os.path.exists(self.file_):
|
||||
return os.path.abspath(self.file_)
|
||||
|
||||
if self.url:
|
||||
try:
|
||||
loop = asyncio.get_event_loop()
|
||||
if loop.is_running():
|
||||
logger.warning(
|
||||
"不可以在异步上下文中同步等待下载! "
|
||||
"这个警告通常发生于某些逻辑试图通过 <File>.file 获取文件消息段的文件内容。"
|
||||
"请使用 await get_file() 代替直接获取 <File>.file 字段",
|
||||
)
|
||||
return ""
|
||||
# 等待下载完成
|
||||
loop.run_until_complete(self._download_file())
|
||||
|
||||
if self.file_ and os.path.exists(self.file_):
|
||||
return os.path.abspath(self.file_)
|
||||
except Exception as e:
|
||||
logger.error(f"文件下载失败: {e}")
|
||||
|
||||
return ""
|
||||
|
||||
@file.setter
|
||||
@@ -714,15 +701,18 @@ class File(BaseMessageComponent):
|
||||
|
||||
if self.url:
|
||||
await self._download_file()
|
||||
return os.path.abspath(self.file_)
|
||||
if self.file_:
|
||||
return os.path.abspath(self.file_)
|
||||
|
||||
return ""
|
||||
|
||||
async def _download_file(self):
|
||||
"""下载文件"""
|
||||
download_dir = os.path.join(get_astrbot_data_path(), "temp")
|
||||
if not self.url:
|
||||
raise ValueError("No URL provided for download")
|
||||
download_dir = str(AstrbotPaths.astrbot_root / "temp")
|
||||
os.makedirs(download_dir, exist_ok=True)
|
||||
file_path = os.path.join(download_dir, f"{uuid.uuid4().hex}")
|
||||
file_path = str(AstrbotPaths.astrbot_root / "temp" / f"{uuid.uuid4().hex}")
|
||||
await download_file(self.url, file_path)
|
||||
self.file_ = os.path.abspath(file_path)
|
||||
|
||||
|
||||
@@ -1,7 +1,6 @@
|
||||
import asyncio
|
||||
import math
|
||||
import random
|
||||
from collections.abc import AsyncGenerator
|
||||
|
||||
import astrbot.core.message.components as Comp
|
||||
from astrbot.core import logger
|
||||
@@ -153,7 +152,7 @@ class RespondStage(Stage):
|
||||
async def process(
|
||||
self,
|
||||
event: AstrMessageEvent,
|
||||
) -> None | AsyncGenerator[None, None]:
|
||||
) -> None:
|
||||
result = event.get_result()
|
||||
if result is None:
|
||||
return
|
||||
|
||||
@@ -5,6 +5,8 @@ import uuid
|
||||
from collections.abc import Awaitable, Callable
|
||||
from typing import Any
|
||||
|
||||
from astrbot_api.abc import IAstrbotPaths
|
||||
|
||||
from astrbot import logger
|
||||
from astrbot.core.message.components import Image, Plain, Record
|
||||
from astrbot.core.message.message_event_result import MessageChain
|
||||
@@ -16,12 +18,13 @@ from astrbot.core.platform import (
|
||||
PlatformMetadata,
|
||||
)
|
||||
from astrbot.core.platform.astr_message_event import MessageSesion
|
||||
from astrbot.core.utils.astrbot_path import get_astrbot_data_path
|
||||
from astrbot_sdk import sync_base_container
|
||||
|
||||
from ...register import register_platform_adapter
|
||||
from .webchat_event import WebChatMessageEvent
|
||||
from .webchat_queue_mgr import WebChatQueueMgr, webchat_queue_mgr
|
||||
|
||||
AstrbotPaths: type[IAstrbotPaths] = sync_base_container.get(type[IAstrbotPaths])
|
||||
|
||||
class QueueListener:
|
||||
def __init__(self, webchat_queue_mgr: WebChatQueueMgr, callback: Callable) -> None:
|
||||
@@ -79,7 +82,7 @@ class WebChatAdapter(Platform):
|
||||
self.config = platform_config
|
||||
self.settings = platform_settings
|
||||
self.unique_session = platform_settings["unique_session"]
|
||||
self.imgs_dir = os.path.join(get_astrbot_data_path(), "webchat", "imgs")
|
||||
self.imgs_dir = str(AstrbotPaths.astrbot_root / "webchat" / "imgs")
|
||||
os.makedirs(self.imgs_dir, exist_ok=True)
|
||||
|
||||
self.metadata = PlatformMetadata(
|
||||
|
||||
@@ -8,10 +8,14 @@ import traceback
|
||||
import aiohttp
|
||||
import anyio
|
||||
import websockets
|
||||
from astrbot_api.abc import IAstrbotPaths
|
||||
|
||||
from astrbot import logger
|
||||
from astrbot.api.message_components import At, Image, Plain, Record
|
||||
from astrbot.api.platform import Platform, PlatformMetadata
|
||||
from astrbot_sdk import sync_base_container
|
||||
|
||||
AstrbotPaths: type[IAstrbotPaths] = sync_base_container.get(type[IAstrbotPaths])
|
||||
from astrbot.core.message.message_event_result import MessageChain
|
||||
from astrbot.core.platform.astr_message_event import MessageSesion
|
||||
from astrbot.core.platform.astrbot_message import (
|
||||
@@ -19,7 +23,6 @@ from astrbot.core.platform.astrbot_message import (
|
||||
MessageMember,
|
||||
MessageType,
|
||||
)
|
||||
from astrbot.core.utils.astrbot_path import get_astrbot_data_path
|
||||
|
||||
from ...register import register_platform_adapter
|
||||
from .wechatpadpro_message_event import WeChatPadProMessageEvent
|
||||
@@ -68,9 +71,8 @@ class WeChatPadProAdapter(Platform):
|
||||
self.base_url = f"http://{self.host}:{self.port}"
|
||||
self.auth_key = None # 用于保存生成的授权码
|
||||
self.wxid = None # 用于保存登录成功后的 wxid
|
||||
self.credentials_file = os.path.join(
|
||||
get_astrbot_data_path(),
|
||||
"wechatpadpro_credentials.json",
|
||||
self.credentials_file = str(
|
||||
AstrbotPaths.astrbot_root / "wechatpadpro_credentials.json"
|
||||
) # 持久化文件路径
|
||||
self.ws_handle_task = None
|
||||
|
||||
@@ -155,8 +157,8 @@ class WeChatPadProAdapter(Platform):
|
||||
}
|
||||
try:
|
||||
# 确保数据目录存在
|
||||
data_dir = os.path.dirname(self.credentials_file)
|
||||
os.makedirs(data_dir, exist_ok=True)
|
||||
config_dir = AstrbotPaths.astrbot_root / "config"
|
||||
config_dir.mkdir(parents=True, exist_ok=True)
|
||||
with open(self.credentials_file, "w") as f:
|
||||
json.dump(credentials, f)
|
||||
except Exception as e:
|
||||
@@ -787,10 +789,10 @@ class WeChatPadProAdapter(Platform):
|
||||
voice_bs64_data = voice_resp.get("Data", {}).get("Base64", None)
|
||||
if voice_bs64_data:
|
||||
voice_bs64_data = base64.b64decode(voice_bs64_data)
|
||||
temp_dir = os.path.join(get_astrbot_data_path(), "temp")
|
||||
file_path = os.path.join(
|
||||
temp_dir,
|
||||
f"wechatpadpro_voice_{abm.message_id}.silk",
|
||||
file_path = str(
|
||||
AstrbotPaths.astrbot_root
|
||||
/ "temp"
|
||||
/ f"wechatpadpro_voice_{abm.message_id}.silk"
|
||||
)
|
||||
|
||||
async with await anyio.open_file(file_path, "wb") as f:
|
||||
|
||||
@@ -15,7 +15,11 @@ except (
|
||||
): # pragma: no cover - older dashscope versions without Qwen TTS support
|
||||
MultiModalConversation = None
|
||||
|
||||
from astrbot.core.utils.astrbot_path import get_astrbot_data_path
|
||||
from astrbot_api.abc import IAstrbotPaths
|
||||
|
||||
from astrbot_sdk import sync_base_container
|
||||
|
||||
AstrbotPaths: type[IAstrbotPaths] = sync_base_container.get(type[IAstrbotPaths])
|
||||
|
||||
from ..entities import ProviderType
|
||||
from ..provider import TTSProvider
|
||||
@@ -45,7 +49,7 @@ class ProviderDashscopeTTSAPI(TTSProvider):
|
||||
if not model:
|
||||
raise RuntimeError("Dashscope TTS model is not configured.")
|
||||
|
||||
temp_dir = os.path.join(get_astrbot_data_path(), "temp")
|
||||
temp_dir = str(AstrbotPaths.astrbot_root / "temp")
|
||||
os.makedirs(temp_dir, exist_ok=True)
|
||||
|
||||
if self._is_qwen_tts_model(model):
|
||||
|
||||
@@ -4,9 +4,12 @@ import subprocess
|
||||
import uuid
|
||||
|
||||
import edge_tts
|
||||
from astrbot_api.abc import IAstrbotPaths
|
||||
|
||||
from astrbot_sdk import sync_base_container
|
||||
|
||||
AstrbotPaths: type[IAstrbotPaths] = sync_base_container.get(type[IAstrbotPaths])
|
||||
from astrbot.core import logger
|
||||
from astrbot.core.utils.astrbot_path import get_astrbot_data_path
|
||||
|
||||
from ..entities import ProviderType
|
||||
from ..provider import TTSProvider
|
||||
@@ -46,8 +49,10 @@ class ProviderEdgeTTS(TTSProvider):
|
||||
self.set_model("edge_tts")
|
||||
|
||||
async def get_audio(self, text: str) -> str:
|
||||
temp_dir = os.path.join(get_astrbot_data_path(), "temp")
|
||||
mp3_path = os.path.join(temp_dir, f"edge_tts_temp_{uuid.uuid4()}.mp3")
|
||||
temp_dir = str(AstrbotPaths.astrbot_root / "temp")
|
||||
mp3_path = str(
|
||||
AstrbotPaths.astrbot_root / "temp" / f"edge_tts_temp_{uuid.uuid4()}.mp3"
|
||||
)
|
||||
wav_path = os.path.join(temp_dir, f"edge_tts_{uuid.uuid4()}.wav")
|
||||
|
||||
# 构建 Edge TTS 参数
|
||||
|
||||
@@ -1,13 +1,16 @@
|
||||
import asyncio
|
||||
import base64
|
||||
import json
|
||||
import os
|
||||
import traceback
|
||||
import uuid
|
||||
|
||||
import aiohttp
|
||||
from astrbot_api.abc import IAstrbotPaths
|
||||
|
||||
from astrbot import logger
|
||||
from astrbot_sdk import sync_base_container
|
||||
|
||||
AstrbotPaths: type[IAstrbotPaths] = sync_base_container.get(type[IAstrbotPaths])
|
||||
|
||||
from ..entities import ProviderType
|
||||
from ..provider import TTSProvider
|
||||
@@ -92,9 +95,10 @@ class ProviderVolcengineTTS(TTSProvider):
|
||||
if "data" in resp_data:
|
||||
audio_data = base64.b64decode(resp_data["data"])
|
||||
|
||||
os.makedirs("data/temp", exist_ok=True)
|
||||
temp_dir = AstrbotPaths.astrbot_root / "temp"
|
||||
temp_dir.mkdir(parents=True, exist_ok=True)
|
||||
|
||||
file_path = f"data/temp/volcengine_tts_{uuid.uuid4()}.mp3"
|
||||
file_path = str(temp_dir / f"volcengine_tts_{uuid.uuid4()}.mp3")
|
||||
|
||||
loop = asyncio.get_running_loop()
|
||||
await loop.run_in_executor(
|
||||
|
||||
@@ -1,10 +1,13 @@
|
||||
import os
|
||||
import uuid
|
||||
|
||||
from astrbot_api.abc import IAstrbotPaths
|
||||
from openai import NOT_GIVEN, AsyncOpenAI
|
||||
|
||||
from astrbot_sdk import sync_base_container
|
||||
|
||||
AstrbotPaths: type[IAstrbotPaths] = sync_base_container.get(type[IAstrbotPaths])
|
||||
from astrbot.core import logger
|
||||
from astrbot.core.utils.astrbot_path import get_astrbot_data_path
|
||||
from astrbot.core.utils.io import download_file
|
||||
from astrbot.core.utils.tencent_record_helper import tencent_silk_to_wav
|
||||
|
||||
@@ -53,8 +56,7 @@ class ProviderOpenAIWhisperAPI(STTProvider):
|
||||
is_tencent = True
|
||||
|
||||
name = str(uuid.uuid4())
|
||||
temp_dir = os.path.join(get_astrbot_data_path(), "temp")
|
||||
path = os.path.join(temp_dir, name)
|
||||
path = str(AstrbotPaths.astrbot_root / "temp" / name)
|
||||
await download_file(audio_url, path)
|
||||
audio_url = path
|
||||
|
||||
@@ -65,8 +67,9 @@ class ProviderOpenAIWhisperAPI(STTProvider):
|
||||
is_silk = await self._is_silk_file(audio_url)
|
||||
if is_silk:
|
||||
logger.info("Converting silk file to wav ...")
|
||||
temp_dir = os.path.join(get_astrbot_data_path(), "temp")
|
||||
output_path = os.path.join(temp_dir, str(uuid.uuid4()) + ".wav")
|
||||
output_path = str(
|
||||
AstrbotPaths.astrbot_root / "temp" / f"{uuid.uuid4()}.wav"
|
||||
)
|
||||
await tencent_silk_to_wav(audio_url, output_path)
|
||||
audio_url = output_path
|
||||
|
||||
|
||||
@@ -3,7 +3,11 @@
|
||||
import json
|
||||
import os
|
||||
|
||||
from astrbot.core.utils.astrbot_path import get_astrbot_data_path
|
||||
from astrbot_api.abc import IAstrbotPaths
|
||||
|
||||
from astrbot_sdk import sync_base_container
|
||||
|
||||
AstrbotPaths: type[IAstrbotPaths] = sync_base_container.get(type[IAstrbotPaths])
|
||||
|
||||
|
||||
def load_config(namespace: str) -> dict | bool:
|
||||
@@ -11,7 +15,7 @@ def load_config(namespace: str) -> dict | bool:
|
||||
namespace: str, 配置的唯一识别符,也就是配置文件的名字。
|
||||
返回值: 当配置文件存在时,返回 namespace 对应配置文件的内容dict,否则返回 False。
|
||||
"""
|
||||
path = os.path.join(get_astrbot_data_path(), "config", f"{namespace}.json")
|
||||
path = str(AstrbotPaths.astrbot_root / "config" / f"{namespace}.json")
|
||||
if not os.path.exists(path):
|
||||
return False
|
||||
with open(path, encoding="utf-8-sig") as f:
|
||||
@@ -41,8 +45,7 @@ def put_config(namespace: str, name: str, key: str, value, description: str):
|
||||
if not isinstance(value, (str, int, float, bool, list)):
|
||||
raise ValueError("value 只支持 str, int, float, bool, list 类型。")
|
||||
|
||||
config_dir = os.path.join(get_astrbot_data_path(), "config")
|
||||
path = os.path.join(config_dir, f"{namespace}.json")
|
||||
path = str(AstrbotPaths.astrbot_root / "config" / f"{namespace}.json")
|
||||
|
||||
if not os.path.exists(path):
|
||||
with open(path, "w", encoding="utf-8-sig") as f:
|
||||
@@ -70,7 +73,7 @@ def update_config(namespace: str, key: str, value):
|
||||
key: str, 配置项的键。
|
||||
value: str, int, float, bool, list, 配置项的值。
|
||||
"""
|
||||
path = os.path.join(get_astrbot_data_path(), "config", f"{namespace}.json")
|
||||
path = str(AstrbotPaths.astrbot_root / "config" / f"{namespace}.json")
|
||||
if not os.path.exists(path):
|
||||
raise FileNotFoundError(f"配置文件 {namespace}.json 不存在。")
|
||||
with open(path, encoding="utf-8-sig") as f:
|
||||
|
||||
@@ -6,7 +6,7 @@ import psutil
|
||||
|
||||
from astrbot.core import logger
|
||||
from astrbot.core.config.default import VERSION
|
||||
from astrbot.core.utils.astrbot_path import get_astrbot_path
|
||||
from astrbot.core.utils.astrbot_path import get_astrbot_data_path
|
||||
from astrbot.core.utils.io import download_file
|
||||
|
||||
from .zip_updator import ReleaseInfo, RepoZipUpdator
|
||||
@@ -20,7 +20,7 @@ class AstrBotUpdator(RepoZipUpdator):
|
||||
|
||||
def __init__(self, repo_mirror: str = "") -> None:
|
||||
super().__init__(repo_mirror)
|
||||
self.MAIN_PATH = get_astrbot_path()
|
||||
self.astrbot_root = get_astrbot_data_path()
|
||||
self.ASTRBOT_RELEASE_API = "https://api.soulter.top/releases"
|
||||
|
||||
def terminate_child_processes(self):
|
||||
@@ -117,7 +117,7 @@ class AstrBotUpdator(RepoZipUpdator):
|
||||
try:
|
||||
await download_file(file_url, "temp.zip")
|
||||
logger.info("下载 AstrBot Core 更新文件完成,正在执行解压...")
|
||||
self.unzip_file("temp.zip", self.MAIN_PATH)
|
||||
self.unzip_file("temp.zip", self.astrbot_root)
|
||||
except BaseException as e:
|
||||
raise e
|
||||
|
||||
|
||||
@@ -1,39 +1,70 @@
|
||||
"""Astrbot统一路径获取
|
||||
|
||||
项目路径:固定为源码所在路径
|
||||
根目录路径:默认为当前工作目录,可通过环境变量 ASTRBOT_ROOT 指定
|
||||
数据目录路径:固定为根目录下的 data 目录
|
||||
配置文件路径:固定为数据目录下的 config 目录
|
||||
插件目录路径:固定为数据目录下的 plugins 目录
|
||||
|
||||
"""
|
||||
|
||||
import os
|
||||
import warnings
|
||||
|
||||
from astrbot_api.abc import IAstrbotPaths
|
||||
|
||||
from astrbot_sdk import sync_base_container
|
||||
|
||||
AstrbotPaths = sync_base_container.get(type[IAstrbotPaths])
|
||||
|
||||
def get_astrbot_path() -> str:
|
||||
"""获取Astrbot项目路径"""
|
||||
"""获取Astrbot项目路径 -- 请勿使用本函数!!! -- 仅供兼容旧代码使用"""
|
||||
warnings.warn(
|
||||
"get_astrbot_path is deprecated. Use AstrbotPaths class instead.",
|
||||
DeprecationWarning,
|
||||
stacklevel=2,
|
||||
)
|
||||
return os.path.realpath(
|
||||
os.path.join(os.path.dirname(os.path.abspath(__file__)), "../../../"),
|
||||
)
|
||||
|
||||
|
||||
def get_astrbot_root() -> str:
|
||||
"""获取Astrbot根目录路径"""
|
||||
if path := os.environ.get("ASTRBOT_ROOT"):
|
||||
return os.path.realpath(path)
|
||||
return os.path.realpath(os.getcwd())
|
||||
"""获取Astrbot根目录路径 --> get_astrbot_data_path"""
|
||||
warnings.warn(
|
||||
"不要再使用本函数!等效于: AstrbotPaths.astrbot_root",
|
||||
DeprecationWarning,
|
||||
stacklevel=2,
|
||||
)
|
||||
return str(AstrbotPaths.astrbot_root)
|
||||
|
||||
|
||||
def get_astrbot_data_path() -> str:
|
||||
"""获取Astrbot数据目录路径"""
|
||||
return os.path.realpath(os.path.join(get_astrbot_root(), "data"))
|
||||
"""获取Astrbot数据目录路径
|
||||
特别注意!
|
||||
这里的data目录指的就是.astrbot根目录!
|
||||
两者是等价的!
|
||||
不要和AstrbotPaths.data混淆!
|
||||
"""
|
||||
warnings.warn(
|
||||
"等效于: AstrbotPaths.astrbot_root",
|
||||
DeprecationWarning,
|
||||
stacklevel=2,
|
||||
)
|
||||
return str(AstrbotPaths.astrbot_root)
|
||||
|
||||
|
||||
def get_astrbot_config_path() -> str:
|
||||
"""获取Astrbot配置文件路径"""
|
||||
return os.path.realpath(os.path.join(get_astrbot_data_path(), "config"))
|
||||
warnings.warn(
|
||||
"get_astrbot_config_path is deprecated. Use AstrbotPaths class instead.",
|
||||
DeprecationWarning,
|
||||
stacklevel=2,
|
||||
)
|
||||
return str(AstrbotPaths.astrbot_root / "config")
|
||||
|
||||
|
||||
def get_astrbot_plugin_path() -> str:
|
||||
"""获取Astrbot插件目录路径"""
|
||||
return os.path.realpath(os.path.join(get_astrbot_data_path(), "plugins"))
|
||||
warnings.warn(
|
||||
"get_astrbot_plugin_path is deprecated. Use AstrbotPaths class instead.",
|
||||
DeprecationWarning,
|
||||
stacklevel=2,
|
||||
)
|
||||
return str(AstrbotPaths.astrbot_root / "plugins")
|
||||
|
||||
@@ -12,12 +12,16 @@ from pathlib import Path
|
||||
import aiohttp
|
||||
import certifi
|
||||
import psutil
|
||||
from astrbot_api.abc import IAstrbotPaths
|
||||
from PIL import Image
|
||||
|
||||
from astrbot_sdk import sync_base_container
|
||||
|
||||
from .astrbot_path import get_astrbot_data_path
|
||||
|
||||
logger = logging.getLogger("astrbot")
|
||||
|
||||
AstrbotPaths: type[IAstrbotPaths] = sync_base_container.get(type[IAstrbotPaths])
|
||||
|
||||
def on_error(func, path, exc_info):
|
||||
"""A callback of the rmtree function."""
|
||||
@@ -50,11 +54,11 @@ def port_checker(port: int, host: str = "localhost"):
|
||||
|
||||
|
||||
def save_temp_img(img: Image.Image | str) -> str:
|
||||
temp_dir = os.path.join(get_astrbot_data_path(), "temp")
|
||||
temp_dir = str(AstrbotPaths.astrbot_root / "temp")
|
||||
# 获得文件创建时间,清除超过 12 小时的
|
||||
try:
|
||||
for f in os.listdir(temp_dir):
|
||||
path = os.path.join(temp_dir, f)
|
||||
path = str(AstrbotPaths.astrbot_root / "temp" / f)
|
||||
if os.path.isfile(path):
|
||||
ctime = os.path.getctime(path)
|
||||
if time.time() - ctime > 3600 * 12:
|
||||
@@ -64,7 +68,7 @@ def save_temp_img(img: Image.Image | str) -> str:
|
||||
|
||||
# 获得时间戳
|
||||
timestamp = f"{int(time.time())}_{uuid.uuid4().hex[:8]}"
|
||||
p = os.path.join(temp_dir, f"{timestamp}.jpg")
|
||||
p = str(AstrbotPaths.astrbot_root / "temp" / f"{timestamp}.jpg")
|
||||
|
||||
if isinstance(img, Image.Image):
|
||||
img.save(p)
|
||||
@@ -205,9 +209,9 @@ def get_local_ip_addresses():
|
||||
|
||||
|
||||
async def get_dashboard_version():
|
||||
dist_dir = os.path.join(get_astrbot_data_path(), "dist")
|
||||
dist_dir = str(AstrbotPaths.astrbot_root / "dist")
|
||||
if os.path.exists(dist_dir):
|
||||
version_file = os.path.join(dist_dir, "assets", "version")
|
||||
version_file = str(AstrbotPaths.astrbot_root / "dist" / "assets" / "version")
|
||||
if os.path.exists(version_file):
|
||||
with open(version_file, encoding="utf-8") as f:
|
||||
v = f.read().strip()
|
||||
|
||||
@@ -2,8 +2,9 @@
|
||||
|
||||
import os
|
||||
import shutil
|
||||
from importlib import resources
|
||||
|
||||
from astrbot.core.utils.astrbot_path import get_astrbot_data_path, get_astrbot_path
|
||||
from astrbot.core.utils.astrbot_path import get_astrbot_data_path
|
||||
|
||||
|
||||
class TemplateManager:
|
||||
@@ -15,14 +16,10 @@ class TemplateManager:
|
||||
CORE_TEMPLATES = ["base.html", "astrbot_powershell.html"]
|
||||
|
||||
def __init__(self):
|
||||
self.builtin_template_dir = os.path.join(
|
||||
get_astrbot_path(),
|
||||
"astrbot",
|
||||
"core",
|
||||
"utils",
|
||||
"t2i",
|
||||
"template",
|
||||
self.builtin_template_dir = str(
|
||||
resources.files("astrbot.core.utils.t2i.template")
|
||||
)
|
||||
|
||||
self.user_template_dir = os.path.join(get_astrbot_data_path(), "t2i_templates")
|
||||
|
||||
os.makedirs(self.user_template_dir, exist_ok=True)
|
||||
|
||||
@@ -4,15 +4,18 @@ import os
|
||||
import uuid
|
||||
from contextlib import asynccontextmanager
|
||||
|
||||
from astrbot_api.abc import IAstrbotPaths
|
||||
from quart import Response as QuartResponse
|
||||
from quart import g, make_response, request
|
||||
|
||||
from astrbot_sdk import sync_base_container
|
||||
|
||||
AstrbotPaths: type[IAstrbotPaths] = sync_base_container.get(type[IAstrbotPaths])
|
||||
from astrbot.core import logger
|
||||
from astrbot.core.core_lifecycle import AstrBotCoreLifecycle
|
||||
from astrbot.core.db import BaseDatabase
|
||||
from astrbot.core.platform.astr_message_event import MessageSession
|
||||
from astrbot.core.platform.sources.webchat.webchat_queue_mgr import webchat_queue_mgr
|
||||
from astrbot.core.utils.astrbot_path import get_astrbot_data_path
|
||||
|
||||
from .route import Response, Route, RouteContext
|
||||
|
||||
@@ -47,7 +50,7 @@ class ChatRoute(Route):
|
||||
}
|
||||
self.core_lifecycle = core_lifecycle
|
||||
self.register_routes()
|
||||
self.imgs_dir = os.path.join(get_astrbot_data_path(), "webchat", "imgs")
|
||||
self.imgs_dir = str(AstrbotPaths.astrbot_root / "webchat" / "imgs")
|
||||
os.makedirs(self.imgs_dir, exist_ok=True)
|
||||
|
||||
self.supported_imgs = ["jpg", "jpeg", "png", "gif", "webp"]
|
||||
|
||||
@@ -1,4 +1,5 @@
|
||||
import asyncio
|
||||
import importlib.resources
|
||||
import inspect
|
||||
import os
|
||||
import traceback
|
||||
@@ -21,7 +22,6 @@ from astrbot.core.provider.entities import ProviderType
|
||||
from astrbot.core.provider.provider import RerankProvider
|
||||
from astrbot.core.provider.register import provider_registry
|
||||
from astrbot.core.star.star import star_registry
|
||||
from astrbot.core.utils.astrbot_path import get_astrbot_path
|
||||
|
||||
from .route import Response, Route, RouteContext
|
||||
|
||||
@@ -461,11 +461,12 @@ class ConfigRoute(Route):
|
||||
logger.debug(
|
||||
f"Sending health check audio to provider: {status_info['name']}",
|
||||
)
|
||||
sample_audio_path = os.path.join(
|
||||
get_astrbot_path(),
|
||||
"samples",
|
||||
"stt_health_check.wav",
|
||||
sample_audio_path = str(
|
||||
importlib.resources.files("astrbot")
|
||||
/ "samples"
|
||||
/ "stt_health_check.wav"
|
||||
)
|
||||
|
||||
if not os.path.exists(sample_audio_path):
|
||||
status_info["status"] = "unavailable"
|
||||
status_info["error"] = (
|
||||
|
||||
@@ -5,6 +5,7 @@ import socket
|
||||
|
||||
import jwt
|
||||
import psutil
|
||||
from astrbot_api.abc import IAstrbotPaths
|
||||
from quart import Quart, g, jsonify, request
|
||||
from quart.logging import default_handler
|
||||
|
||||
@@ -12,8 +13,8 @@ from astrbot.core import logger
|
||||
from astrbot.core.config.default import VERSION
|
||||
from astrbot.core.core_lifecycle import AstrBotCoreLifecycle
|
||||
from astrbot.core.db import BaseDatabase
|
||||
from astrbot.core.utils.astrbot_path import get_astrbot_data_path
|
||||
from astrbot.core.utils.io import get_local_ip_addresses
|
||||
from astrbot_sdk import sync_base_container
|
||||
|
||||
from .routes import *
|
||||
from .routes.route import Response, RouteContext
|
||||
@@ -22,7 +23,7 @@ from .routes.t2i import T2iRoute
|
||||
|
||||
APP: Quart = None
|
||||
|
||||
|
||||
AstrbotPaths: type[IAstrbotPaths] = sync_base_container.get(type[IAstrbotPaths])
|
||||
class AstrBotDashboard:
|
||||
def __init__(
|
||||
self,
|
||||
@@ -38,9 +39,7 @@ class AstrBotDashboard:
|
||||
if webui_dir and os.path.exists(webui_dir):
|
||||
self.data_path = os.path.abspath(webui_dir)
|
||||
else:
|
||||
self.data_path = os.path.abspath(
|
||||
os.path.join(get_astrbot_data_path(), "dist"),
|
||||
)
|
||||
self.data_path = os.path.abspath(str(AstrbotPaths.astrbot_root / "dist"))
|
||||
|
||||
self.app = Quart("dashboard", static_folder=self.data_path, static_url_path="/")
|
||||
APP = self.app # noqa
|
||||
|
||||
0
astrbot_api/README.md
Normal file
0
astrbot_api/README.md
Normal file
25
astrbot_api/pyproject.toml
Normal file
25
astrbot_api/pyproject.toml
Normal file
@@ -0,0 +1,25 @@
|
||||
[project]
|
||||
name = "astrbot-api"
|
||||
dynamic = ["version"]
|
||||
description = "Astrbot Python API"
|
||||
readme = "README.md"
|
||||
authors = [
|
||||
{ name = "AstrBotDevs", email = "community@astrbot.app" }
|
||||
]
|
||||
|
||||
requires-python = ">=3.10"
|
||||
dependencies = [
|
||||
"anyio>=4.11.0",
|
||||
"mcp>=1.12.4",
|
||||
"pydantic>=2.10.6",
|
||||
"pyyaml>=6.0.3",
|
||||
"types-pyyaml>=6.0.12.20250915",
|
||||
"yarl>=1.22.0",
|
||||
]
|
||||
|
||||
[build-system]
|
||||
requires = ["hatchling", "uv-dynamic-versioning"]
|
||||
build-backend = "hatchling.build"
|
||||
|
||||
[tool.hatch.version]
|
||||
source = "uv-dynamic-versioning"
|
||||
7
astrbot_api/src/astrbot_api/__init__.py
Normal file
7
astrbot_api/src/astrbot_api/__init__.py
Normal file
@@ -0,0 +1,7 @@
|
||||
from .abc import IAstrbotPaths
|
||||
from .const import LOGO
|
||||
|
||||
__all__ = [
|
||||
"IAstrbotPaths",
|
||||
"LOGO",
|
||||
]
|
||||
379
astrbot_api/src/astrbot_api/abc.py
Normal file
379
astrbot_api/src/astrbot_api/abc.py
Normal file
@@ -0,0 +1,379 @@
|
||||
from __future__ import annotations
|
||||
|
||||
import json
|
||||
from abc import ABC, abstractmethod
|
||||
from collections.abc import AsyncGenerator, Callable, Generator
|
||||
from contextlib import (
|
||||
AbstractAsyncContextManager,
|
||||
AbstractContextManager,
|
||||
asynccontextmanager,
|
||||
contextmanager,
|
||||
)
|
||||
from importlib.abc import Traversable
|
||||
from pathlib import Path
|
||||
from typing import ClassVar
|
||||
|
||||
import anyio
|
||||
import tomli
|
||||
import yaml
|
||||
from yarl import URL
|
||||
|
||||
from .models import AstrbotPluginMetadata
|
||||
|
||||
# region 核心运行时协议
|
||||
|
||||
|
||||
class IAstrbotContainerMgr(ABC):
|
||||
"""AstrBot 容器管理器的抽象基类."""
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
class IAstrbotPaths(ABC):
|
||||
"""路径管理的抽象基类."""
|
||||
astrbot_root: ClassVar[Path]
|
||||
"""Astrbot 根目录路径."""
|
||||
|
||||
@abstractmethod
|
||||
def __init__(self, name: str) -> None:
|
||||
"""初始化路径管理器."""
|
||||
|
||||
@classmethod
|
||||
@abstractmethod
|
||||
def getPaths(cls, name: str) -> IAstrbotPaths:
|
||||
"""返回Paths实例,用于访问模块的各类目录."""
|
||||
|
||||
@property
|
||||
@abstractmethod
|
||||
def root(self) -> Path:
|
||||
"""获取根目录."""
|
||||
|
||||
@property
|
||||
@abstractmethod
|
||||
def home(self) -> Path:
|
||||
"""获取模块/插件主目录."""
|
||||
|
||||
@property
|
||||
@abstractmethod
|
||||
def config(self) -> Path:
|
||||
"""获取模块配置目录."""
|
||||
|
||||
@property
|
||||
@abstractmethod
|
||||
def data(self) -> Path:
|
||||
"""获取模块数据目录."""
|
||||
|
||||
@property
|
||||
@abstractmethod
|
||||
def log(self) -> Path:
|
||||
"""获取模块日志目录."""
|
||||
|
||||
@property
|
||||
@abstractmethod
|
||||
def temp(self) -> Path:
|
||||
"""获取模块临时目录."""
|
||||
|
||||
@property
|
||||
@abstractmethod
|
||||
def plugins(self) -> Path:
|
||||
"""获取插件目录."""
|
||||
|
||||
@abstractmethod
|
||||
def reload(self) -> None:
|
||||
"""重新加载环境变量."""
|
||||
|
||||
@classmethod
|
||||
@abstractmethod
|
||||
def is_root(cls, path: Path) -> bool:
|
||||
"""判断路径是否为根目录."""
|
||||
|
||||
@abstractmethod
|
||||
def chdir(self, cwd: str = "home") -> AbstractContextManager[Path]:
|
||||
"""临时切换到指定目录, 子进程将继承此 CWD。"""
|
||||
|
||||
@abstractmethod
|
||||
async def achdir(self, cwd: str = "home") -> AbstractAsyncContextManager[Path]:
|
||||
"""异步临时切换到指定目录, 子进程将继承此 CWD。"""
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
class AstrbotPluginBaseSession(ABC):
|
||||
"""插件会话的基类."""
|
||||
|
||||
url: URL
|
||||
"""插件的astrbot专有协议URL地址.
|
||||
|
||||
协议: astrbot://{stdio/web/legacy}/plugin_id
|
||||
"""
|
||||
|
||||
|
||||
@abstractmethod
|
||||
def connect(self) -> AstrbotPluginBaseSession:
|
||||
"""连接到插件."""
|
||||
...
|
||||
|
||||
@abstractmethod
|
||||
def disconnect(self) -> None:
|
||||
"""断开与插件的连接."""
|
||||
...
|
||||
|
||||
@abstractmethod
|
||||
async def async_connect(self) -> AstrbotPluginBaseSession:
|
||||
"""异步连接到插件."""
|
||||
...
|
||||
|
||||
@abstractmethod
|
||||
async def async_disconnect(self) -> None:
|
||||
"""异步断开与插件的连接."""
|
||||
...
|
||||
|
||||
@contextmanager
|
||||
def session_scope(self) -> Generator[AstrbotPluginBaseSession]:
|
||||
"""插件会话的上下文管理器."""
|
||||
try:
|
||||
yield self.connect()
|
||||
finally:
|
||||
self.disconnect()
|
||||
|
||||
@asynccontextmanager
|
||||
async def async_session_scope(self) -> AsyncGenerator[AstrbotPluginBaseSession]:
|
||||
"""异步插件会话的上下文管理器."""
|
||||
try:
|
||||
yield await self.async_connect()
|
||||
finally:
|
||||
await self.async_disconnect()
|
||||
|
||||
# region 数据传输方法
|
||||
|
||||
# 主动发送数据
|
||||
@abstractmethod
|
||||
def send(self, data: bytes) -> bytes:
|
||||
"""发送数据到插件并接收响应."""
|
||||
...
|
||||
|
||||
@abstractmethod
|
||||
async def async_send(self, data: bytes) -> bytes:
|
||||
"""异步发送数据到插件并接收响应."""
|
||||
...
|
||||
|
||||
# 被动接收数据
|
||||
@abstractmethod
|
||||
def listen(self, callback: Callable[[bytes], bytes | None]) -> None:
|
||||
"""监听插件发送的数据.
|
||||
|
||||
callback: 一个接受 bytes 类型参数并返回 bytes 或 None 的函数.
|
||||
如果返回 None, 则不给插件发送响应.
|
||||
|
||||
"""
|
||||
...
|
||||
|
||||
@abstractmethod
|
||||
async def async_listen(self, callback: Callable[[bytes], bytes | None]) -> None:
|
||||
"""异步监听插件发送的数据.
|
||||
|
||||
callback: 一个接受 bytes 类型参数并返回 bytes 或 None 的函数.
|
||||
如果返回 None, 则不给插件发送响应.
|
||||
"""
|
||||
...
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
class IVirtualAstrbotPlugin(ABC):
|
||||
"""AstrBot 虚拟插件的基类协议."""
|
||||
|
||||
vpo_map: ClassVar[dict[URL, IVirtualAstrbotPlugin]] = {}
|
||||
"""虚拟插件对象映射表, key 是插件的 astrbot 协议 URL 地址."""
|
||||
|
||||
url: URL
|
||||
"""AstrBot 插件的 astrbot 专有协议 URL 地址.
|
||||
协议:
|
||||
astrbot://{stdio/web/legacy}/plugin_id
|
||||
"""
|
||||
|
||||
metadata: AstrbotPluginMetadata
|
||||
"""插件元数据."""
|
||||
|
||||
session: AstrbotPluginBaseSession
|
||||
"""插件会话对象."""
|
||||
|
||||
# region 公共方法
|
||||
@classmethod
|
||||
@abstractmethod
|
||||
def handshake(cls) -> None:
|
||||
"""通用插件握手方法.
|
||||
|
||||
1. 发送握手请求
|
||||
2. 接受插件元数据响应,并设置 metadata 属性
|
||||
3. 返回确认消息,表示握手成功
|
||||
"""
|
||||
...
|
||||
|
||||
# part1 :工厂方法
|
||||
@classmethod
|
||||
@abstractmethod
|
||||
def fromFile(cls, path: Path, * , stdio: bool = False) -> IVirtualAstrbotPlugin:
|
||||
"""从文件加载插件/插件包的公共方法.
|
||||
|
||||
通过此方法加载经典插件: stdio=False
|
||||
或者子进程插件: stdio=True
|
||||
|
||||
任务:
|
||||
1. 从 path 加载插件,调用私有方法 _load_metadata 加载插件元数据.
|
||||
"""
|
||||
...
|
||||
|
||||
|
||||
@classmethod
|
||||
@abstractmethod
|
||||
def fromURL(cls, url: URL) -> IVirtualAstrbotPlugin:
|
||||
"""从URL加载插件/插件包的公共方法."""
|
||||
...
|
||||
|
||||
|
||||
# part2 :实例方法
|
||||
|
||||
@abstractmethod
|
||||
def get_logo(self) -> Traversable | None:
|
||||
"""获取插件的logo文件路径(Optional).
|
||||
|
||||
请使用importlib.resources.files来访问并返回Traversable对象.
|
||||
示例:
|
||||
from importlib.resources import files
|
||||
logo_path = files("plugin_a/assets/logo.png")
|
||||
# 如果插件安装在虚拟环境,可以直接这样获取
|
||||
|
||||
# 如果插件不可以安装到虚拟环境,适用于子进程插件/网络插件,可以这样获取
|
||||
|
||||
|
||||
返回None表示没有logo文件.
|
||||
Returns:
|
||||
Traversable | None: 插件logo文件的路径或None.
|
||||
"""
|
||||
...
|
||||
|
||||
@abstractmethod
|
||||
def get_metadata(self) -> AstrbotPluginMetadata:
|
||||
"""获取插件元数据的公共方法."""
|
||||
...
|
||||
|
||||
@abstractmethod
|
||||
def get_url(self) -> URL:
|
||||
"""获取插件URL(astrbot协议)的公共方法."""
|
||||
...
|
||||
|
||||
@abstractmethod
|
||||
def get_session(self) -> AstrbotPluginBaseSession:
|
||||
"""获取插件会话对象的公共方法."""
|
||||
...
|
||||
|
||||
|
||||
|
||||
# region 魔术方法
|
||||
|
||||
@abstractmethod
|
||||
def __str__(self) -> str:
|
||||
"""返回插件元数据的字符串表示."""
|
||||
...
|
||||
|
||||
@abstractmethod
|
||||
def __repr__(self) -> str:
|
||||
"""返回插件元数据的正式字符串表示."""
|
||||
...
|
||||
|
||||
# region 私有方法
|
||||
@classmethod
|
||||
def _load_metadata(cls, path: Path) -> AstrbotPluginMetadata:
|
||||
"""加载插件元数据的私有方法,自动按下列优先级加载: pyproject -> yaml -> toml -> json.
|
||||
|
||||
其中yaml: plugin.yaml > metadata.yaml
|
||||
|
||||
此函数是上述工厂方法的辅助函数,用于加载插件元数据.
|
||||
"""
|
||||
match path.suffix.lower():
|
||||
case ".json":
|
||||
return cls._load_metadata_json(path)
|
||||
case ".toml":
|
||||
return cls._load_metadata_toml(path)
|
||||
case ".yaml" | ".yml":
|
||||
return cls._load_metadata_yaml(path)
|
||||
case _:
|
||||
raise ValueError(f"不支持的插件元数据文件格式: {path.suffix}")
|
||||
|
||||
@classmethod
|
||||
async def _load_metadata_async(cls, path: Path) -> AstrbotPluginMetadata:
|
||||
"""异步加载插件元数据的私有方法,自动按下列优先级加载: pyproject -> yaml -> toml -> json.
|
||||
|
||||
其中yaml: plugin.yaml > metadata.yaml
|
||||
|
||||
此函数是上述工厂方法的辅助函数,用于加载插件元数据.
|
||||
"""
|
||||
return await anyio.to_thread.run_sync(cls._load_metadata, path) # type: ignore[attr-defined]
|
||||
|
||||
@classmethod
|
||||
def _load_metadata_json(cls, path: Path) -> AstrbotPluginMetadata:
|
||||
"""从json文件加载插件元数据的私有方法."""
|
||||
with path.open(encoding="utf-8") as f:
|
||||
data = json.load(f)
|
||||
return AstrbotPluginMetadata.model_validate(**data)
|
||||
|
||||
@classmethod
|
||||
async def _load_metadata_json_async(cls, path: Path) -> AstrbotPluginMetadata:
|
||||
"""异步从json文件加载插件元数据的私有方法."""
|
||||
async with await anyio.open_file(path, "r", encoding="utf-8") as f:
|
||||
content = await f.read()
|
||||
data = json.loads(content)
|
||||
return AstrbotPluginMetadata.model_validate(**data)
|
||||
|
||||
@classmethod
|
||||
def _load_metadata_toml(cls, path: Path) -> AstrbotPluginMetadata:
|
||||
"""从toml文件加载插件元数据的私有方法."""
|
||||
with path.open("rb") as f:
|
||||
data = tomli.load(f)
|
||||
return AstrbotPluginMetadata.model_validate(**data)
|
||||
|
||||
@classmethod
|
||||
async def _load_metadata_toml_async(cls, path: Path) -> AstrbotPluginMetadata:
|
||||
"""异步从toml文件加载插件元数据的私有方法."""
|
||||
async with await anyio.open_file(path, "rb") as f:
|
||||
content = await f.read()
|
||||
content_str = content.decode("utf-8")
|
||||
data = tomli.loads(content_str)
|
||||
return AstrbotPluginMetadata.model_validate(**data)
|
||||
|
||||
@classmethod
|
||||
def _load_metadata_yaml(cls, path: Path) -> AstrbotPluginMetadata:
|
||||
"""从yaml文件加载插件元数据的私有方法."""
|
||||
with path.open(encoding="utf-8") as f:
|
||||
data = yaml.safe_load(f)
|
||||
return AstrbotPluginMetadata.model_validate(**data)
|
||||
|
||||
@classmethod
|
||||
async def _load_metadata_yaml_async(cls, path: Path) -> AstrbotPluginMetadata:
|
||||
"""异步从yaml文件加载插件元数据的私有方法."""
|
||||
async with await anyio.open_file(path, "r", encoding="utf-8") as f:
|
||||
content = await f.read()
|
||||
data = yaml.safe_load(content)
|
||||
return AstrbotPluginMetadata.model_validate(**data)
|
||||
|
||||
|
||||
# region 插件运行时协议
|
||||
|
||||
class IAstrbotPluginRuntime(ABC):
|
||||
"""AstrBot 插件运行时的基类协议."""
|
||||
|
||||
@abstractmethod
|
||||
def start(self) -> None:
|
||||
"""启动插件运行时."""
|
||||
...
|
||||
|
||||
@abstractmethod
|
||||
def stop(self) -> None:
|
||||
"""停止插件运行时."""
|
||||
...
|
||||
|
||||
9
astrbot_api/src/astrbot_api/const.py
Normal file
9
astrbot_api/src/astrbot_api/const.py
Normal file
@@ -0,0 +1,9 @@
|
||||
LOGO = r"""
|
||||
___ _______.___________..______ .______ ______ .___________.
|
||||
/ \ / | || _ \ | _ \ / __ \ | |
|
||||
/ ^ \ | (----`---| |----`| |_) | | |_) | | | | | `---| |----`
|
||||
/ /_\ \ \ \ | | | / | _ < | | | | | |
|
||||
/ _____ \ .----) | | | | |\ \----.| |_) | | `--' | | |
|
||||
/__/ \__\ |_______/ |__| | _| `._____||______/ \______/ |__|
|
||||
|
||||
"""
|
||||
11
astrbot_api/src/astrbot_api/enums.py
Normal file
11
astrbot_api/src/astrbot_api/enums.py
Normal file
@@ -0,0 +1,11 @@
|
||||
from enum import Enum
|
||||
|
||||
|
||||
class PluginType(Enum):
|
||||
"""插件大类."""
|
||||
# 兼容保留
|
||||
LEGACY = "legacy" # 经典插件,直接在主进程中运行 -- 逐渐淘汰
|
||||
|
||||
# 后两者为新插件机制
|
||||
STDIO = "stdio" # 子进程 -- 进程间通信
|
||||
WEB = "web" # 通过 HTTP/HTTPS 协议调用
|
||||
3
astrbot_api/src/astrbot_api/exceptions.py
Normal file
3
astrbot_api/src/astrbot_api/exceptions.py
Normal file
@@ -0,0 +1,3 @@
|
||||
|
||||
class AstrbotBaseError(Exception):
|
||||
"""Base exception for Astrbot API errors."""
|
||||
28
astrbot_api/src/astrbot_api/models.py
Normal file
28
astrbot_api/src/astrbot_api/models.py
Normal file
@@ -0,0 +1,28 @@
|
||||
from __future__ import annotations
|
||||
|
||||
from pydantic import BaseModel
|
||||
|
||||
|
||||
class AstrbotPluginMetadata(BaseModel):
|
||||
"""AstrBot 插件元数据模型."""
|
||||
name: str | None = None
|
||||
"""插件名"""
|
||||
author: str | None = None
|
||||
"""插件作者"""
|
||||
desc: str | None = None
|
||||
"""插件简介"""
|
||||
version: str | None = None
|
||||
"""插件版本"""
|
||||
repo: str | None = None
|
||||
"""插件仓库地址"""
|
||||
|
||||
reserved: bool = False
|
||||
"""是否是 AstrBot 的保留插件"""
|
||||
|
||||
activated: bool = True
|
||||
"""是否被激活"""
|
||||
|
||||
display_name: str | None = None
|
||||
"""用于展示的插件名称"""
|
||||
|
||||
|
||||
1
astrbot_api/src/astrbot_api/protocols.py
Normal file
1
astrbot_api/src/astrbot_api/protocols.py
Normal file
@@ -0,0 +1 @@
|
||||
|
||||
0
astrbot_api/src/astrbot_api/py.typed
Normal file
0
astrbot_api/src/astrbot_api/py.typed
Normal file
0
astrbot_api/src/astrbot_api/types.py
Normal file
0
astrbot_api/src/astrbot_api/types.py
Normal file
0
astrbot_sdk/README.md
Normal file
0
astrbot_sdk/README.md
Normal file
25
astrbot_sdk/pyproject.toml
Normal file
25
astrbot_sdk/pyproject.toml
Normal file
@@ -0,0 +1,25 @@
|
||||
[project]
|
||||
name = "astrbot-sdk"
|
||||
dynamic = ["version"]
|
||||
description = "Astrbot Python SDK"
|
||||
readme = "README.md"
|
||||
authors = [
|
||||
{ name = "AstrBotDevs", email = "community@astrbot.app" }
|
||||
]
|
||||
|
||||
requires-python = ">=3.10"
|
||||
dependencies = [
|
||||
"astrbot-api",
|
||||
"dishka>=1.7.2",
|
||||
"dotenv>=0.9.9",
|
||||
]
|
||||
|
||||
[build-system]
|
||||
requires = ["hatchling", "uv-dynamic-versioning"]
|
||||
build-backend = "hatchling.build"
|
||||
|
||||
[tool.hatch.version]
|
||||
source = "uv-dynamic-versioning"
|
||||
|
||||
[tool.uv.sources]
|
||||
astrbot-api = { workspace = true }
|
||||
23
astrbot_sdk/src/astrbot_sdk/__init__.py
Normal file
23
astrbot_sdk/src/astrbot_sdk/__init__.py
Normal file
@@ -0,0 +1,23 @@
|
||||
from dishka import (
|
||||
AsyncContainer,
|
||||
Container,
|
||||
Provider,
|
||||
make_async_container,
|
||||
make_container,
|
||||
)
|
||||
|
||||
from .base import AstrbotBaseProvider
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
base_provider: Provider = AstrbotBaseProvider()
|
||||
async_base_container: AsyncContainer = make_async_container(base_provider)
|
||||
sync_base_container: Container = make_container(base_provider)
|
||||
|
||||
__all__ = [
|
||||
"async_base_container",
|
||||
"sync_base_container",
|
||||
]
|
||||
3
astrbot_sdk/src/astrbot_sdk/base/README.md
Normal file
3
astrbot_sdk/src/astrbot_sdk/base/README.md
Normal file
@@ -0,0 +1,3 @@
|
||||
# 基础类的实现
|
||||
|
||||
- 单向导出
|
||||
3
astrbot_sdk/src/astrbot_sdk/base/__init__.py
Normal file
3
astrbot_sdk/src/astrbot_sdk/base/__init__.py
Normal file
@@ -0,0 +1,3 @@
|
||||
from .provider import AstrbotBaseProvider
|
||||
|
||||
__all__ = ["AstrbotBaseProvider"]
|
||||
131
astrbot_sdk/src/astrbot_sdk/base/paths.py
Normal file
131
astrbot_sdk/src/astrbot_sdk/base/paths.py
Normal file
@@ -0,0 +1,131 @@
|
||||
from __future__ import annotations
|
||||
|
||||
from collections.abc import AsyncGenerator, Generator
|
||||
from contextlib import asynccontextmanager, contextmanager
|
||||
from os import chdir, getenv
|
||||
from pathlib import Path
|
||||
from typing import ClassVar
|
||||
|
||||
from dotenv import load_dotenv
|
||||
from packaging.utils import NormalizedName, canonicalize_name
|
||||
|
||||
from astrbot_api import IAstrbotPaths
|
||||
|
||||
|
||||
class AstrbotPaths(IAstrbotPaths):
|
||||
"""统一化路径获取."""
|
||||
|
||||
load_dotenv()
|
||||
astrbot_root: ClassVar[Path] = Path(
|
||||
getenv("ASTRBOT_ROOT", Path.home() / ".astrbot")
|
||||
).absolute()
|
||||
|
||||
_instances: ClassVar[dict[str, AstrbotPaths]] = {}
|
||||
|
||||
def __init__(self, name: str) -> None:
|
||||
self.name: str = name
|
||||
# 确保根目录存在
|
||||
self.astrbot_root.mkdir(parents=True, exist_ok=True)
|
||||
|
||||
@classmethod
|
||||
def getPaths(cls, name: str) -> AstrbotPaths:
|
||||
"""返回Paths实例,用于访问模块的各类目录."""
|
||||
normalized_name: NormalizedName = canonicalize_name(name)
|
||||
if normalized_name in cls._instances:
|
||||
return cls._instances[normalized_name]
|
||||
instance: AstrbotPaths = cls(normalized_name)
|
||||
instance.name = normalized_name
|
||||
cls._instances[normalized_name] = instance
|
||||
return instance
|
||||
|
||||
@property
|
||||
def root(self) -> Path:
|
||||
"""返回根目录."""
|
||||
return (
|
||||
self.astrbot_root if self.astrbot_root.exists() else Path.cwd() / ".astrbot"
|
||||
)
|
||||
|
||||
@property
|
||||
def home(self) -> Path:
|
||||
"""模块/插件主目录.
|
||||
|
||||
通过此属性获取模块/插件主目录.
|
||||
"""
|
||||
my_home = self.astrbot_root / "home" / self.name
|
||||
my_home.mkdir(parents=True, exist_ok=True)
|
||||
return my_home
|
||||
|
||||
@property
|
||||
def config(self) -> Path:
|
||||
"""返回模块/插件配置目录.
|
||||
|
||||
搭配 astrbot_config 使用.
|
||||
"""
|
||||
config_path = self.astrbot_root / "config" / self.name
|
||||
config_path.mkdir(parents=True, exist_ok=True)
|
||||
return config_path
|
||||
|
||||
@property
|
||||
def data(self) -> Path:
|
||||
"""返回模块/插件数据目录."""
|
||||
data_path = self.astrbot_root / "data" / self.name
|
||||
data_path.mkdir(parents=True, exist_ok=True)
|
||||
return data_path
|
||||
|
||||
@property
|
||||
def log(self) -> Path:
|
||||
"""返回模块日志目录."""
|
||||
log_path = self.astrbot_root / "logs" / self.name
|
||||
log_path.mkdir(parents=True, exist_ok=True)
|
||||
return log_path
|
||||
|
||||
@property
|
||||
def temp(self) -> Path:
|
||||
"""返回模块临时文件目录."""
|
||||
temp_path = self.astrbot_root / "temp" / self.name
|
||||
temp_path.mkdir(parents=True, exist_ok=True)
|
||||
return temp_path
|
||||
|
||||
@property
|
||||
def plugins(self) -> Path:
|
||||
"""返回插件目录."""
|
||||
plugin_path = self.astrbot_root / "plugins" / self.name
|
||||
plugin_path.mkdir(parents=True, exist_ok=True)
|
||||
return plugin_path
|
||||
|
||||
@classmethod
|
||||
def is_root(cls, path: Path) -> bool:
|
||||
"""检查路径是否为 Astrbot 根目录."""
|
||||
if not path.exists() or not path.is_dir():
|
||||
return False
|
||||
# 检查此目录内是是否包含.astrbot标记文件
|
||||
return bool((path / ".astrbot").exists())
|
||||
|
||||
def reload(self) -> None:
|
||||
"""重新加载环境变量."""
|
||||
load_dotenv()
|
||||
self.__class__.astrbot_root = Path(
|
||||
getenv("ASTRBOT_ROOT", Path.home() / ".astrbot")
|
||||
).absolute()
|
||||
|
||||
@contextmanager
|
||||
def chdir(self, cwd: str = "home") -> Generator[Path]:
|
||||
"""临时切换到指定目录, 子进程将继承此 CWD。"""
|
||||
original_cwd = Path.cwd()
|
||||
target_dir = self.root / cwd
|
||||
try:
|
||||
chdir(target_dir)
|
||||
yield target_dir
|
||||
finally:
|
||||
chdir(original_cwd)
|
||||
|
||||
@asynccontextmanager
|
||||
async def achdir(self, cwd: str = "home") -> AsyncGenerator[Path]:
|
||||
"""异步上下文管理器: 临时切换到指定目录, 子进程将继承此 CWD。"""
|
||||
original_cwd = Path.cwd()
|
||||
target_dir = self.root / cwd
|
||||
try:
|
||||
chdir(target_dir)
|
||||
yield target_dir
|
||||
finally:
|
||||
chdir(original_cwd)
|
||||
19
astrbot_sdk/src/astrbot_sdk/base/provider.py
Normal file
19
astrbot_sdk/src/astrbot_sdk/base/provider.py
Normal file
@@ -0,0 +1,19 @@
|
||||
from dishka import Provider, Scope, provide
|
||||
|
||||
from astrbot_api import IAstrbotPaths
|
||||
|
||||
from .paths import AstrbotPaths
|
||||
|
||||
|
||||
class AstrbotBaseProvider(Provider):
|
||||
scope = Scope.APP # 基础Provider的作用域设为APP
|
||||
|
||||
@provide
|
||||
def get_astrbot_paths_cls(self) -> type[IAstrbotPaths]:
|
||||
return AstrbotPaths
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
0
astrbot_sdk/src/astrbot_sdk/py.typed
Normal file
0
astrbot_sdk/src/astrbot_sdk/py.typed
Normal file
105
main.py
105
main.py
@@ -1,105 +1,4 @@
|
||||
import argparse
|
||||
import asyncio
|
||||
import mimetypes
|
||||
import os
|
||||
import sys
|
||||
from pathlib import Path
|
||||
|
||||
from astrbot.core import LogBroker, LogManager, db_helper, logger
|
||||
from astrbot.core.config.default import VERSION
|
||||
from astrbot.core.initial_loader import InitialLoader
|
||||
from astrbot.core.utils.astrbot_path import get_astrbot_data_path
|
||||
from astrbot.core.utils.io import download_dashboard, get_dashboard_version
|
||||
|
||||
# 将父目录添加到 sys.path
|
||||
sys.path.append(Path(__file__).parent.as_posix())
|
||||
|
||||
logo_tmpl = r"""
|
||||
___ _______.___________..______ .______ ______ .___________.
|
||||
/ \ / | || _ \ | _ \ / __ \ | |
|
||||
/ ^ \ | (----`---| |----`| |_) | | |_) | | | | | `---| |----`
|
||||
/ /_\ \ \ \ | | | / | _ < | | | | | |
|
||||
/ _____ \ .----) | | | | |\ \----.| |_) | | `--' | | |
|
||||
/__/ \__\ |_______/ |__| | _| `._____||______/ \______/ |__|
|
||||
|
||||
"""
|
||||
|
||||
|
||||
def check_env():
|
||||
if not (sys.version_info.major == 3 and sys.version_info.minor >= 10):
|
||||
logger.error("请使用 Python3.10+ 运行本项目。")
|
||||
exit()
|
||||
|
||||
os.makedirs("data/config", exist_ok=True)
|
||||
os.makedirs("data/plugins", exist_ok=True)
|
||||
os.makedirs("data/temp", exist_ok=True)
|
||||
|
||||
# 针对问题 #181 的临时解决方案
|
||||
mimetypes.add_type("text/javascript", ".js")
|
||||
mimetypes.add_type("text/javascript", ".mjs")
|
||||
mimetypes.add_type("application/json", ".json")
|
||||
|
||||
|
||||
async def check_dashboard_files(webui_dir: str | None = None):
|
||||
"""下载管理面板文件"""
|
||||
# 指定webui目录
|
||||
if webui_dir:
|
||||
if os.path.exists(webui_dir):
|
||||
logger.info(f"使用指定的 WebUI 目录: {webui_dir}")
|
||||
return webui_dir
|
||||
logger.warning(f"指定的 WebUI 目录 {webui_dir} 不存在,将使用默认逻辑。")
|
||||
|
||||
data_dist_path = os.path.join(get_astrbot_data_path(), "dist")
|
||||
if os.path.exists(data_dist_path):
|
||||
v = await get_dashboard_version()
|
||||
if v is not None:
|
||||
# 存在文件
|
||||
if v == f"v{VERSION}":
|
||||
logger.info("WebUI 版本已是最新。")
|
||||
else:
|
||||
logger.warning(
|
||||
f"检测到 WebUI 版本 ({v}) 与当前 AstrBot 版本 (v{VERSION}) 不符。",
|
||||
)
|
||||
return data_dist_path
|
||||
|
||||
logger.info(
|
||||
"开始下载管理面板文件...高峰期(晚上)可能导致较慢的速度。如多次下载失败,请前往 https://github.com/AstrBotDevs/AstrBot/releases/latest 下载 dist.zip,并将其中的 dist 文件夹解压至 data 目录下。",
|
||||
)
|
||||
|
||||
try:
|
||||
await download_dashboard(version=f"v{VERSION}", latest=False)
|
||||
except Exception as e:
|
||||
logger.critical(f"下载管理面板文件失败: {e}。")
|
||||
return None
|
||||
|
||||
logger.info("管理面板下载完成。")
|
||||
return data_dist_path
|
||||
|
||||
from astrbot.__main__ import main
|
||||
|
||||
if __name__ == "__main__":
|
||||
parser = argparse.ArgumentParser(description="AstrBot")
|
||||
parser.add_argument(
|
||||
"--webui-dir",
|
||||
type=str,
|
||||
help="指定 WebUI 静态文件目录路径",
|
||||
default=None,
|
||||
)
|
||||
args = parser.parse_args()
|
||||
|
||||
check_env()
|
||||
|
||||
# 启动日志代理
|
||||
log_broker = LogBroker()
|
||||
LogManager.set_queue_handler(logger, log_broker)
|
||||
|
||||
# 检查仪表板文件
|
||||
webui_dir = asyncio.run(check_dashboard_files(args.webui_dir))
|
||||
|
||||
db = db_helper
|
||||
|
||||
# 打印 logo
|
||||
logger.info(logo_tmpl)
|
||||
|
||||
core_lifecycle = InitialLoader(db, log_broker)
|
||||
core_lifecycle.webui_dir = webui_dir
|
||||
asyncio.run(core_lifecycle.start())
|
||||
main()
|
||||
|
||||
@@ -9,12 +9,15 @@ from collections import defaultdict
|
||||
|
||||
import aiodocker
|
||||
import aiohttp
|
||||
from astrbot_api.abc import IAstrbotPaths
|
||||
|
||||
from astrbot.api import llm_tool, logger, star
|
||||
from astrbot.api.event import AstrMessageEvent, MessageEventResult, filter
|
||||
from astrbot.api.message_components import File, Image
|
||||
from astrbot.api.provider import ProviderRequest
|
||||
from astrbot.core.utils.astrbot_path import get_astrbot_data_path
|
||||
from astrbot_sdk import sync_base_container
|
||||
|
||||
AstrbotPaths: type[IAstrbotPaths] = sync_base_container.get(type[IAstrbotPaths])
|
||||
from astrbot.core.utils.io import download_file, download_image_by_url
|
||||
|
||||
PROMPT = """
|
||||
@@ -91,7 +94,7 @@ DEFAULT_CONFIG = {
|
||||
},
|
||||
"docker_host_astrbot_abs_path": "",
|
||||
}
|
||||
PATH = os.path.join(get_astrbot_data_path(), "config", "python_interpreter.json")
|
||||
PATH = str(AstrbotPaths.astrbot_root / "config" / "python_interpreter.json")
|
||||
|
||||
|
||||
class Main(star.Star):
|
||||
@@ -101,13 +104,15 @@ class Main(star.Star):
|
||||
self.context = context
|
||||
self.curr_dir = os.path.dirname(os.path.abspath(__file__))
|
||||
|
||||
self.shared_path = os.path.join("data", "py_interpreter_shared")
|
||||
self.shared_path = str(AstrbotPaths.astrbot_root / "py_interpreter_shared")
|
||||
if not os.path.exists(self.shared_path):
|
||||
# 复制 api.py 到 shared 目录
|
||||
os.makedirs(self.shared_path, exist_ok=True)
|
||||
shared_api_file = os.path.join(self.curr_dir, "shared", "api.py")
|
||||
shutil.copy(shared_api_file, self.shared_path)
|
||||
self.workplace_path = os.path.join("data", "py_interpreter_workplace")
|
||||
self.workplace_path = str(
|
||||
AstrbotPaths.astrbot_root / "py_interpreter_workplace"
|
||||
)
|
||||
os.makedirs(self.workplace_path, exist_ok=True)
|
||||
|
||||
self.user_file_msg_buffer = defaultdict(list)
|
||||
@@ -212,8 +217,7 @@ class Main(star.Star):
|
||||
file_path = await comp.get_file()
|
||||
if file_path.startswith("http"):
|
||||
name = comp.name if comp.name else uuid.uuid4().hex[:8]
|
||||
temp_dir = os.path.join(get_astrbot_data_path(), "temp")
|
||||
path = os.path.join(temp_dir, name)
|
||||
path = str(AstrbotPaths.astrbot_root / "temp" / name)
|
||||
await download_file(file_path, path)
|
||||
else:
|
||||
path = file_path
|
||||
|
||||
@@ -5,9 +5,13 @@ import uuid
|
||||
import zoneinfo
|
||||
|
||||
from apscheduler.schedulers.asyncio import AsyncIOScheduler
|
||||
from astrbot_api.abc import IAstrbotPaths
|
||||
|
||||
from astrbot.api import llm_tool, logger, star
|
||||
from astrbot.api.event import AstrMessageEvent, MessageEventResult, filter
|
||||
from astrbot_sdk import sync_base_container
|
||||
|
||||
AstrbotPaths: type[IAstrbotPaths] = sync_base_container.get(type[IAstrbotPaths])
|
||||
from astrbot.core.utils.astrbot_path import get_astrbot_data_path
|
||||
|
||||
|
||||
@@ -27,7 +31,7 @@ class Main(star.Star):
|
||||
self.scheduler = AsyncIOScheduler(timezone=self.timezone)
|
||||
|
||||
# set and load config
|
||||
reminder_file = os.path.join(get_astrbot_data_path(), "astrbot-reminder.json")
|
||||
reminder_file = str(AstrbotPaths.astrbot_root / "astrbot-reminder.json")
|
||||
if not os.path.exists(reminder_file):
|
||||
with open(reminder_file, "w", encoding="utf-8") as f:
|
||||
f.write("{}")
|
||||
|
||||
@@ -63,6 +63,10 @@ dependencies = [
|
||||
"jieba>=0.42.1",
|
||||
"markitdown-no-magika[docx,xls,xlsx]>=0.1.2",
|
||||
"xinference-client",
|
||||
"dotenv>=0.9.9",
|
||||
"astrbot-api",
|
||||
"astrbot-sdk",
|
||||
"dishka>=1.7.2",
|
||||
]
|
||||
|
||||
[dependency-groups]
|
||||
@@ -103,13 +107,22 @@ ignore = [
|
||||
"E501",
|
||||
"ASYNC230" # TODO: handle ASYNC230 in AstrBot
|
||||
]
|
||||
[tool.uv.workspace]
|
||||
members = [
|
||||
"astrbot_api",
|
||||
"astrbot_sdk",
|
||||
]
|
||||
|
||||
[tool.uv.sources]
|
||||
astrbot-api = { workspace = true }
|
||||
astrbot-sdk = { workspace = true }
|
||||
|
||||
[tool.pyright]
|
||||
typeCheckingMode = "basic"
|
||||
pythonVersion = "3.10"
|
||||
reportMissingTypeStubs = false
|
||||
reportMissingImports = false
|
||||
include = ["astrbot","packages"]
|
||||
include = ["astrbot","astrbot_api","astrbot_sdk","packages"]
|
||||
exclude = ["dashboard", "node_modules", "dist", "data", "tests"]
|
||||
|
||||
[tool.hatch.version]
|
||||
|
||||
517
tests/test_paths.py
Normal file
517
tests/test_paths.py
Normal file
@@ -0,0 +1,517 @@
|
||||
"""测试 AstrbotPaths 路径类的综合测试."""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import os
|
||||
import tempfile
|
||||
from pathlib import Path
|
||||
from typing import TYPE_CHECKING
|
||||
|
||||
import pytest
|
||||
from astrbot_api.abc import IAstrbotPaths
|
||||
|
||||
from astrbot_sdk import sync_base_container
|
||||
|
||||
AstrbotPaths: type[IAstrbotPaths] = sync_base_container.get(type[IAstrbotPaths])
|
||||
|
||||
if TYPE_CHECKING:
|
||||
from collections.abc import Generator
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def temp_root(monkeypatch: pytest.MonkeyPatch) -> Generator[Path]:
|
||||
"""创建一个临时根目录用于测试."""
|
||||
with tempfile.TemporaryDirectory() as tmpdir:
|
||||
temp_path = Path(tmpdir)
|
||||
monkeypatch.setenv("ASTRBOT_ROOT", str(temp_path))
|
||||
# 清除类变量和实例缓存
|
||||
AstrbotPaths._instances.clear()
|
||||
# 重新加载环境变量
|
||||
from dotenv import load_dotenv
|
||||
|
||||
load_dotenv(override=True)
|
||||
AstrbotPaths.astrbot_root = temp_path
|
||||
yield temp_path
|
||||
# 清理
|
||||
AstrbotPaths._instances.clear()
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def paths_instance(temp_root: Path) -> AstrbotPaths:
|
||||
"""创建一个 AstrbotPaths 实例用于测试."""
|
||||
return AstrbotPaths.getPaths("test-module")
|
||||
|
||||
|
||||
class TestAstrbotPathsInit:
|
||||
"""测试 AstrbotPaths 初始化."""
|
||||
|
||||
def test_init_creates_root_directory(self, temp_root: Path) -> None:
|
||||
"""测试初始化时创建根目录."""
|
||||
# 删除根目录以测试自动创建
|
||||
if temp_root.exists():
|
||||
import shutil
|
||||
|
||||
shutil.rmtree(temp_root)
|
||||
|
||||
AstrbotPaths("test-init")
|
||||
assert temp_root.exists()
|
||||
assert temp_root.is_dir()
|
||||
|
||||
def test_init_with_name(self, temp_root: Path) -> None:
|
||||
"""测试使用名称初始化."""
|
||||
paths = AstrbotPaths("my-module")
|
||||
assert paths.name == "my-module"
|
||||
|
||||
def test_astrbot_root_from_env(
|
||||
self, monkeypatch: pytest.MonkeyPatch, tmp_path: Path
|
||||
) -> None:
|
||||
"""测试从环境变量读取根目录."""
|
||||
custom_root = tmp_path / "custom_root"
|
||||
custom_root.mkdir(parents=True, exist_ok=True)
|
||||
|
||||
# 清除实例缓存
|
||||
AstrbotPaths._instances.clear()
|
||||
|
||||
# 直接设置环境变量(在 load_dotenv 之前)
|
||||
monkeypatch.setenv("ASTRBOT_ROOT", str(custom_root))
|
||||
|
||||
# 直接更新 astrbot_root(模拟 load_dotenv 的效果但使用我们设置的环境变量)
|
||||
AstrbotPaths.astrbot_root = Path(
|
||||
os.getenv("ASTRBOT_ROOT", Path.home() / ".astrbot")
|
||||
).absolute()
|
||||
|
||||
assert AstrbotPaths.astrbot_root == custom_root.absolute()
|
||||
|
||||
def test_astrbot_root_default(
|
||||
self, monkeypatch: pytest.MonkeyPatch, tmp_path: Path
|
||||
) -> None:
|
||||
"""测试默认根目录."""
|
||||
# 清除环境变量
|
||||
monkeypatch.delenv("ASTRBOT_ROOT", raising=False)
|
||||
# 清除任何可能存在的 .env 文件影响
|
||||
monkeypatch.setattr("os.environ", {**os.environ})
|
||||
|
||||
# 清除实例缓存
|
||||
AstrbotPaths._instances.clear()
|
||||
|
||||
# 重新计算根目录
|
||||
AstrbotPaths.astrbot_root = Path(
|
||||
os.getenv("ASTRBOT_ROOT", Path.home() / ".astrbot")
|
||||
).absolute()
|
||||
|
||||
expected = (Path.home() / ".astrbot").absolute()
|
||||
assert AstrbotPaths.astrbot_root == expected
|
||||
|
||||
|
||||
class TestGetPaths:
|
||||
"""测试 getPaths 单例模式."""
|
||||
|
||||
def test_get_paths_returns_same_instance(self, temp_root: Path) -> None:
|
||||
"""测试多次调用返回同一个实例."""
|
||||
paths1 = AstrbotPaths.getPaths("test-module")
|
||||
paths2 = AstrbotPaths.getPaths("test-module")
|
||||
assert paths1 is paths2
|
||||
|
||||
def test_get_paths_different_names(self, temp_root: Path) -> None:
|
||||
"""测试不同名称返回不同实例."""
|
||||
paths1 = AstrbotPaths.getPaths("module-a")
|
||||
paths2 = AstrbotPaths.getPaths("module-b")
|
||||
assert paths1 is not paths2
|
||||
assert paths1.name == "module-a"
|
||||
assert paths2.name == "module-b"
|
||||
|
||||
def test_get_paths_normalizes_name(self, temp_root: Path) -> None:
|
||||
"""测试名称规范化."""
|
||||
# PEP 503 规范化: 转小写, 替换 -, _, .
|
||||
paths1 = AstrbotPaths.getPaths("Test_Module")
|
||||
paths2 = AstrbotPaths.getPaths("test-module")
|
||||
paths3 = AstrbotPaths.getPaths("TEST.MODULE")
|
||||
|
||||
# 所有这些名称应该被规范化为相同的名称
|
||||
assert paths1 is paths2
|
||||
assert paths2 is paths3
|
||||
|
||||
|
||||
class TestProperties:
|
||||
"""测试所有属性访问器."""
|
||||
|
||||
def test_root_property(self, paths_instance: AstrbotPaths, temp_root: Path) -> None:
|
||||
"""测试 root 属性."""
|
||||
assert paths_instance.root == temp_root
|
||||
assert paths_instance.root.exists()
|
||||
|
||||
def test_root_property_when_not_exists(
|
||||
self, monkeypatch: pytest.MonkeyPatch, tmp_path: Path
|
||||
) -> None:
|
||||
"""测试 root 属性当根目录不存在时."""
|
||||
non_existent = tmp_path / "non_existent_path"
|
||||
# 确保目录不存在
|
||||
if non_existent.exists():
|
||||
import shutil
|
||||
|
||||
shutil.rmtree(non_existent)
|
||||
|
||||
# 清除实例缓存
|
||||
AstrbotPaths._instances.clear()
|
||||
# 设置不存在的路径
|
||||
AstrbotPaths.astrbot_root = non_existent
|
||||
|
||||
# __init__ 会创建根目录,所以 getPaths 会使根目录存在
|
||||
# 我们测试的是在 __init__ 创建目录之前访问 root 属性的行为
|
||||
# 但由于 getPaths 总是调用 __init__,目录总是会被创建
|
||||
# 所以这个测试应该验证即使最初不存在,getPaths 之后也会存在
|
||||
paths = AstrbotPaths.getPaths("test")
|
||||
# getPaths 调用 __init__,__init__ 会创建根目录
|
||||
# 所以 root 应该返回 astrbot_root(现在已存在)
|
||||
assert paths.root == non_existent
|
||||
assert non_existent.exists()
|
||||
|
||||
def test_root_property_fallback_to_cwd(
|
||||
self, monkeypatch: pytest.MonkeyPatch, tmp_path: Path
|
||||
) -> None:
|
||||
"""测试 root 属性在根目录被删除后回退到 cwd/.astrbot."""
|
||||
import shutil
|
||||
|
||||
# 创建并设置一个根目录
|
||||
temp_root = tmp_path / "test_root"
|
||||
temp_root.mkdir(parents=True, exist_ok=True)
|
||||
|
||||
# 清除实例缓存
|
||||
AstrbotPaths._instances.clear()
|
||||
AstrbotPaths.astrbot_root = temp_root
|
||||
|
||||
# 创建实例
|
||||
paths = AstrbotPaths.getPaths("test-fallback")
|
||||
|
||||
# 删除根目录(模拟被外部删除的情况)
|
||||
shutil.rmtree(temp_root)
|
||||
|
||||
# 现在访问 root 应该回退到 cwd/.astrbot
|
||||
expected = Path.cwd() / ".astrbot"
|
||||
assert paths.root == expected
|
||||
|
||||
def test_home_property(self, paths_instance: AstrbotPaths, temp_root: Path) -> None:
|
||||
"""测试 home 属性."""
|
||||
home_path = paths_instance.home
|
||||
expected = temp_root / "home" / paths_instance.name
|
||||
assert home_path == expected
|
||||
assert home_path.exists()
|
||||
assert home_path.is_dir()
|
||||
|
||||
def test_config_property(
|
||||
self, paths_instance: AstrbotPaths, temp_root: Path
|
||||
) -> None:
|
||||
"""测试 config 属性."""
|
||||
config_path = paths_instance.config
|
||||
expected = temp_root / "config" / paths_instance.name
|
||||
assert config_path == expected
|
||||
assert config_path.exists()
|
||||
assert config_path.is_dir()
|
||||
|
||||
def test_data_property(self, paths_instance: AstrbotPaths, temp_root: Path) -> None:
|
||||
"""测试 data 属性."""
|
||||
data_path = paths_instance.data
|
||||
expected = temp_root / "data" / paths_instance.name
|
||||
assert data_path == expected
|
||||
assert data_path.exists()
|
||||
assert data_path.is_dir()
|
||||
|
||||
def test_log_property(self, paths_instance: AstrbotPaths, temp_root: Path) -> None:
|
||||
"""测试 log 属性."""
|
||||
log_path = paths_instance.log
|
||||
expected = temp_root / "logs" / paths_instance.name
|
||||
assert log_path == expected
|
||||
assert log_path.exists()
|
||||
assert log_path.is_dir()
|
||||
|
||||
def test_temp_property(self, paths_instance: AstrbotPaths, temp_root: Path) -> None:
|
||||
"""测试 temp 属性."""
|
||||
temp_path = paths_instance.temp
|
||||
expected = temp_root / "temp" / paths_instance.name
|
||||
assert temp_path == expected
|
||||
assert temp_path.exists()
|
||||
assert temp_path.is_dir()
|
||||
|
||||
def test_plugins_property(
|
||||
self, paths_instance: AstrbotPaths, temp_root: Path
|
||||
) -> None:
|
||||
"""测试 plugins 属性."""
|
||||
plugins_path = paths_instance.plugins
|
||||
expected = temp_root / "plugins" / paths_instance.name
|
||||
assert plugins_path == expected
|
||||
assert plugins_path.exists()
|
||||
assert plugins_path.is_dir()
|
||||
|
||||
def test_properties_create_nested_directories(
|
||||
self, paths_instance: AstrbotPaths, temp_root: Path
|
||||
) -> None:
|
||||
"""测试属性访问时创建嵌套目录."""
|
||||
# 清空目录
|
||||
import shutil
|
||||
|
||||
if temp_root.exists():
|
||||
for item in temp_root.iterdir():
|
||||
if item.is_dir():
|
||||
shutil.rmtree(item)
|
||||
else:
|
||||
item.unlink()
|
||||
|
||||
# 访问所有属性
|
||||
_ = paths_instance.home
|
||||
_ = paths_instance.config
|
||||
_ = paths_instance.data
|
||||
_ = paths_instance.log
|
||||
_ = paths_instance.temp
|
||||
_ = paths_instance.plugins
|
||||
|
||||
# 验证所有目录都已创建
|
||||
assert (temp_root / "home" / paths_instance.name).exists()
|
||||
assert (temp_root / "config" / paths_instance.name).exists()
|
||||
assert (temp_root / "data" / paths_instance.name).exists()
|
||||
assert (temp_root / "logs" / paths_instance.name).exists()
|
||||
assert (temp_root / "temp" / paths_instance.name).exists()
|
||||
assert (temp_root / "plugins" / paths_instance.name).exists()
|
||||
|
||||
|
||||
class TestIsRoot:
|
||||
"""测试 is_root 类方法."""
|
||||
|
||||
def test_is_root_with_marker_file(self, temp_root: Path) -> None:
|
||||
"""测试带有标记文件的根目录识别."""
|
||||
marker_file = temp_root / ".astrbot"
|
||||
marker_file.touch()
|
||||
|
||||
assert AstrbotPaths.is_root(temp_root) is True
|
||||
|
||||
def test_is_root_without_marker_file(self, temp_root: Path) -> None:
|
||||
"""测试没有标记文件的目录."""
|
||||
marker_file = temp_root / ".astrbot"
|
||||
if marker_file.exists():
|
||||
marker_file.unlink()
|
||||
|
||||
assert AstrbotPaths.is_root(temp_root) is False
|
||||
|
||||
def test_is_root_with_non_existent_path(self) -> None:
|
||||
"""测试不存在的路径."""
|
||||
non_existent = Path("/definitely/not/exist/path")
|
||||
assert AstrbotPaths.is_root(non_existent) is False
|
||||
|
||||
def test_is_root_with_file_not_directory(self, temp_root: Path) -> None:
|
||||
"""测试路径是文件而非目录."""
|
||||
test_file = temp_root / "test.txt"
|
||||
test_file.touch()
|
||||
|
||||
assert AstrbotPaths.is_root(test_file) is False
|
||||
|
||||
|
||||
class TestReload:
|
||||
"""测试 reload 方法."""
|
||||
|
||||
def test_reload_updates_root(
|
||||
self, temp_root: Path, monkeypatch: pytest.MonkeyPatch
|
||||
) -> None:
|
||||
"""测试 reload 更新根目录."""
|
||||
paths = AstrbotPaths.getPaths("test-reload")
|
||||
|
||||
# 修改环境变量
|
||||
new_root = temp_root / "new_root"
|
||||
new_root.mkdir(parents=True, exist_ok=True)
|
||||
monkeypatch.setenv("ASTRBOT_ROOT", str(new_root))
|
||||
|
||||
# 重新加载
|
||||
paths.reload()
|
||||
|
||||
# 验证根目录已更新
|
||||
assert AstrbotPaths.astrbot_root == new_root.absolute()
|
||||
|
||||
def test_reload_clears_old_env(
|
||||
self, temp_root: Path, monkeypatch: pytest.MonkeyPatch
|
||||
) -> None:
|
||||
"""测试 reload 在环境变量被删除后使用默认值."""
|
||||
paths = AstrbotPaths.getPaths("test-reload-default")
|
||||
|
||||
# 删除环境变量
|
||||
monkeypatch.delenv("ASTRBOT_ROOT", raising=False)
|
||||
|
||||
# 重新加载
|
||||
paths.reload()
|
||||
|
||||
# 应该使用默认值
|
||||
(Path.home() / ".astrbot").absolute()
|
||||
# 由于 .env 文件可能存在,实际结果可能不变
|
||||
# 所以我们只验证 reload 没有抛出异常
|
||||
assert AstrbotPaths.astrbot_root is not None
|
||||
assert isinstance(AstrbotPaths.astrbot_root, Path)
|
||||
|
||||
|
||||
class TestChdir:
|
||||
"""测试 chdir 上下文管理器."""
|
||||
|
||||
def test_chdir_changes_directory(
|
||||
self, paths_instance: AstrbotPaths, temp_root: Path
|
||||
) -> None:
|
||||
"""测试 chdir 切换目录."""
|
||||
original_cwd = Path.cwd()
|
||||
|
||||
# 创建目标目录
|
||||
target_path = temp_root / "home"
|
||||
target_path.mkdir(parents=True, exist_ok=True)
|
||||
|
||||
with paths_instance.chdir("home") as target_dir:
|
||||
current_cwd = Path.cwd()
|
||||
expected_dir = temp_root / "home"
|
||||
assert current_cwd == expected_dir
|
||||
assert target_dir == expected_dir
|
||||
|
||||
# 验证已恢复原目录
|
||||
assert Path.cwd() == original_cwd
|
||||
|
||||
def test_chdir_restores_on_exception(
|
||||
self, paths_instance: AstrbotPaths, temp_root: Path
|
||||
) -> None:
|
||||
"""测试 chdir 在异常时恢复原目录."""
|
||||
original_cwd = Path.cwd()
|
||||
|
||||
# 创建目标目录
|
||||
target_path = temp_root / "home"
|
||||
target_path.mkdir(parents=True, exist_ok=True)
|
||||
|
||||
with pytest.raises(ValueError):
|
||||
with paths_instance.chdir("home"):
|
||||
raise ValueError("Test exception")
|
||||
|
||||
# 验证已恢复原目录
|
||||
assert Path.cwd() == original_cwd
|
||||
|
||||
def test_chdir_with_different_subdirectories(
|
||||
self, paths_instance: AstrbotPaths, temp_root: Path
|
||||
) -> None:
|
||||
"""测试 chdir 使用不同的子目录."""
|
||||
original_cwd = Path.cwd()
|
||||
|
||||
# 创建测试目录
|
||||
test_dir = temp_root / "test_subdir"
|
||||
test_dir.mkdir(parents=True, exist_ok=True)
|
||||
|
||||
with paths_instance.chdir("test_subdir") as target_dir:
|
||||
assert Path.cwd() == test_dir
|
||||
assert target_dir == test_dir
|
||||
|
||||
assert Path.cwd() == original_cwd
|
||||
|
||||
|
||||
class TestAchdir:
|
||||
"""测试 achdir 异步上下文管理器."""
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_achdir_changes_directory(
|
||||
self, paths_instance: AstrbotPaths, temp_root: Path
|
||||
) -> None:
|
||||
"""测试 achdir 异步切换目录."""
|
||||
original_cwd = Path.cwd()
|
||||
|
||||
# 创建目标目录
|
||||
target_path = temp_root / "home"
|
||||
target_path.mkdir(parents=True, exist_ok=True)
|
||||
|
||||
async with paths_instance.achdir("home") as target_dir:
|
||||
current_cwd = Path.cwd()
|
||||
expected_dir = temp_root / "home"
|
||||
assert current_cwd == expected_dir
|
||||
assert target_dir == expected_dir
|
||||
|
||||
# 验证已恢复原目录
|
||||
assert Path.cwd() == original_cwd
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_achdir_restores_on_exception(
|
||||
self, paths_instance: AstrbotPaths, temp_root: Path
|
||||
) -> None:
|
||||
"""测试 achdir 在异常时恢复原目录."""
|
||||
original_cwd = Path.cwd()
|
||||
|
||||
# 创建目标目录
|
||||
target_path = temp_root / "home"
|
||||
target_path.mkdir(parents=True, exist_ok=True)
|
||||
|
||||
with pytest.raises(ValueError):
|
||||
async with paths_instance.achdir("home"):
|
||||
raise ValueError("Test exception")
|
||||
|
||||
# 验证已恢复原目录
|
||||
assert Path.cwd() == original_cwd
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_achdir_with_different_subdirectories(
|
||||
self, paths_instance: AstrbotPaths, temp_root: Path
|
||||
) -> None:
|
||||
"""测试 achdir 使用不同的子目录."""
|
||||
original_cwd = Path.cwd()
|
||||
|
||||
# 创建测试目录
|
||||
test_dir = temp_root / "async_test_subdir"
|
||||
test_dir.mkdir(parents=True, exist_ok=True)
|
||||
|
||||
async with paths_instance.achdir("async_test_subdir") as target_dir:
|
||||
assert Path.cwd() == test_dir
|
||||
assert target_dir == test_dir
|
||||
|
||||
assert Path.cwd() == original_cwd
|
||||
|
||||
|
||||
class TestIntegration:
|
||||
"""集成测试."""
|
||||
|
||||
def test_multiple_modules_isolated(self, temp_root: Path) -> None:
|
||||
"""测试多个模块之间的隔离."""
|
||||
module_a = AstrbotPaths.getPaths("module-a")
|
||||
module_b = AstrbotPaths.getPaths("module-b")
|
||||
|
||||
# 访问各自的 home 目录
|
||||
home_a = module_a.home
|
||||
home_b = module_b.home
|
||||
|
||||
# 验证目录不同
|
||||
assert home_a != home_b
|
||||
assert home_a == temp_root / "home" / "module-a"
|
||||
assert home_b == temp_root / "home" / "module-b"
|
||||
|
||||
# 验证都存在
|
||||
assert home_a.exists()
|
||||
assert home_b.exists()
|
||||
|
||||
def test_full_workflow(self, temp_root: Path) -> None:
|
||||
"""测试完整工作流."""
|
||||
# 创建一个模块
|
||||
module = AstrbotPaths.getPaths("my-plugin")
|
||||
|
||||
# 创建各种文件
|
||||
config_file = module.config / "settings.json"
|
||||
config_file.write_text('{"key": "value"}')
|
||||
|
||||
data_file = module.data / "data.txt"
|
||||
data_file.write_text("some data")
|
||||
|
||||
log_file = module.log / "app.log"
|
||||
log_file.write_text("log entry")
|
||||
|
||||
# 验证文件存在
|
||||
assert config_file.exists()
|
||||
assert data_file.exists()
|
||||
assert log_file.exists()
|
||||
|
||||
# 验证内容
|
||||
assert config_file.read_text() == '{"key": "value"}'
|
||||
assert data_file.read_text() == "some data"
|
||||
assert log_file.read_text() == "log entry"
|
||||
|
||||
def test_singleton_pattern_thread_safe(self, temp_root: Path) -> None:
|
||||
"""测试单例模式的基本行为(注意:不是真正的线程安全测试)."""
|
||||
instances = []
|
||||
for _ in range(10):
|
||||
instances.append(AstrbotPaths.getPaths("singleton-test"))
|
||||
|
||||
# 所有实例应该是同一个对象
|
||||
first = instances[0]
|
||||
for instance in instances[1:]:
|
||||
assert instance is first
|
||||
Reference in New Issue
Block a user