update
This commit is contained in:
Iliyan Angelov
437
Backend/src/guest_management/routes/complaint_routes.py
Normal file
437
Backend/src/guest_management/routes/complaint_routes.py
Normal file
@@ -0,0 +1,437 @@
|
||||
"""
|
||||
Routes for guest complaint management.
|
||||
"""
|
||||
from fastapi import APIRouter, Depends, HTTPException, Query
|
||||
from sqlalchemy.orm import Session
|
||||
from sqlalchemy import and_, or_, func
|
||||
from typing import Optional
|
||||
from datetime import datetime
|
||||
from ...shared.config.database import get_db
|
||||
from ...shared.config.logging_config import get_logger
|
||||
from ...security.middleware.auth import get_current_user, authorize_roles
|
||||
from ...auth.models.user import User
|
||||
from ..models.guest_complaint import (
|
||||
GuestComplaint, ComplaintStatus, ComplaintPriority, ComplaintCategory, ComplaintUpdate
|
||||
)
|
||||
from ..schemas.complaint import (
|
||||
CreateComplaintRequest, UpdateComplaintRequest,
|
||||
AddComplaintUpdateRequest, ResolveComplaintRequest
|
||||
)
|
||||
from ...shared.utils.response_helpers import success_response
|
||||
|
||||
logger = get_logger(__name__)
|
||||
router = APIRouter(prefix='/complaints', tags=['complaints'])
|
||||
|
||||
|
||||
@router.post('/')
|
||||
async def create_complaint(
|
||||
complaint_data: CreateComplaintRequest,
|
||||
current_user: User = Depends(get_current_user),
|
||||
db: Session = Depends(get_db)
|
||||
):
|
||||
"""Create a new guest complaint."""
|
||||
try:
|
||||
# Verify booking ownership if booking_id provided
|
||||
if complaint_data.booking_id:
|
||||
from ...bookings.models.booking import Booking
|
||||
booking = db.query(Booking).filter(Booking.id == complaint_data.booking_id).first()
|
||||
if not booking:
|
||||
db.rollback()
|
||||
raise HTTPException(status_code=404, detail='Booking not found')
|
||||
|
||||
# Check if user owns the booking (unless admin/staff)
|
||||
from ...shared.utils.role_helpers import is_admin, is_staff
|
||||
if not (is_admin(current_user, db) or is_staff(current_user, db)):
|
||||
if booking.user_id != current_user.id:
|
||||
db.rollback()
|
||||
raise HTTPException(status_code=403, detail='Access denied')
|
||||
|
||||
complaint = GuestComplaint(
|
||||
guest_id=current_user.id,
|
||||
booking_id=complaint_data.booking_id,
|
||||
room_id=complaint_data.room_id,
|
||||
category=ComplaintCategory(complaint_data.category),
|
||||
priority=ComplaintPriority(complaint_data.priority),
|
||||
status=ComplaintStatus.open,
|
||||
title=complaint_data.title,
|
||||
description=complaint_data.description,
|
||||
attachments=complaint_data.attachments or []
|
||||
)
|
||||
|
||||
db.add(complaint)
|
||||
db.flush()
|
||||
|
||||
# Create initial update
|
||||
update = ComplaintUpdate(
|
||||
complaint_id=complaint.id,
|
||||
update_type='status_change',
|
||||
description=f'Complaint created: {complaint_data.title}',
|
||||
updated_by=current_user.id,
|
||||
update_metadata={'status': 'open', 'priority': complaint_data.priority}
|
||||
)
|
||||
db.add(update)
|
||||
|
||||
db.commit()
|
||||
db.refresh(complaint)
|
||||
|
||||
return success_response(
|
||||
data={'complaint': {
|
||||
'id': complaint.id,
|
||||
'title': complaint.title,
|
||||
'status': complaint.status.value,
|
||||
'priority': complaint.priority.value,
|
||||
'category': complaint.category.value,
|
||||
'created_at': complaint.created_at.isoformat()
|
||||
}},
|
||||
message='Complaint created successfully'
|
||||
)
|
||||
except HTTPException:
|
||||
db.rollback()
|
||||
raise
|
||||
except Exception as e:
|
||||
db.rollback()
|
||||
logger.error(f'Error creating complaint: {str(e)}', exc_info=True)
|
||||
raise HTTPException(status_code=500, detail='An error occurred while creating the complaint')
|
||||
|
||||
|
||||
@router.get('/')
|
||||
async def get_complaints(
|
||||
status: Optional[str] = Query(None),
|
||||
priority: Optional[str] = Query(None),
|
||||
category: Optional[str] = Query(None),
|
||||
assigned_to: Optional[int] = Query(None),
|
||||
page: int = Query(1, ge=1),
|
||||
limit: int = Query(20, ge=1, le=100),
|
||||
current_user: User = Depends(get_current_user),
|
||||
db: Session = Depends(get_db)
|
||||
):
|
||||
"""Get complaints with filtering."""
|
||||
try:
|
||||
from ...shared.utils.role_helpers import is_admin, is_staff
|
||||
|
||||
query = db.query(GuestComplaint)
|
||||
|
||||
# Filter by user role
|
||||
try:
|
||||
is_admin_user = is_admin(current_user, db)
|
||||
is_staff_user = is_staff(current_user, db)
|
||||
except Exception as role_error:
|
||||
logger.warning(f'Error checking user role: {str(role_error)}')
|
||||
is_admin_user = False
|
||||
is_staff_user = False
|
||||
|
||||
if not (is_admin_user or is_staff_user):
|
||||
# Customers can only see their own complaints
|
||||
query = query.filter(GuestComplaint.guest_id == current_user.id)
|
||||
elif assigned_to:
|
||||
# Staff/admin can filter by assignee
|
||||
query = query.filter(GuestComplaint.assigned_to == assigned_to)
|
||||
|
||||
# Apply filters
|
||||
if status:
|
||||
try:
|
||||
query = query.filter(GuestComplaint.status == ComplaintStatus(status))
|
||||
except ValueError:
|
||||
logger.warning(f'Invalid status filter: {status}')
|
||||
if priority:
|
||||
try:
|
||||
query = query.filter(GuestComplaint.priority == ComplaintPriority(priority))
|
||||
except ValueError:
|
||||
logger.warning(f'Invalid priority filter: {priority}')
|
||||
if category:
|
||||
try:
|
||||
query = query.filter(GuestComplaint.category == ComplaintCategory(category))
|
||||
except ValueError:
|
||||
logger.warning(f'Invalid category filter: {category}')
|
||||
|
||||
# Pagination
|
||||
total = query.count()
|
||||
offset = (page - 1) * limit
|
||||
complaints = query.order_by(GuestComplaint.created_at.desc()).offset(offset).limit(limit).all()
|
||||
|
||||
complaints_data = []
|
||||
for complaint in complaints:
|
||||
complaints_data.append({
|
||||
'id': complaint.id,
|
||||
'title': complaint.title,
|
||||
'category': complaint.category.value,
|
||||
'priority': complaint.priority.value,
|
||||
'status': complaint.status.value,
|
||||
'guest_id': complaint.guest_id,
|
||||
'booking_id': complaint.booking_id,
|
||||
'room_id': complaint.room_id,
|
||||
'assigned_to': complaint.assigned_to,
|
||||
'created_at': complaint.created_at.isoformat(),
|
||||
'updated_at': complaint.updated_at.isoformat()
|
||||
})
|
||||
|
||||
return success_response(
|
||||
data={
|
||||
'complaints': complaints_data,
|
||||
'pagination': {
|
||||
'page': page,
|
||||
'limit': limit,
|
||||
'total': total,
|
||||
'total_pages': (total + limit - 1) // limit
|
||||
}
|
||||
},
|
||||
message='Complaints retrieved successfully'
|
||||
)
|
||||
except HTTPException:
|
||||
raise
|
||||
except Exception as e:
|
||||
logger.error(f'Error retrieving complaints: {str(e)}', exc_info=True)
|
||||
raise HTTPException(status_code=500, detail=f'An error occurred while retrieving complaints: {str(e)}')
|
||||
|
||||
|
||||
@router.get('/{complaint_id}')
|
||||
async def get_complaint(
|
||||
complaint_id: int,
|
||||
current_user: User = Depends(get_current_user),
|
||||
db: Session = Depends(get_db)
|
||||
):
|
||||
"""Get a specific complaint with details."""
|
||||
try:
|
||||
complaint = db.query(GuestComplaint).filter(GuestComplaint.id == complaint_id).first()
|
||||
if not complaint:
|
||||
raise HTTPException(status_code=404, detail='Complaint not found')
|
||||
|
||||
# Check access
|
||||
from ...shared.utils.role_helpers import is_admin, is_staff
|
||||
if not (is_admin(current_user, db) or is_staff(current_user, db)):
|
||||
if complaint.guest_id != current_user.id:
|
||||
raise HTTPException(status_code=403, detail='Access denied')
|
||||
|
||||
# Get updates
|
||||
updates = db.query(ComplaintUpdate).filter(
|
||||
ComplaintUpdate.complaint_id == complaint_id
|
||||
).order_by(ComplaintUpdate.created_at.asc()).all()
|
||||
|
||||
complaint_data = {
|
||||
'id': complaint.id,
|
||||
'title': complaint.title,
|
||||
'description': complaint.description,
|
||||
'category': complaint.category.value,
|
||||
'priority': complaint.priority.value,
|
||||
'status': complaint.status.value,
|
||||
'guest_id': complaint.guest_id,
|
||||
'booking_id': complaint.booking_id,
|
||||
'room_id': complaint.room_id,
|
||||
'assigned_to': complaint.assigned_to,
|
||||
'resolution': complaint.resolution,
|
||||
'resolved_at': complaint.resolved_at.isoformat() if complaint.resolved_at else None,
|
||||
'resolved_by': complaint.resolved_by,
|
||||
'guest_satisfaction_rating': complaint.guest_satisfaction_rating,
|
||||
'guest_feedback': complaint.guest_feedback,
|
||||
'internal_notes': complaint.internal_notes if (is_admin(current_user, db) or is_staff(current_user, db)) else None,
|
||||
'attachments': complaint.attachments,
|
||||
'requires_follow_up': complaint.requires_follow_up,
|
||||
'follow_up_date': complaint.follow_up_date.isoformat() if complaint.follow_up_date else None,
|
||||
'created_at': complaint.created_at.isoformat(),
|
||||
'updated_at': complaint.updated_at.isoformat(),
|
||||
'updates': [{
|
||||
'id': u.id,
|
||||
'update_type': u.update_type,
|
||||
'description': u.description,
|
||||
'updated_by': u.updated_by,
|
||||
'created_at': u.created_at.isoformat()
|
||||
} for u in updates]
|
||||
}
|
||||
|
||||
return success_response(
|
||||
data={'complaint': complaint_data},
|
||||
message='Complaint retrieved successfully'
|
||||
)
|
||||
except HTTPException:
|
||||
raise
|
||||
except Exception as e:
|
||||
logger.error(f'Error retrieving complaint: {str(e)}', exc_info=True)
|
||||
raise HTTPException(status_code=500, detail='An error occurred while retrieving complaint')
|
||||
|
||||
|
||||
@router.put('/{complaint_id}')
|
||||
async def update_complaint(
|
||||
complaint_id: int,
|
||||
update_data: UpdateComplaintRequest,
|
||||
current_user: User = Depends(authorize_roles('admin', 'staff')),
|
||||
db: Session = Depends(get_db)
|
||||
):
|
||||
"""Update a complaint (admin/staff only)."""
|
||||
try:
|
||||
complaint = db.query(GuestComplaint).filter(GuestComplaint.id == complaint_id).first()
|
||||
if not complaint:
|
||||
db.rollback()
|
||||
raise HTTPException(status_code=404, detail='Complaint not found')
|
||||
|
||||
# Track changes
|
||||
changes = []
|
||||
|
||||
if update_data.status:
|
||||
old_status = complaint.status.value
|
||||
complaint.status = ComplaintStatus(update_data.status)
|
||||
changes.append(f'Status changed from {old_status} to {update_data.status}')
|
||||
|
||||
if update_data.status == 'resolved' and not complaint.resolved_at:
|
||||
complaint.resolved_at = datetime.utcnow()
|
||||
complaint.resolved_by = current_user.id
|
||||
|
||||
if update_data.priority:
|
||||
old_priority = complaint.priority.value
|
||||
complaint.priority = ComplaintPriority(update_data.priority)
|
||||
changes.append(f'Priority changed from {old_priority} to {update_data.priority}')
|
||||
|
||||
if update_data.assigned_to is not None:
|
||||
old_assignee = complaint.assigned_to
|
||||
complaint.assigned_to = update_data.assigned_to
|
||||
changes.append(f'Assigned to user {update_data.assigned_to}')
|
||||
|
||||
if update_data.resolution:
|
||||
complaint.resolution = update_data.resolution
|
||||
|
||||
if update_data.internal_notes is not None:
|
||||
complaint.internal_notes = update_data.internal_notes
|
||||
|
||||
if update_data.requires_follow_up is not None:
|
||||
complaint.requires_follow_up = update_data.requires_follow_up
|
||||
|
||||
if update_data.follow_up_date:
|
||||
complaint.follow_up_date = datetime.fromisoformat(update_data.follow_up_date.replace('Z', '+00:00'))
|
||||
|
||||
# Create update record
|
||||
if changes:
|
||||
update = ComplaintUpdate(
|
||||
complaint_id=complaint.id,
|
||||
update_type='status_change' if update_data.status else 'note',
|
||||
description='; '.join(changes),
|
||||
updated_by=current_user.id
|
||||
)
|
||||
db.add(update)
|
||||
|
||||
db.commit()
|
||||
db.refresh(complaint)
|
||||
|
||||
return success_response(
|
||||
data={'complaint': {
|
||||
'id': complaint.id,
|
||||
'status': complaint.status.value,
|
||||
'priority': complaint.priority.value,
|
||||
'assigned_to': complaint.assigned_to
|
||||
}},
|
||||
message='Complaint updated successfully'
|
||||
)
|
||||
except HTTPException:
|
||||
db.rollback()
|
||||
raise
|
||||
except Exception as e:
|
||||
db.rollback()
|
||||
logger.error(f'Error updating complaint: {str(e)}', exc_info=True)
|
||||
raise HTTPException(status_code=500, detail='An error occurred while updating complaint')
|
||||
|
||||
|
||||
@router.post('/{complaint_id}/resolve')
|
||||
async def resolve_complaint(
|
||||
complaint_id: int,
|
||||
resolve_data: ResolveComplaintRequest,
|
||||
current_user: User = Depends(authorize_roles('admin', 'staff')),
|
||||
db: Session = Depends(get_db)
|
||||
):
|
||||
"""Resolve a complaint."""
|
||||
try:
|
||||
complaint = db.query(GuestComplaint).filter(GuestComplaint.id == complaint_id).first()
|
||||
if not complaint:
|
||||
db.rollback()
|
||||
raise HTTPException(status_code=404, detail='Complaint not found')
|
||||
|
||||
complaint.status = ComplaintStatus.resolved
|
||||
complaint.resolution = resolve_data.resolution
|
||||
complaint.resolved_at = datetime.utcnow()
|
||||
complaint.resolved_by = current_user.id
|
||||
|
||||
if resolve_data.guest_satisfaction_rating:
|
||||
complaint.guest_satisfaction_rating = resolve_data.guest_satisfaction_rating
|
||||
if resolve_data.guest_feedback:
|
||||
complaint.guest_feedback = resolve_data.guest_feedback
|
||||
|
||||
# Create update record
|
||||
update = ComplaintUpdate(
|
||||
complaint_id=complaint.id,
|
||||
update_type='resolution',
|
||||
description=f'Complaint resolved: {resolve_data.resolution}',
|
||||
updated_by=current_user.id,
|
||||
update_metadata={
|
||||
'satisfaction_rating': resolve_data.guest_satisfaction_rating,
|
||||
'guest_feedback': resolve_data.guest_feedback
|
||||
}
|
||||
)
|
||||
db.add(update)
|
||||
|
||||
db.commit()
|
||||
db.refresh(complaint)
|
||||
|
||||
return success_response(
|
||||
data={'complaint': {
|
||||
'id': complaint.id,
|
||||
'status': complaint.status.value,
|
||||
'resolved_at': complaint.resolved_at.isoformat()
|
||||
}},
|
||||
message='Complaint resolved successfully'
|
||||
)
|
||||
except HTTPException:
|
||||
db.rollback()
|
||||
raise
|
||||
except Exception as e:
|
||||
db.rollback()
|
||||
logger.error(f'Error resolving complaint: {str(e)}', exc_info=True)
|
||||
raise HTTPException(status_code=500, detail='An error occurred while resolving complaint')
|
||||
|
||||
|
||||
@router.post('/{complaint_id}/updates')
|
||||
async def add_complaint_update(
|
||||
complaint_id: int,
|
||||
update_data: AddComplaintUpdateRequest,
|
||||
current_user: User = Depends(get_current_user),
|
||||
db: Session = Depends(get_db)
|
||||
):
|
||||
"""Add an update to a complaint."""
|
||||
try:
|
||||
complaint = db.query(GuestComplaint).filter(GuestComplaint.id == complaint_id).first()
|
||||
if not complaint:
|
||||
db.rollback()
|
||||
raise HTTPException(status_code=404, detail='Complaint not found')
|
||||
|
||||
# Check access
|
||||
from ...shared.utils.role_helpers import is_admin, is_staff
|
||||
if not (is_admin(current_user, db) or is_staff(current_user, db)):
|
||||
if complaint.guest_id != current_user.id:
|
||||
db.rollback()
|
||||
raise HTTPException(status_code=403, detail='Access denied')
|
||||
|
||||
update = ComplaintUpdate(
|
||||
complaint_id=complaint_id,
|
||||
update_type=update_data.update_type,
|
||||
description=update_data.description,
|
||||
updated_by=current_user.id,
|
||||
update_metadata=update_data.metadata or {}
|
||||
)
|
||||
|
||||
db.add(update)
|
||||
db.commit()
|
||||
db.refresh(update)
|
||||
|
||||
return success_response(
|
||||
data={'update': {
|
||||
'id': update.id,
|
||||
'update_type': update.update_type,
|
||||
'description': update.description,
|
||||
'created_at': update.created_at.isoformat()
|
||||
}},
|
||||
message='Update added successfully'
|
||||
)
|
||||
except HTTPException:
|
||||
db.rollback()
|
||||
raise
|
||||
except Exception as e:
|
||||
db.rollback()
|
||||
logger.error(f'Error adding complaint update: {str(e)}', exc_info=True)
|
||||
raise HTTPException(status_code=500, detail='An error occurred while adding update')
|
||||
|
||||
Reference in New Issue
Block a user