修改路径,从src放到根目录

This commit is contained in:
zqc
2026-01-08 10:32:36 +08:00
parent 96589ebdbd
commit f86effd63c
37 changed files with 51 additions and 410 deletions

View File

@@ -0,0 +1,74 @@
"""
配置相关数据访问层
"""
from typing import List, Dict, Optional
from sqlalchemy.orm import Session
from models.sur_config import SurConfig, SurConfigBase
from config import settings
class SurConfigRepository:
"""配置数据访问类"""
def __init__(self, db: Session):
self.db = db
def get_face_config_params(self) -> Dict[str, str]:
"""
获取人脸识别配置参数
返回:
配置参数字典 {config_key: config_value}
"""
try:
# 查询全局人脸配置
config_records = self.db.query(SurConfig).filter(
SurConfig.scope == settings.SUR_CONFIG_SCOPE_GLOBAL,
SurConfig.config_type == settings.SUR_CONFIG_TYPE_FACE
).all()
# 查询配置组对应的基础配置
config_group_ids = [record.config_group_id for record in config_records if record.config_group_id]
if config_group_ids:
base_configs = self.db.query(SurConfigBase).filter(
SurConfigBase.group_id.in_(config_group_ids)
).all()
# 合并配置
config_dict = {}
for record in config_records:
if record.config_key and record.config_value:
config_dict[record.config_key] = record.config_value
return config_dict
return {}
except Exception as e:
print(f"获取人脸配置参数失败: {e}")
return {}
def get_face_config_value(self, config_key: str) -> Optional[str]:
"""
获取指定配置键的值
参数:
config_key: 配置键
返回:
配置值如果不存在返回None
"""
try:
config_record = self.db.query(SurConfig).filter(
SurConfig.scope == settings.SUR_CONFIG_SCOPE_GLOBAL,
SurConfig.config_type == settings.SUR_CONFIG_TYPE_FACE,
SurConfig.config_key == config_key
).first()
return config_record.config_value if config_record else None
except Exception as e:
print(f"获取配置值失败 {config_key}: {e}")
return None