DRF вложенный маршрутизатор сериализатор исходных полей из url
У меня есть модель автора и книги. У автора есть много книг
class Author(Model):
id = UUIDField(primary_key=True, default=uuid4, editable=False)
name = CharField(max_length=50)
email = CharField(max_length=50)
class Book(Model):
id = UUIDField(primary_key=True, default=uuid4, editable=False)
name = CharField(max_length=50)
author = ForeignKey(Author, on_delete=models.CASCADE)
В моем urls.py
author_router = SimpleRouter()
author_router.register(
r"author", AuthorViewSet, basename=author"
)
nested_author_router = NestedSimpleRouter(author_router, r"author", lookup="author")
nested_author_router.register(r"book", BookViewSet)
В моем searlizers.py
class BookSerializer(ModelSerializer):
class Meta:
model = Book
fields = (
"id",
"name",
"author",
)
extra_kwargs = {
"id": {"required": False},
"author": {"required": False},
}
class AuthorSerialzer(ModelSerializer):
class Meta:
model = Author
fields = (
"id",
"name",
"email",
)
extra_kwargs = {
"id": {"required": False},
}
В views.py
class BookViewSet(GenericViewSet):
queryset = Book.objects.all()
serializer_class = BookSerializer
def create(self, request, author_pk):
data = request.data
data["author"] = author_pk
serializer = self.get_serializer(data=data)
serializer.is_valid(raise_exception=True)
serializer.save()
return Response(serializer.data)
Поскольку книги связаны с автором, и я использую вложенные маршрутизаторы, вызов curl будет выглядеть так
curl --location --request POST 'localhost:8000/author/1/book' --data '{"name": "Book Name"}'
В моем BookViewSet
случае я вручную добавляю author_pk в объект данных перед вызовом метода сериализатора is_valid
. Есть ли способ указать источник из маршрута URL или какой-либо лучший способ сделать это?
В этом случае вы можете передать author_pk
в save()
, чтобы автоматически установить id автора вновь созданной книги, как объясняется here
:
def create(self, request, author_pk):
serializer = self.get_serializer(data=request.data)
serializer.is_valid(raise_exception=True)
serializer.save(author_id=author_pk)
return Response(serializer.data)