employees api
This commit is contained in:
12
employees/filters.py
Normal file
12
employees/filters.py
Normal file
@@ -0,0 +1,12 @@
|
||||
import django_filters
|
||||
from .models import Employee
|
||||
|
||||
class EmployeeFilter(django_filters.FilterSet):
|
||||
first_name = django_filters.CharFilter(field_name="account__first_name", lookup_expr="icontains")
|
||||
last_name = django_filters.CharFilter(field_name="account__last_name", lookup_expr="icontains")
|
||||
department_name = django_filters.CharFilter(field_name="department__name", lookup_expr="icontains")
|
||||
position_name = django_filters.CharFilter(field_name="position__name", lookup_expr="icontains")
|
||||
|
||||
class Meta:
|
||||
model = Employee
|
||||
fields = ["first_name", "last_name", "department_name", "position_name"]
|
||||
44
employees/serializers.py
Normal file
44
employees/serializers.py
Normal file
@@ -0,0 +1,44 @@
|
||||
from rest_framework import serializers
|
||||
from .models import Position, Department, Employee
|
||||
|
||||
class PositionSerializer(serializers.ModelSerializer):
|
||||
class Meta:
|
||||
model = Position
|
||||
fields = "__all__"
|
||||
|
||||
|
||||
class DepartmentSerializer(serializers.ModelSerializer):
|
||||
class Meta:
|
||||
model = Department
|
||||
fields = "__all__"
|
||||
|
||||
|
||||
class EmployeeSerializer(serializers.ModelSerializer):
|
||||
first_name = serializers.CharField(source="account.first_name")
|
||||
last_name = serializers.CharField(source="account.last_name")
|
||||
middle_name = serializers.CharField(source="account.middle_name", allow_null=True)
|
||||
phone = serializers.CharField(source="account.phone", allow_null=True)
|
||||
email = serializers.CharField(source="account.email")
|
||||
|
||||
class Meta:
|
||||
model = Employee
|
||||
fields = [
|
||||
"id",
|
||||
"department",
|
||||
"position",
|
||||
"first_name",
|
||||
"last_name",
|
||||
"middle_name",
|
||||
"phone",
|
||||
"email",
|
||||
]
|
||||
|
||||
def update(self, instance, validated_data):
|
||||
account_data = validated_data.pop("account", {})
|
||||
instance.account.first_name = account_data.get("first_name", instance.account.first_name)
|
||||
instance.account.last_name = account_data.get("last_name", instance.account.last_name)
|
||||
instance.account.middle_name = account_data.get("middle_name", instance.account.middle_name)
|
||||
instance.account.phone = account_data.get("phone", instance.account.phone)
|
||||
instance.account.email = account_data.get("email", instance.account.email)
|
||||
instance.account.save()
|
||||
return super().update(instance, validated_data)
|
||||
12
employees/urls.py
Normal file
12
employees/urls.py
Normal file
@@ -0,0 +1,12 @@
|
||||
from django.urls import path, include
|
||||
from rest_framework.routers import DefaultRouter
|
||||
from .views import PositionViewSet, DepartmentViewSet, EmployeeViewSet
|
||||
|
||||
router = DefaultRouter()
|
||||
router.register(r"positions", PositionViewSet, basename="position")
|
||||
router.register(r"departments", DepartmentViewSet, basename="department")
|
||||
router.register(r"employees", EmployeeViewSet, basename="employee")
|
||||
|
||||
urlpatterns = [
|
||||
path("", include(router.urls)),
|
||||
]
|
||||
@@ -1,3 +1,30 @@
|
||||
from django.shortcuts import render
|
||||
from rest_framework import viewsets
|
||||
from rest_framework.filters import SearchFilter, OrderingFilter
|
||||
from django_filters.rest_framework import DjangoFilterBackend
|
||||
from .models import Position, Department, Employee
|
||||
from .serializers import PositionSerializer, DepartmentSerializer, EmployeeSerializer
|
||||
from .filters import EmployeeFilter # Фильтры будут добавлены позже
|
||||
|
||||
# Create your views here.
|
||||
class PositionViewSet(viewsets.ModelViewSet):
|
||||
queryset = Position.objects.all()
|
||||
serializer_class = PositionSerializer
|
||||
filter_backends = [SearchFilter, OrderingFilter]
|
||||
search_fields = ["name"]
|
||||
ordering_fields = ["name"]
|
||||
|
||||
|
||||
class DepartmentViewSet(viewsets.ModelViewSet):
|
||||
queryset = Department.objects.all()
|
||||
serializer_class = DepartmentSerializer
|
||||
filter_backends = [SearchFilter, OrderingFilter]
|
||||
search_fields = ["name"]
|
||||
ordering_fields = ["name"]
|
||||
|
||||
|
||||
class EmployeeViewSet(viewsets.ModelViewSet):
|
||||
queryset = Employee.objects.all()
|
||||
serializer_class = EmployeeSerializer
|
||||
filter_backends = [DjangoFilterBackend, SearchFilter, OrderingFilter]
|
||||
filterset_class = EmployeeFilter
|
||||
search_fields = ["account__first_name", "account__last_name", "position__name", "department__name"]
|
||||
ordering_fields = ["department", "position", "account__last_name"]
|
||||
Reference in New Issue
Block a user