151 lines
5.4 KiB
Python
151 lines
5.4 KiB
Python
from typing import Optional
|
||
from fastapi import APIRouter, Depends, HTTPException, Query, status
|
||
from sqlalchemy.orm import Session
|
||
from fastapi.responses import StreamingResponse
|
||
|
||
from app.database import get_db
|
||
from app.models import User
|
||
from app.routers.auth import get_current_user, get_current_admin
|
||
from app.services.report_service import (
|
||
export_employee_report,
|
||
export_company_report,
|
||
export_performance_report
|
||
)
|
||
|
||
router = APIRouter(prefix="/reports", tags=["报表导出"])
|
||
|
||
|
||
@router.get("/employee/{employee_id}/excel")
|
||
def export_employee_excel(
|
||
employee_id: int,
|
||
period: str = Query(..., description="计算周期: monthly/quarterly/half_yearly/yearly"),
|
||
year: int = Query(..., description="年份"),
|
||
month: Optional[int] = Query(None, description="月份（月度周期需要）"),
|
||
db: Session = Depends(get_db),
|
||
current_user: User = Depends(get_current_user)
|
||
):
|
||
"""导出员工收益Excel报表"""
|
||
# 权限检查：非管理员只能导出自己的报表
|
||
if current_user.role != "admin":
|
||
from app.models import Employee
|
||
employee = db.query(Employee).filter(Employee.user_id == current_user.id).first()
|
||
if not employee or employee.id != employee_id:
|
||
raise HTTPException(
|
||
status_code=status.HTTP_403_FORBIDDEN,
|
||
detail="权限不足，只能导出自己的报表"
|
||
)
|
||
|
||
try:
|
||
excel_bytes, filename = export_employee_report(
|
||
db=db,
|
||
employee_id=employee_id,
|
||
period=period,
|
||
year=year,
|
||
month=month
|
||
)
|
||
|
||
return StreamingResponse(
|
||
iter([excel_bytes]),
|
||
media_type="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet",
|
||
headers={
|
||
"Content-Disposition": f"attachment; filename={filename}"
|
||
}
|
||
)
|
||
except ValueError as e:
|
||
raise HTTPException(status_code=400, detail=str(e))
|
||
except Exception as e:
|
||
raise HTTPException(status_code=500, detail=f"导出失败: {str(e)}")
|
||
|
||
|
||
@router.get("/company/excel")
|
||
def export_company_excel(
|
||
period: str = Query(..., description="计算周期: monthly/quarterly/half_yearly/yearly"),
|
||
year: int = Query(..., description="年份"),
|
||
month: Optional[int] = Query(None, description="月份（月度周期需要）"),
|
||
db: Session = Depends(get_db),
|
||
current_admin: User = Depends(get_current_admin)
|
||
):
|
||
"""导出公司收益Excel报表（仅管理员）"""
|
||
try:
|
||
excel_bytes, filename = export_company_report(
|
||
db=db,
|
||
period=period,
|
||
year=year,
|
||
month=month
|
||
)
|
||
|
||
return StreamingResponse(
|
||
iter([excel_bytes]),
|
||
media_type="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet",
|
||
headers={
|
||
"Content-Disposition": f"attachment; filename={filename}"
|
||
}
|
||
)
|
||
except ValueError as e:
|
||
raise HTTPException(status_code=400, detail=str(e))
|
||
except Exception as e:
|
||
raise HTTPException(status_code=500, detail=f"导出失败: {str(e)}")
|
||
|
||
|
||
@router.get("/performance/excel")
|
||
def export_performance_excel(
|
||
start_date: Optional[str] = Query(None, description="开始日期 (YYYY-MM-DD)"),
|
||
end_date: Optional[str] = Query(None, description="结束日期 (YYYY-MM-DD)"),
|
||
employee_id: Optional[int] = Query(None, description="员工ID"),
|
||
agent_id: Optional[int] = Query(None, description="代理ID"),
|
||
category_id: Optional[int] = Query(None, description="分类ID"),
|
||
record_type: Optional[str] = Query(None, description="记录类型: employee/agent"),
|
||
db: Session = Depends(get_db),
|
||
current_user: User = Depends(get_current_user)
|
||
):
|
||
"""导出业绩报表Excel"""
|
||
from datetime import datetime
|
||
|
||
# 构建筛选条件
|
||
filters = {}
|
||
|
||
if start_date:
|
||
try:
|
||
filters['start_date'] = datetime.strptime(start_date, "%Y-%m-%d").date()
|
||
except ValueError:
|
||
raise HTTPException(status_code=400, detail="开始日期格式错误，应为YYYY-MM-DD")
|
||
|
||
if end_date:
|
||
try:
|
||
filters['end_date'] = datetime.strptime(end_date, "%Y-%m-%d").date()
|
||
except ValueError:
|
||
raise HTTPException(status_code=400, detail="结束日期格式错误，应为YYYY-MM-DD")
|
||
|
||
# 权限检查：非管理员只能查看自己的业绩
|
||
if current_user.role != "admin":
|
||
from app.models import Employee
|
||
employee = db.query(Employee).filter(Employee.user_id == current_user.id).first()
|
||
if employee:
|
||
filters['employee_id'] = employee.id
|
||
else:
|
||
if employee_id:
|
||
filters['employee_id'] = employee_id
|
||
|
||
if agent_id:
|
||
filters['agent_id'] = agent_id
|
||
if category_id:
|
||
filters['category_id'] = category_id
|
||
if record_type:
|
||
filters['record_type'] = record_type
|
||
|
||
try:
|
||
excel_bytes, filename = export_performance_report(
|
||
db=db,
|
||
filters=filters
|
||
)
|
||
|
||
return StreamingResponse(
|
||
iter([excel_bytes]),
|
||
media_type="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet",
|
||
headers={
|
||
"Content-Disposition": f"attachment; filename={filename}"
|
||
}
|
||
)
|
||
except Exception as e:
|
||
raise HTTPException(status_code=500, detail=f"导出失败: {str(e)}")
|