HW12_Tymofieieva

hr/api_urls.py

@@ -8,13 +8,14 @@
 from hr.api_views import (
     EmployeeViewSet,
     PositionViewSet,
-    SalaryCalculatorView,
+    SalaryCalculatorView, DepartmentViewSet,
 )
-
+from hr.models import Department
 
 router = DefaultRouter()
 router.register(r'employees', EmployeeViewSet)
 router.register(r'positions', PositionViewSet)
+router.register(r'department', DepartmentViewSet)
 
 urlpatterns = [
     path('', include(router.urls)),

hr/api_views.py

@@ -7,16 +7,28 @@
 from hr.calculate_salary import CalculateMonthRateSalary
 from hr.models import (
     Employee,
-    Position,
+    Position, Department,
 )
 from hr.pydantic_models import WorkingDays
 from hr.serializers import (
     EmployeeSerializer,
     PositionSerializer,
     SalarySerializer,
+    DepartmentSerializer
 )
 
 
+class DepartmentViewSet(viewsets.ModelViewSet):
+    queryset = Department.objects.all()
+    serializer_class = DepartmentSerializer
+
+    @action(detail=True, methods=['get'])
+    def employee_count(self, request, pk=None):
+        department = self.get_object()
+        count_emp = Employee.objects.filter(position__department=department).count()
+        return Response({'employee_count': count_emp})
+
+
 class EmployeeViewSet(viewsets.ModelViewSet):
     queryset = Employee.objects.all()
     serializer_class = EmployeeSerializer

hr/serializers.py

@@ -2,9 +2,14 @@
 
 from hr.models import (
     Employee,
-    Position,
+    Position, Department,
 )
 
+class DepartmentSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = Department
+        fields = ('id', 'name', 'parent_department')
+
 
 class EmployeeSerializer(serializers.ModelSerializer):
     class Meta: