初始化

backend/app/routers/reports.py

@@ -0,0 +1,150 @@
+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)}")