25 lines
978 B
Python
25 lines
978 B
Python
from rest_framework import viewsets, permissions
|
|
from django_filters.rest_framework import DjangoFilterBackend
|
|
from rest_framework import filters
|
|
from ..models import Admission
|
|
from .serializers import AdmissionSerializer
|
|
|
|
|
|
class AdmissionViewSet(viewsets.ModelViewSet):
|
|
"""ViewSet for admissions"""
|
|
queryset = Admission.objects.all()
|
|
serializer_class = AdmissionSerializer
|
|
permission_classes = [permissions.IsAuthenticated]
|
|
filter_backends = [DjangoFilterBackend, filters.SearchFilter, filters.OrderingFilter]
|
|
filterset_fields = ['status', 'admission_type', 'ward']
|
|
search_fields = ['patient__first_name', 'patient__last_name', 'admission_reason']
|
|
ordering_fields = ['admission_date', 'created_at']
|
|
ordering = ['-admission_date']
|
|
|
|
def get_queryset(self):
|
|
return Admission.objects.filter(tenant=self.request.user.tenant)
|
|
|
|
def perform_create(self, serializer):
|
|
serializer.save(tenant=self.request.user.tenant)
|
|
|