141 lines
3.8 KiB
Python
141 lines
3.8 KiB
Python
"""
|
|
配置管理模块
|
|
"""
|
|
import os
|
|
import json
|
|
from pathlib import Path
|
|
from datetime import datetime
|
|
|
|
# 基础路径
|
|
BASE_DIR = Path(__file__).parent.absolute()
|
|
DATA_DIR = BASE_DIR / "data"
|
|
DB_PATH = DATA_DIR / "events.db"
|
|
CONFIG_PATH = DATA_DIR / "config.json"
|
|
|
|
# 确保目录存在
|
|
DATA_DIR.mkdir(parents=True, exist_ok=True)
|
|
|
|
# Web 服务配置
|
|
WEB_PORT = 19016
|
|
WEB_HOST = "0.0.0.0"
|
|
|
|
# 默认配置
|
|
DEFAULT_CONFIG = {
|
|
"images_dir": str(DATA_DIR / "images"),
|
|
"camera_index": 0,
|
|
"capture_interval": 60,
|
|
"auto_analyze": True,
|
|
"refresh_interval": 5, # 页面刷新间隔（秒）
|
|
"display_limit": 20, # 显示最近多少条
|
|
"vision_api_url": os.environ.get("VISION_API_URL", "https://dashscope.aliyuncs.com/compatible-mode/v1"),
|
|
"vision_api_key": os.environ.get("VISION_API_KEY", "sk-lm-fuP5tGU8:Hi7YU87jHyDP6Ay8Tl2j"),
|
|
"vision_model": os.environ.get("VISION_MODEL", "qwen-vl-plus")
|
|
}
|
|
|
|
# 分析配置
|
|
ANALYSIS_PROMPT = """请分析这张图片，识别其中的重要事件或变化。
|
|
|
|
重点关注：
|
|
1. 人物活动（有人出现、离开、动作等）
|
|
2. 物体变化（物品移动、新增、消失等）
|
|
3. 环境变化（光线、天气、状态等）
|
|
4. 异常情况（潜在危险、异常行为等）
|
|
|
|
请用简洁的中文描述你观察到的事件，格式如下：
|
|
事件类型: [类型]
|
|
描述: [简短描述]
|
|
置信度: [高/中/低]
|
|
|
|
如果没有明显事件，请回答"无明显事件"。
|
|
"""
|
|
|
|
# 数据库配置
|
|
DB_SCHEMA = """
|
|
CREATE TABLE IF NOT EXISTS config (
|
|
key TEXT PRIMARY KEY,
|
|
value TEXT
|
|
);
|
|
|
|
CREATE TABLE IF NOT EXISTS images (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
path TEXT NOT NULL,
|
|
timestamp DATETIME DEFAULT CURRENT_TIMESTAMP,
|
|
camera_id INTEGER DEFAULT 0,
|
|
analyzed BOOLEAN DEFAULT 0,
|
|
date_folder TEXT
|
|
);
|
|
|
|
CREATE TABLE IF NOT EXISTS events (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
image_id INTEGER NOT NULL,
|
|
event_type TEXT,
|
|
description TEXT,
|
|
confidence TEXT,
|
|
timestamp DATETIME DEFAULT CURRENT_TIMESTAMP,
|
|
FOREIGN KEY (image_id) REFERENCES images(id)
|
|
);
|
|
|
|
CREATE INDEX IF NOT EXISTS idx_images_timestamp ON images(timestamp);
|
|
CREATE INDEX IF NOT EXISTS idx_images_date_folder ON images(date_folder);
|
|
CREATE INDEX IF NOT EXISTS idx_events_type ON events(event_type);
|
|
"""
|
|
|
|
|
|
class ConfigManager:
|
|
"""配置管理器"""
|
|
|
|
def __init__(self):
|
|
self.config = DEFAULT_CONFIG.copy()
|
|
self.load()
|
|
|
|
def load(self):
|
|
"""加载配置"""
|
|
if CONFIG_PATH.exists():
|
|
try:
|
|
with open(CONFIG_PATH, 'r', encoding='utf-8') as f:
|
|
saved = json.load(f)
|
|
self.config.update(saved)
|
|
except:
|
|
pass
|
|
|
|
def save(self):
|
|
"""保存配置"""
|
|
with open(CONFIG_PATH, 'w', encoding='utf-8') as f:
|
|
json.dump(self.config, f, ensure_ascii=False, indent=2)
|
|
|
|
def get(self, key, default=None):
|
|
"""获取配置项"""
|
|
return self.config.get(key, default)
|
|
|
|
def set(self, key, value):
|
|
"""设置配置项"""
|
|
self.config[key] = value
|
|
self.save()
|
|
|
|
def get_images_dir(self, date=None):
|
|
"""获取图片保存目录（按日期）"""
|
|
base_dir = Path(self.config['images_dir'])
|
|
|
|
if date is None:
|
|
date = datetime.now().strftime('%Y-%m-%d')
|
|
|
|
# 每天的图片放在单独文件夹
|
|
date_folder = base_dir / date
|
|
date_folder.mkdir(parents=True, exist_ok=True)
|
|
|
|
return date_folder
|
|
|
|
def get_all(self):
|
|
"""获取所有配置"""
|
|
return self.config.copy()
|
|
|
|
def update(self, updates):
|
|
"""批量更新配置"""
|
|
for key, value in updates.items():
|
|
if key in self.config:
|
|
self.config[key] = value
|
|
self.save()
|
|
|
|
|
|
# 全局配置实例
|
|
config_mgr = ConfigManager() |