butubb
55b1b74944
导出：sqlite3 在线备份 API 对 data/users.db 做一致快照 → zip（users.db + manifest.json：schema_version/逐表行数/apk 清单）+ 可选 data/apks/*.apk。 导出文件读入内存(BytesIO)发送后即删磁盘副本，避免 Windows 流式句柄锁残留。 导入：上传 zip/db → 暂存校验（integrity + 必需表 app_meta/user/task_job/ device_group + schema 版本提示）→ 确认应用：先自动快照当前库到 data/backups/pre_restore_*.db，再把备份落到 data/restore_pending/，由 web_server.py 在 init_db 之前 consume 换库——TaskManager 启动时才读库入内存、 Windows 不能热替换正被持有的库文件，故导入必须重启生效。 新增 web/system_api.py（export/preview/apply，全 @admin_required）与目录常量 BACKUP_DIR/RESTORE_STAGING_DIR/RESTORE_PENDING_DIR；.gitignore 排除运行产物。
99 lines
4.8 KiB
Python
99 lines
4.8 KiB
Python
"""项目根配置：只放核心程序级配置。
|
||
|
||
任务自身的参数（观看时长、点赞开关、滑动速度等）不在这里，
|
||
放各自 tasks/xxx.py 顶部，保持每个任务自包含、互不干扰。
|
||
|
||
改 STF 地址/token、adb 路径、web 端口都改这里。
|
||
|
||
安全：STF_TOKEN / WEB_SECRET_KEY 建议用环境变量注入（生产不要提交到 git）。
|
||
支持 `.env` 文件（项目根目录），也可直接 `export STF_TOKEN=xxx`。
|
||
"""
|
||
import os
|
||
|
||
# 可选加载 .env 文件（不存在则忽略）
|
||
_env_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), ".env")
|
||
if os.path.exists(_env_path):
|
||
try:
|
||
with open(_env_path, encoding="utf-8") as _f:
|
||
for _line in _f:
|
||
_line = _line.strip()
|
||
if _line and not _line.startswith("#") and "=" in _line:
|
||
_k, _v = _line.split("=", 1)
|
||
os.environ.setdefault(_k.strip(), _v.strip())
|
||
except OSError:
|
||
pass
|
||
|
||
|
||
def _env(key, default):
|
||
"""读环境变量，未设置则返回默认值。"""
|
||
return os.environ.get(key, default)
|
||
|
||
|
||
# ================== STF 配置（已废弃，阶段 3 摘除 STF 后仅保留供历史参考） ==================
|
||
STF_URL = _env("STF_URL", "")
|
||
STF_TOKEN = _env("STF_TOKEN", "") # 写入 .env，不要提交到 git
|
||
|
||
# ================== adb 路径 ==================
|
||
# adb 二进制及依赖统一放在 bin/adb/ 下
|
||
# Windows: adb.exe（依赖同目录的 dll）
|
||
# Linux: adb（需 chmod +x，Linux 版二进制）
|
||
# 代码自动识别操作系统，用户只需把对应平台的 adb 放进 bin/adb/
|
||
import sys as _sys
|
||
_ADB_DIR = os.path.join(os.path.dirname(os.path.abspath(__file__)), "bin", "adb")
|
||
if _sys.platform == "win32":
|
||
ADB_PATH = os.path.join(_ADB_DIR, "adb.exe")
|
||
else:
|
||
# Linux / macOS：用 bin/adb/adb（无 .exe 后缀）
|
||
# 部署后需执行 chmod +x bin/adb/adb
|
||
ADB_PATH = os.path.join(_ADB_DIR, "adb")
|
||
|
||
# ================== Web 后台 ==================
|
||
WEB_HOST = "0.0.0.0"
|
||
# 端口必须避开 Windows 动态端口范围（本机被改成 1024-15000，默认 49152-65535）
|
||
# 用 18050 处于动态范围之外，不会被出站连接随机占用，避免 WinError 10013
|
||
WEB_PORT = 18050
|
||
|
||
# ================== 数据目录 ==================
|
||
# 持久化数据（分组、任务计划）存放目录
|
||
DATA_DIR = os.path.join(os.path.dirname(os.path.abspath(__file__)), "data")
|
||
|
||
# APK 文件存储目录（应用管理功能）
|
||
APK_DIR = os.path.join(DATA_DIR, "apks")
|
||
|
||
# ================== 系统数据备份（导出/导入） ==================
|
||
# 导出 zip 与恢复前快照存放；导入暂存目录；待下次启动生效的恢复目录
|
||
# （运行时产物，不入 git，见 .gitignore data/backups 等）
|
||
BACKUP_DIR = os.path.join(DATA_DIR, "backups")
|
||
RESTORE_STAGING_DIR = os.path.join(DATA_DIR, "restore_staging")
|
||
RESTORE_PENDING_DIR = os.path.join(DATA_DIR, "restore_pending")
|
||
|
||
# ================== SSH 到部署机（已废弃，STF 相关操作退役） ==================
|
||
# 阶段 3 后代码不再使用 SSH；保留配置仅供手动运维 220（如 docker stop stf）。
|
||
STF_SSH_TARGET = _env("STF_SSH_TARGET", "")
|
||
STF_SSH_PASSWORD = _env("STF_SSH_PASSWORD", "")
|
||
STF_DOCKER_CONTAINER = _env("STF_DOCKER_CONTAINER", "")
|
||
STF_ADB_CONTAINER = _env("STF_ADB_CONTAINER", "")
|
||
STF_SCRIPT_PATH = _env("STF_SCRIPT_PATH", "")
|
||
|
||
# ================== USB 设备远程 adb server（阶段 2） ==================
|
||
# USB 设备插在 220（STF 主机）上，本机 adb 看不到；220 的 adb 容器是 host 网络
|
||
# 模式，5037 已监听所有网卡（含 Tailscale），平台用 adb -H/-P 直接驱动。
|
||
# 平台自身 USB（serial 无冒号且在本机 adb 里）不受影响，仍走本地。
|
||
USB_ADB_HOST = _env("USB_ADB_HOST", "100.100.10.1") # 220 的 Tailscale IP
|
||
USB_ADB_PORT = int(_env("USB_ADB_PORT", "5037"))
|
||
|
||
# ================== Tailscale 管理（工具页） ==================
|
||
# 调用官方 API v2 管理 tailnet 设备。配置从 .env 读取：
|
||
# TAILSCALE_API_KEY：后台 → Settings → API Access Tokens 生成（或 OAuth client_id:secret）
|
||
# TAILSCALE_TAILNET：tailnet 名称/ID，个人账号一般是登录邮箱前缀
|
||
# 安全：密钥用环境变量/.env 注入，不要写死在代码或提交到 git。
|
||
TAILSCALE_API_KEY = _env("TAILSCALE_API_KEY", "") # 写入 .env，不要提交到 git
|
||
TAILSCALE_TAILNET = _env("TAILSCALE_TAILNET", "")
|
||
|
||
# ================== 设备自动发现 ==================
|
||
# 定时扫描网段中开放 adb 5555 的设备 → 待连接池（用户确认后才加入正式池）。
|
||
# 网段默认：局域网 + Tailscale；可在工具页设备池面板修改（存 app_meta）。
|
||
DISCOVERY_PORT = int(_env("DISCOVERY_PORT", "5555"))
|
||
DISCOVERY_SUBNETS = ["192.168.20.0/24", "100.100.10.0/24"]
|
||
DISCOVERY_INTERVAL = int(_env("DISCOVERY_INTERVAL", "60"))
|