video/service/face_service.py

from fastapi import APIRouter, HTTPException, UploadFile, File, Form, Query, Request
from fastapi.responses import FileResponse
from mysql.connector import Error as MySQLError
import os
from pathlib import Path

from ds.db import db
from schema.face_schema import (
    FaceCreateRequest,
    FaceUpdateRequest,
    FaceResponse,
    FaceListResponse
)
from schema.response_schema import APIResponse

from util.face_util import add_binary_data, get_average_feature
from util.file_util import save_face_to_up_images

router = APIRouter(prefix="/faces", tags=["人脸管理"])


# ------------------------------
# 1. 创建人脸记录（使用修复后的路径）
# ------------------------------
@router.post("", response_model=APIResponse, summary="创建人脸记录")
async def create_face(
        request: Request,
        name: str = Form(None, max_length=255, description="名称（可选）"),
        file: UploadFile = File(..., description="人脸文件（必传）")
):
    conn = None
    cursor = None
    try:
        face_create = FaceCreateRequest(name=name)
        client_ip = request.client.host if request.client else ""
        if not client_ip:
            raise HTTPException(status_code=400, detail="无法获取客户端IP")

        conn = db.get_connection()
        cursor = conn.cursor(dictionary=True)

        # 读取图片并保存（使用修复后的路径逻辑）
        file_content = await file.read()
        file_ext = file.filename.split(".")[-1].lower() if "." in file.filename else "jpg"
        save_result = save_face_to_up_images(
            client_ip=client_ip,
            face_name=name,
            image_bytes=file_content,
            image_format=file_ext
        )
        if not save_result["success"]:
            raise HTTPException(status_code=500, detail=f"图片保存失败：{save_result['msg']}")
        db_image_path = save_result["db_path"]  # 从修复后的方法获取路径

        # 提取人脸特征
        detect_success, detect_result = add_binary_data(file_content)
        if not detect_success:
            raise HTTPException(status_code=400, detail=f"人脸检测失败：{detect_result}")
        eigenvalue = detect_result

        # 插入数据库
        insert_query = """
            INSERT INTO face (name, eigenvalue, address)
            VALUES (%s, %s, %s)
        """
        cursor.execute(insert_query, (face_create.name, str(eigenvalue), db_image_path))
        conn.commit()

        # 查询新记录
        cursor.execute("""
            SELECT id, name, address, created_at, updated_at 
            FROM face 
            WHERE id = LAST_INSERT_ID()
        """)
        created_face = cursor.fetchone()
        if not created_face:
            raise HTTPException(status_code=500, detail="创建成功但无法获取记录")

        return APIResponse(
            code=201,
            message=f"人脸记录创建成功（ID: {created_face['id']}）",
            data=FaceResponse(**created_face)
        )
    except MySQLError as e:
        if conn:
            conn.rollback()
        raise HTTPException(status_code=500, detail=f"创建失败: {str(e)}") from e
    except Exception as e:
        raise HTTPException(status_code=500, detail=f"服务器错误: {str(e)}") from e
    finally:
        await file.close()
        db.close_connection(conn, cursor)


# 其他接口（获取单条/列表、更新、删除、获取图片）与之前一致，无需修改
# ------------------------------
# 2. 获取单个人脸记录
# ------------------------------
@router.get("/{face_id}", response_model=APIResponse, summary="获取单个人脸记录")
async def get_face(face_id: int):
    conn = None
    cursor = None
    try:
        conn = db.get_connection()
        cursor = conn.cursor(dictionary=True)

        query = """
            SELECT id, name, address, created_at, updated_at 
            FROM face 
            WHERE id = %s
        """
        cursor.execute(query, (face_id,))
        face = cursor.fetchone()

        if not face:
            raise HTTPException(status_code=404, detail=f"ID为 {face_id} 的记录不存在")

        return APIResponse(
            code=200,
            message="查询成功",
            data=FaceResponse(**face)
        )
    except MySQLError as e:
        raise HTTPException(status_code=500, detail=f"查询失败: {str(e)}") from e
    finally:
        db.close_connection(conn, cursor)


# ------------------------------
# 3. 获取人脸列表
# ------------------------------
@router.get("", response_model=APIResponse, summary="获取人脸列表（分页+筛选）")
async def get_face_list(
        page: int = Query(1, ge=1),
        page_size: int = Query(10, ge=1, le=100),
        name: str = Query(None),
        has_eigenvalue: bool = Query(None)
):
    conn = None
    cursor = None
    try:
        conn = db.get_connection()
        cursor = conn.cursor(dictionary=True)

        where_clause = []
        params = []
        if name:
            where_clause.append("name LIKE %s")
            params.append(f"%{name}%")
        if has_eigenvalue is not None:
            where_clause.append("eigenvalue IS NOT NULL" if has_eigenvalue else "eigenvalue IS NULL")

        # 总记录数
        count_query = "SELECT COUNT(*) AS total FROM face"
        if where_clause:
            count_query += " WHERE " + " AND ".join(where_clause)
        cursor.execute(count_query, params)
        total = cursor.fetchone()["total"]

        # 列表数据
        offset = (page - 1) * page_size
        list_query = """
            SELECT id, name, address, created_at, updated_at 
            FROM face
        """
        if where_clause:
            list_query += " WHERE " + " AND ".join(where_clause)
        list_query += " ORDER BY id DESC LIMIT %s OFFSET %s"
        params.extend([page_size, offset])

        cursor.execute(list_query, params)
        face_list = cursor.fetchall()

        return APIResponse(
            code=200,
            message=f"获取成功（共{total}条）",
            data=FaceListResponse(
                total=total,
                faces=[FaceResponse(**face) for face in face_list]
            )
        )
    except MySQLError as e:
        raise HTTPException(status_code=500, detail=f"查询失败: {str(e)}") from e
    finally:
        db.close_connection(conn, cursor)


# ------------------------------
# 4. 更新人脸记录
# ------------------------------
@router.put("/{face_id}", response_model=APIResponse, summary="更新人脸记录")
async def update_face(face_id: int, face_update: FaceUpdateRequest):
    conn = None
    cursor = None
    try:
        conn = db.get_connection()
        cursor = conn.cursor(dictionary=True)

        cursor.execute("SELECT id, address FROM face WHERE id = %s", (face_id,))
        exist_face = cursor.fetchone()
        if not exist_face:
            raise HTTPException(status_code=404, detail=f"ID为 {face_id} 的记录不存在")
        old_db_path = exist_face["address"]

        update_fields = []
        params = []
        if face_update.name is not None:
            update_fields.append("name = %s")
            params.append(face_update.name)
        if face_update.eigenvalue is not None:
            update_fields.append("eigenvalue = %s")
            params.append(face_update.eigenvalue)
        if face_update.address is not None:
            # 删除旧图片（相对路径转绝对路径）
            if old_db_path:
                old_abs_path = Path(old_db_path).resolve()
                if old_abs_path.exists():
                    try:
                        old_abs_path.unlink()  # 使用Path方法删除更安全
                        print(f"[FaceRouter] 已删除旧图片：{old_abs_path}")
                    except Exception as e:
                        print(f"[FaceRouter] 删除旧图片失败：{str(e)}")
            update_fields.append("address = %s")
            params.append(face_update.address)

        if not update_fields:
            raise HTTPException(status_code=400, detail="至少需提供一个更新字段")

        params.append(face_id)
        update_query = f"UPDATE face SET {', '.join(update_fields)}, updated_at = CURRENT_TIMESTAMP WHERE id = %s"
        cursor.execute(update_query, params)
        conn.commit()

        cursor.execute("""
            SELECT id, name, address, created_at, updated_at 
            FROM face 
            WHERE id = %s
        """, (face_id,))
        updated_face = cursor.fetchone()

        return APIResponse(
            code=200,
            message="更新成功",
            data=FaceResponse(**updated_face)
        )
    except MySQLError as e:
        if conn:
            conn.rollback()
        raise HTTPException(status_code=500, detail=f"更新失败: {str(e)}") from e
    finally:
        db.close_connection(conn, cursor)


# ------------------------------
# 5. 删除人脸记录
# ------------------------------
@router.delete("/{face_id}", response_model=APIResponse, summary="删除人脸记录")
async def delete_face(face_id: int):
    conn = None
    cursor = None
    try:
        conn = db.get_connection()
        cursor = conn.cursor(dictionary=True)

        cursor.execute("SELECT id, address FROM face WHERE id = %s", (face_id,))
        exist_face = cursor.fetchone()
        if not exist_face:
            raise HTTPException(status_code=404, detail=f"ID为 {face_id} 的记录不存在")
        old_db_path = exist_face["address"]

        cursor.execute("DELETE FROM face WHERE id = %s", (face_id,))
        conn.commit()

        # 删除图片
        extra_msg = ""
        if old_db_path:
            old_abs_path = Path(old_db_path).resolve()
            if old_abs_path.exists():
                try:
                    old_abs_path.unlink()
                    print(f"[FaceRouter] 已删除图片：{old_abs_path}")
                    extra_msg = "（已同步删除图片）"
                except Exception as e:
                    print(f"[FaceRouter] 删除图片失败：{str(e)}")
                    extra_msg = "（图片删除失败）"
            else:
                extra_msg = "（图片不存在）"
        else:
            extra_msg = "（无关联图片）"

        return APIResponse(
            code=200,
            message=f"ID为 {face_id} 的记录删除成功 {extra_msg}",
            data=None
        )
    except MySQLError as e:
        if conn:
            conn.rollback()
        raise HTTPException(status_code=500, detail=f"删除失败: {str(e)}") from e
    finally:
        db.close_connection(conn, cursor)


# ------------------------------
# 6. 获取人脸图片
# ------------------------------
@router.get("/{face_id}/image", summary="获取人脸图片")
async def get_face_image(face_id: int):
    conn = None
    cursor = None
    try:
        conn = db.get_connection()
        cursor = conn.cursor(dictionary=True)

        query = "SELECT address, name FROM face WHERE id = %s"
        cursor.execute(query, (face_id,))
        face = cursor.fetchone()

        if not face:
            raise HTTPException(status_code=404, detail=f"ID为 {face_id} 的记录不存在")

        db_path = face["address"]
        abs_path = Path(db_path).resolve()  # 转为绝对路径
        if not db_path or not abs_path.exists():
            raise HTTPException(status_code=404, detail=f"图片不存在（路径：{db_path}）")

        return FileResponse(
            path=abs_path,
            filename=f"face_{face_id}_{face['name'] or '未命名'}.{db_path.split('.')[-1]}",
            media_type=f"image/{db_path.split('.')[-1]}"
        )
    except MySQLError as e:
        raise HTTPException(status_code=500, detail=f"获取图片失败: {str(e)}") from e
    finally:
        db.close_connection(conn, cursor)


# ------------------------------
# 内部工具方法
# ------------------------------
def get_all_face_name_with_eigenvalue() -> dict:
    conn = None
    cursor = None
    try:
        conn = db.get_connection()
        cursor = conn.cursor(dictionary=True)

        query = "SELECT name, eigenvalue FROM face WHERE name IS NOT NULL"
        cursor.execute(query)
        faces = cursor.fetchall()

        name_to_eigenvalues = {}
        for face in faces:
            name = face["name"]
            eigenvalue = face["eigenvalue"]
            if name in name_to_eigenvalues:
                name_to_eigenvalues[name].append(eigenvalue)
            else:
                name_to_eigenvalues[name] = [eigenvalue]

        face_dict = {}
        for name, eigenvalues in name_to_eigenvalues.items():
            if len(eigenvalues) > 1:
                face_dict[name] = get_average_feature(eigenvalues)
            else:
                face_dict[name] = eigenvalues[0]

        return face_dict

    except MySQLError as e:
        raise Exception(f"获取人脸特征失败: {str(e)}") from e
    finally:
        db.close_connection(conn, cursor)