Django 用戶模型外鍵

Question

我正在努力在 Django 中編寫關系數據庫。 基本上我想要實現的是為其中一個表創建一個用戶外鍵。

我在這里發現了很多類似的問題，但由於某種原因，它似乎對我的事業不起作用，也許是因為版本的原因，或者我不知道。

這是我的模型：

from django.contrib.auth.models import User


class Category(models.Model):
    name = models.CharField(max_length=255)
    user = models.ForeignKey(User, on_delete=models.CASCADE)

看法：

from rest_framework import viewsets
from rest_framework.authentication import SessionAuthentication, BasicAuthentication
from rest_framework.permissions import IsAuthenticated

from .models import Category
from .serializers import CategorySerializer


class CategoryView(viewsets.ModelViewSet):
    queryset = Category.objects.all()
    serializer_class = CategorySerializer
    permission_classes = [IsAuthenticated]
    authentication_classes = [SessionAuthentication, BasicAuthentication]

對於身份驗證，我使用了內置功能

path('api-auth/', include('rest_framework.urls')),

這是我得到的錯誤：

IntegrityError at /category/
NOT NULL constraint failed: category_category.user_id
Request Method: POST
Request URL:    http://127.0.0.1:8000/category/
Django Version: 2.2.5
Exception Type: IntegrityError
Exception Value:    
NOT NULL constraint failed: category_category.user_id
Exception Location: C:\Users\Terkea\Anaconda3\envs\django_api_template\lib\site-packages\django\db\backends\sqlite3\base.py in execute, line 383
Python Executable:  C:\Users\Terkea\Anaconda3\envs\django_api_template\python.exe
Python Version: 3.7.5
Python Path:    
['C:\\Users\\Terkea\\PycharmProjects\\django_api_template',
 'C:\\Users\\Terkea\\PycharmProjects\\django_api_template',
 'C:\\Program Files\\JetBrains\\PyCharm '
 '2019.2.3\\plugins\\python\\helpers\\pycharm_display',
 'C:\\Users\\Terkea\\Anaconda3\\envs\\django_api_template\\python37.zip',
 'C:\\Users\\Terkea\\Anaconda3\\envs\\django_api_template\\DLLs',
 'C:\\Users\\Terkea\\Anaconda3\\envs\\django_api_template\\lib',
 'C:\\Users\\Terkea\\Anaconda3\\envs\\django_api_template',
 'C:\\Users\\Terkea\\Anaconda3\\envs\\django_api_template\\lib\\site-packages',
 'C:\\Program Files\\JetBrains\\PyCharm '
 '2019.2.3\\plugins\\python\\helpers\\pycharm_matplotlib_backend']
Server time:    Wed, 11 Dec 2019 08:08:11 +0000

Answer 1

您可以覆蓋perform_create方法。

class CategoryView(viewsets.ModelViewSet):
    queryset = Category.objects.all()
    serializer_class = CategorySerializer
    permission_classes = [IsAuthenticated]
    authentication_classes = [SessionAuthentication, BasicAuthentication]


    def perform_create(self, serializer):
        serializer.save(user=self.request.user)

從文檔：

perform_create(self, serializer) - 保存新對象實例時由 CreateModelMixin 調用。