Upload routers/tickets.py

routers/tickets.py

@@ -0,0 +1,136 @@
+from fastapi import APIRouter, Depends, HTTPException, status
+from sqlalchemy.ext.asyncio import AsyncSession
+from sqlalchemy.future import select
+from typing import List
+from database.session import get_db
+from database.models import Ticket as TicketModel, TicketMessage as TicketMessageModel, User as UserModel
+from auth import get_current_user
+import schemas
+
+router = APIRouter(
+    prefix="/tickets",
+    tags=["Tickets"],
+    responses={404: {"description": "Not found"}},
+)
+
+@router.post("/", response_model=schemas.Ticket, status_code=status.HTTP_201_CREATED)
+async def create_ticket(
+    ticket: schemas.TicketCreate, 
+    db: AsyncSession = Depends(get_db),
+    current_user: UserModel = Depends(get_current_user)
+):
+    db_ticket = TicketModel(
+        user_id=current_user.id,
+        subject=ticket.subject,
+        status="open",
+        priority="normal"
+    )
+    db.add(db_ticket)
+    await db.commit()
+    await db.refresh(db_ticket)
+    
+    # Add initial message
+    db_message = TicketMessageModel(
+        ticket_id=db_ticket.id,
+        user_id=current_user.id,
+        message=ticket.message,
+        is_admin=False
+    )
+    db.add(db_message)
+    await db.commit()
+    
+    return db_ticket
+
+@router.get("/", response_model=List[schemas.Ticket])
+async def read_tickets(
+    skip: int = 0, 
+    limit: int = 100, 
+    db: AsyncSession = Depends(get_db),
+    current_user: UserModel = Depends(get_current_user)
+):
+    # Retrieve tickets for the current user
+    result = await db.execute(
+        select(TicketModel)
+        .where(TicketModel.user_id == current_user.id)
+        .offset(skip)
+        .limit(limit)
+    )
+    tickets = result.scalars().all()
+    return tickets
+
+@router.get("/{ticket_id}", response_model=schemas.Ticket)
+async def read_ticket(
+    ticket_id: int, 
+    db: AsyncSession = Depends(get_db),
+    current_user: UserModel = Depends(get_current_user)
+):
+    result = await db.execute(
+        select(TicketModel)
+        .where(TicketModel.id == ticket_id)
+        .where(TicketModel.user_id == current_user.id)
+    )
+    ticket = result.scalars().first()
+    if ticket is None:
+        raise HTTPException(status_code=404, detail="Ticket not found")
+        
+    return ticket
+
+@router.post("/{ticket_id}/messages", response_model=schemas.TicketMessageCreate, status_code=status.HTTP_201_CREATED)
+async def reply_ticket(
+    ticket_id: int,
+    message: schemas.TicketMessageCreate,
+    db: AsyncSession = Depends(get_db),
+    current_user: UserModel = Depends(get_current_user)
+):
+    # Verify ticket belongs to user
+    result = await db.execute(
+        select(TicketModel)
+        .where(TicketModel.id == ticket_id)
+        .where(TicketModel.user_id == current_user.id)
+    )
+    ticket = result.scalars().first()
+    
+    if ticket is None:
+        raise HTTPException(status_code=404, detail="Ticket not found")
+        
+    if ticket.status == "closed":
+        raise HTTPException(status_code=400, detail="Cannot reply to a closed ticket")
+        
+    # Create message
+    db_message = TicketMessageModel(
+        ticket_id=ticket.id,
+        user_id=current_user.id,
+        message=message.message,
+        is_admin=False
+    )
+    db.add(db_message)
+    
+    # Update ticket status if it was answered
+    if ticket.status == "answered":
+        ticket.status = "open"
+        
+    await db.commit()
+    await db.refresh(db_message)
+    return message
+
+@router.put("/{ticket_id}/close", response_model=schemas.Ticket)
+async def close_ticket(
+    ticket_id: int,
+    db: AsyncSession = Depends(get_db),
+    current_user: UserModel = Depends(get_current_user)
+):
+    result = await db.execute(
+        select(TicketModel)
+        .where(TicketModel.id == ticket_id)
+        .where(TicketModel.user_id == current_user.id)
+    )
+    ticket = result.scalars().first()
+    
+    if ticket is None:
+        raise HTTPException(status_code=404, detail="Ticket not found")
+        
+    ticket.status = "closed"
+    await db.commit()
+    await db.refresh(ticket)
+    
+    return ticket