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

时间:2019-01-04 19:02:12

标签: python django django-models

我将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)

1 个答案:

答案 0 :(得分:1)

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

现在,当您从管理员exclude来使用client_id时,该字段将不会出现在表单中,因为client_id是必填字段,因此会引发错误。因此,您可以覆盖{{1 }}的管理方法为:

save_model

因此,每当更改/添加对象时,都会将当前用户分配给client_id。希望有帮助