Как добавить дополнительное поле в Django Rest Framework для модели, которая является внешним ключом для другой модели

Я хочу перечислить всех преподавателей определенной кафедры в Django rest framework и не знаю, как это сделать. Я полагаю, что это можно сделать, добавив поле в сериализатор Department, но я не уверен. Есть ли простой способ, как это можно сделать.

models.py

   class Department(models.Model):
       name = models.CharField(max_length=300)
    
        def __str__(self):
            return self.name
    
    
    class Teacher(models.Model):
        name = models.CharField(max_length=300)
        department = models.ForeignKey(Department, on_delete=models.CASCADE)
        tenure = models.BooleanField()

serializers.py

class DepartmentSerializer(serializers.HyperlinkedModelSerializer):
    #What I believe the serializer field would look like to list teachers connected 
    #with a department
    teacher = serializers.DjangoModelField(
        Teacher.objects.filter(department=self.department))

    class Meta:
        model = Department
        fields = ['url', 'name', 'teacher']

urls.py

router = DefaultRouter()
router.register(r'teachers', TeacherViewSet)
router.register(r'departments', DepartmentViewSet)
router.register(r'users', UserViewSet)

urlpatterns = [
    path('api/', include(router.urls)),
]

views.py

class TeacherViewSet(viewsets.ModelViewSet):
    queryset = Teacher.objects.all()
    serializer_class = TeacherSerializer
    permission_classes = [permissions.IsAuthenticatedOrReadOnly]


class DepartmentViewSet(viewsets.ModelViewSet):
    queryset = Department.objects.all()
    serializer_class = DepartmentSerializer
    permission_classes = [permissions.IsAuthenticatedOrReadOnly]

Вы можете получить доступ к Teacher, связанным с department с помощью mydepartment.teacher_set.all(), мы можем использовать это в сериализаторе, например:

class TeacherSerializer(serializers.HyperlinkedModelSerializer):
    class Meta:
        model = Teacher
        fields = ['name', 'tenure']

class DepartmentSerializer(serializers.HyperlinkedModelSerializer):
    teacher_set = serializers.TeacherSerializer(many=True)

    class Meta:
        model = Department
        fields = ['url', 'name', 'teacher_set']
Вернуться на верх