from io import BytesIO from pathlib import Path from fastapi import APIRouter, HTTPException, UploadFile, File, Form, Query, Request from mysql.connector import Error as MySQLError from ds.db import db from encryption.encrypt_decorator import encrypt_response from schema.face_schema import ( FaceCreateRequest, FaceResponse, FaceListResponse ) from schema.response_schema import APIResponse from service.face_service import update_face_data from util.face_util import add_binary_data from service.file_service import save_source_file router = APIRouter(prefix="/api/faces", tags=["人脸管理"]) # 创建人脸记录 @router.post("", response_model=APIResponse, summary="创建人脸记录") @encrypt_response() 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() # 将文件指针重置到开头 await file.seek(0) # 再保存文件 path = save_source_file(file, "face") # 提取人脸特征 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), 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="创建成功但无法获取记录") # TODO 重新加载人脸模型 update_face_data() return APIResponse( code=200, 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) # 获取单个人脸记录 @router.get("/{face_id}", response_model=APIResponse, summary="获取单个人脸记录") @encrypt_response() 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) # 获取人脸列表 @router.get("", response_model=APIResponse, summary="获取人脸列表（分页+筛选）") @encrypt_response() 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) # 删除人脸记录 @router.delete("/{face_id}", response_model=APIResponse, summary="删除人脸记录") @encrypt_response() 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() # 删除图片 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 = "（无关联图片）" # TODO 重新加载人脸模型 update_face_data() 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) @router.post("/batch-import", response_model=APIResponse, summary="批量导入文件夹下的人脸图片") # @encrypt_response() async def batch_import_faces( folder_path: str = Form(..., description="人脸图片所在的**服务器本地文件夹路径**") ): conn = None cursor = None success_count = 0 # 成功导入数量 fail_list = [] # 失败记录（包含文件名、错误原因） try: # 1. 验证文件夹有效性 folder = Path(folder_path) if not folder.exists() or not folder.is_dir(): raise HTTPException(status_code=400, detail=f"文件夹 {folder_path} 不存在或不是有效目录") # 2. 定义支持的图片格式 supported_extensions = {".png", ".jpg", ".jpeg", ".webp"} # 3. 数据库连接初始化 conn = db.get_connection() cursor = conn.cursor(dictionary=True) # 4. 遍历文件夹内所有文件 for file_path in folder.iterdir(): if file_path.is_file() and file_path.suffix.lower() in supported_extensions: file_name = file_path.stem # 提取文件名（不含后缀）作为 `name` try: # 4.1 读取文件二进制内容 with open(file_path, "rb") as f: file_content = f.read() # 4.2 构造模拟的 UploadFile 对象（用于兼容 `save_source_file`） mock_file = UploadFile( filename=file_path.name, file=BytesIO(file_content) ) # 4.3 保存文件到指定目录 saved_path = save_source_file(mock_file, "face") # 4.4 提取人脸特征 detect_success, detect_result = add_binary_data(file_content) if not detect_success: fail_list.append({ "name": file_name, "file_path": str(file_path), "error": f"人脸检测失败：{detect_result}" }) continue # 跳过当前文件，处理下一个 eigenvalue = detect_result # 4.5 插入数据库 insert_sql = """ INSERT INTO face (name, eigenvalue, address) VALUES (%s, %s, %s) """ cursor.execute(insert_sql, (file_name, str(eigenvalue), saved_path)) conn.commit() # 提交当前文件的插入操作 success_count += 1 except Exception as e: # 捕获单文件处理的异常，记录后继续处理其他文件 fail_list.append({ "name": file_name, "file_path": str(file_path), "error": str(e) }) if conn: conn.rollback() # 回滚当前失败文件的插入 # 5. 重新加载人脸模型（确保新增数据生效） update_face_data() # 6. 构造返回结果 return APIResponse( code=200, message=f"批量导入完成，成功 {success_count} 条，失败 {len(fail_list)} 条", data={ "success_count": success_count, "fail_details": fail_list } ) except MySQLError as e: if conn: conn.rollback() raise HTTPException(status_code=500, detail=f"数据库操作失败: {str(e)}") from e except HTTPException: raise # 直接抛出400等由业务逻辑触发的HTTP异常 except Exception as e: raise HTTPException(status_code=500, detail=f"服务器内部错误: {str(e)}") from e finally: db.close_connection(conn, cursor)