161 lines
4.2 KiB
Python
161 lines
4.2 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,
|
||
|
||
# 人体检测设置（YOLO）
|
||
"use_yolo": True,
|
||
"yolo_min_confidence": 0.3,
|
||
|
||
# 人员识别设置
|
||
"use_face_recognition": True,
|
||
"use_mediapipe_face": True,
|
||
"use_color_histogram": True,
|
||
"face_match_threshold": 0.6,
|
||
|
||
# 连续判断设置
|
||
"confirm_frames": 3,
|
||
"leave_frames": 2,
|
||
|
||
# AI大模型分析
|
||
"use_vision_api": False,
|
||
"vision_api_trigger": "person_change",
|
||
|
||
# Vision API 配置
|
||
"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() |