如何使当前用户成为Django模型中的外键

Question

我将Django admin用作客户端的仪表板（受限权限）。 我允许他们通过管理站点添加新产品，并希望将当前用户用作Products模型中的外键。 我在ProductAdmin中使用“排除”，因此客户端将无法选择更改它。 当不使用“排除”时，用户是默认选择，并且可以使用。 当我包含“ exclude”时，表单返回错误，指出该字段为null。 我如何使当前用户成为产品模型中的外键，无论他们是通过管理表单添加项目还是通过csv上传项目？

这是我的产品型号：

#models.py
from django.db import models
from datetime import datetime
from django.contrib.auth.models import User

class Product(models.Model):
    client_id = models.ForeignKey(User, on_delete=models.DO_NOTHING)
    title = models.CharField(max_length=200)
    asin = models.CharField(max_length=200, blank=True)
    sku = models.CharField(max_length=200, blank=True)
    date_added = models.DateTimeField(default=datetime.now, blank=True)
    in_report = models.BooleanField(default=True)

    def get_queryset(self, request):
        qs = super().get_queryset(request)
        if request.user.is_superuser:
            return qs
        return qs.filter(client_id=request.user)

    def __str__(self):
        return self.title

#admin.py
from django.contrib import admin
from .models import Product

class ProductAdmin(admin.ModelAdmin):
    list_display = ('title', 'asin', 'sku', 'in_report')
    list_display_links = ('title', 'asin', 'sku')
    list_editable = ('in_report',)
    search_fields = ('title', 'asin', 'sku', 'in_report')
    list_per_page = 50
    exclude = ('client_id',)

admin.site.register(Product, ProductAdmin)

Answer 1

我相信您想向用户隐藏client_id，但是在通过admin保存/更改对象时，您希望当前用户得到保存。

现在，当您从admin中exclude client_id时，该字段将不会出现在表单中，这将引发错误，因为client_id是必填字段。因此，您可以使用save_model方法覆盖admin的save_model方法：

def save_model(self, request, obj, form, change):
    # associating the current logged in user to the client_id
    obj.client_id = request.user
    super().save_model(request, obj, form, change)

因此，无论何时更改/添加对象，当前用户都将被分配给client_id。希望它会有所帮助