File size: 3,305 Bytes
74de430
 
34b4d1d
 
 
 
 
 
74de430
34b4d1d
 
 
 
 
 
 
 
 
 
 
 
 
 
 
f10602a
74de430
 
 
34b4d1d
f10602a
93e3d85
34b4d1d
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
f10602a
93e3d85
34b4d1d
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
"""
REPORTS Endpoints

Provides structured data generation and CSV exports for:
- SLA Compliance
- User Performance
- Financial Summaries
- Inventory Usage
"""
from fastapi import APIRouter, Depends, HTTPException, status
from fastapi.responses import StreamingResponse
from sqlalchemy.orm import Session
from datetime import datetime
from typing import Any

from app.api.deps import get_db, get_current_user
from app.models.user import User
from app.schemas.report import (
    ReportFilter, 
    ReportResponse, 
    ReportMetaData, 
    ReportFormat
)
from app.services.report_service import ReportService
from app.core.permissions import require_permission

router = APIRouter()

@router.post("/generate", response_model=ReportResponse)
@require_permission("view_reports")
async def generate_report(
    filters: ReportFilter,
    db: Session = Depends(get_db),
    current_user: User = Depends(get_current_user)
) -> Any:
    """
    Generate a JSON report for visualization/tables.
    
    **Report Types:**
    - `sla_compliance`: Tickets vs Due Dates
    - `user_performance`: Aggregated timesheet stats
    - `financial_summary`: Revenue & Expenses ledger
    - `inventory_usage`: Items installed/consumed
    
    **Performance:**
    - Designed for sub-2 second response times.
    - Uses aggregated warehouse tables (Timesheets) where possible.
    """
    try:
        data = ReportService.generate_report_data(db, filters, current_user)
        
        return ReportResponse(
            meta=ReportMetaData(
                generated_at=datetime.utcnow(),
                record_count=len(data),
                filters_applied=filters.model_dump(),
                report_type=filters.report_type
            ),
            data=data
        )
    except Exception as e:
        raise HTTPException(
            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
            detail=f"Report generation failed: {str(e)}"
        )


@router.post("/export")
@require_permission("export_reports")
async def export_report(
    filters: ReportFilter,
    format: ReportFormat = ReportFormat.CSV,
    db: Session = Depends(get_db),
    current_user: User = Depends(get_current_user)
):
    """
    Generate and download a report file (CSV).
    
    Designed to work with the same filters as the `/generate` endpoint.
    Returns a File Download stream.
    """
    if format != ReportFormat.CSV:
        raise HTTPException(
            status_code=status.HTTP_400_BAD_REQUEST,
            detail="Only CSV export is currently supported."
        )

    try:
        # 1. Generate Data
        data = ReportService.generate_report_data(db, filters, current_user)
        
        # 2. Convert to CSV Stream
        csv_file = ReportService.generate_csv_export(data)
        
        # 3. Create Filename
        timestamp = datetime.now().strftime("%Y%m%d_%H%M")
        filename = f"{filters.report_type}_{timestamp}.csv"
        
        return StreamingResponse(
            csv_file,
            media_type="text/csv",
            headers={"Content-Disposition": f"attachment; filename={filename}"}
        )

    except Exception as e:
        raise HTTPException(
            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
            detail=f"Export failed: {str(e)}"
        )