После переопределения функции create, как получить id только что созданного объекта в django?
У меня есть ModelViewSet, и я хочу выполнить еще несколько операций над вновь созданным объектом. Но я не могу получить идентификатор вновь созданного объекта. Как получить идентификатор только что созданного объекта после super()?
def create(self, request, *args, **kwargs):
phone = request.data.get('student_phone_number')
batch_id = request.data.get('batch_id')
class_number = request.data.get('class_number')
if not any([phone, batch_id, class_number]):
return Response({"message": "Missing Params"}, status=status.HTTP_400_BAD_REQUEST)
if self.queryset.filter(student_phone_number=phone, batch_id=batch_id, class_number=class_number,
is_cancelled=False).exists():
return Response({"message": "Compensation class already booked"}, status=status.HTTP_400_BAD_REQUEST)
super().create(request, *args, **kwargs)
вы должны переписать метод perform_create не create на ModelViewSet
def perform_create(self, serializer):
phone_value = self.request.data['phone']
batch_id_value = self.request.data['batch_id']
class_number_value = self.request.data['class_number']
// do your condition here
instance = serializer.save(phone=phone_value, batch_id=batch_id_value, class_number=class_number_value)
instance_id = instance.id
// do whatever you want with instance_id
return Response({"message": "Data Has been saved", Id:instance_id }, status=status.HTTP_201_CREATED)
Вызов serializer.save() должен вернуть экземпляр, который только что был создан.