"""
人员相关数据访问层
"""

from typing import List, Dict, Optional
from sqlalchemy.orm import Session
from models.sur_person import SurPersonBlacklist, SurFaceFeature
from config import settings


class SurPersonRepository:
    """人员数据访问类"""
    
    def __init__(self, db: Session):
        self.db = db
    
    def get_blacklist_face_features(self) -> Dict[int, str]:
        """
        获取黑名单人员的人脸特征
        
        返回:
            字典 {person_id: feature_data}
        """
        try:
            # 查询启用的黑名单人员
            blacklist_persons = self.db.query(SurPersonBlacklist).filter(
                SurPersonBlacklist.status == 1
            ).all()
            
            if not blacklist_persons:
                return {}
            
            person_ids = [person.person_id for person in blacklist_persons]
            
            # 查询对应的人脸特征
            face_features = self.db.query(SurFaceFeature).filter(
                SurFaceFeature.person_id.in_(person_ids),
                SurFaceFeature.feature_type == settings.FACE_MODEL_VERSION,
                SurFaceFeature.status == 2  # 计算成功的特征
            ).all()
            
            # 构建特征字典
            feature_dict = {}
            for feature in face_features:
                if feature.feature_data:
                    feature_dict[feature.person_id] = feature.feature_data
            
            return feature_dict
            
        except Exception as e:
            print(f"获取黑名单人脸特征失败: {e}")
            return {}
    
    def get_blacklist_person_count(self) -> int:
        """获取黑名单人员数量"""
        try:
            count = self.db.query(SurPersonBlacklist).filter(
                SurPersonBlacklist.status == 1
            ).count()
            return count
        except Exception as e:
            print(f"获取黑名单人员数量失败: {e}")
            return 0
    
    def get_blacklist_face_feature_count(self) -> int:
        """获取有特征数据的黑名单人员数量"""
        try:
            # 查询启用的黑名单人员
            blacklist_persons = self.db.query(SurPersonBlacklist).filter(
                SurPersonBlacklist.status == 1
            ).all()
            
            if not blacklist_persons:
                return 0
            
            person_ids = [person.person_id for person in blacklist_persons]
            
            # 查询对应的人脸特征数量
            count = self.db.query(SurFaceFeature).filter(
                SurFaceFeature.person_id.in_(person_ids),
                SurFaceFeature.feature_type == settings.FACE_MODEL_VERSION,
                SurFaceFeature.status == 2  # 计算成功的特征
            ).count()
            
            return count
            
        except Exception as e:
            print(f"获取黑名单人脸特征数量失败: {e}")
            return 0