45 lines
1.8 KiB
Python
45 lines
1.8 KiB
Python
from rest_framework import viewsets, permissions
|
|
from rest_framework.decorators import action
|
|
from rest_framework.response import Response
|
|
from django_filters.rest_framework import DjangoFilterBackend
|
|
from rest_framework import filters
|
|
from ..models import AppointmentRequest
|
|
from .serializers import AppointmentRequestListSerializer
|
|
|
|
|
|
class AppointmentRequestViewSet(viewsets.ModelViewSet):
|
|
"""ViewSet for appointment requests"""
|
|
queryset = AppointmentRequest.objects.all()
|
|
serializer_class = AppointmentRequestListSerializer
|
|
permission_classes = [permissions.IsAuthenticated]
|
|
filter_backends = [DjangoFilterBackend, filters.SearchFilter, filters.OrderingFilter]
|
|
filterset_fields = ['status', 'priority', 'appointment_type', 'provider']
|
|
search_fields = ['patient__first_name', 'patient__last_name', 'chief_complaint']
|
|
ordering_fields = ['scheduled_datetime', 'preferred_date', 'created_at']
|
|
ordering = ['-created_at']
|
|
|
|
def get_queryset(self):
|
|
"""Filter by tenant"""
|
|
return AppointmentRequest.objects.filter(tenant=self.request.user.tenant)
|
|
|
|
def perform_create(self, serializer):
|
|
"""Set tenant when creating"""
|
|
serializer.save(tenant=self.request.user.tenant)
|
|
|
|
@action(detail=True, methods=['post'])
|
|
def approve(self, request, pk=None):
|
|
"""Approve an appointment request"""
|
|
appointment = self.get_object()
|
|
appointment.status = 'CONFIRMED'
|
|
appointment.save()
|
|
return Response({'status': 'approved'})
|
|
|
|
@action(detail=True, methods=['post'])
|
|
def reject(self, request, pk=None):
|
|
"""Reject an appointment request"""
|
|
appointment = self.get_object()
|
|
appointment.status = 'CANCELLED'
|
|
appointment.save()
|
|
return Response({'status': 'rejected'})
|
|
|