326 lines
11 KiB
Python
326 lines
11 KiB
Python
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) |