使模型中的用户默认为当前用户

Question

我有以下问题：

from django.contrib.auth.models import User

class ClientDetails(models.Model):
    created_by = models.ForeignKey(User)
    ...

如何将 created_by 默认为当前登录的用户？

（我想这样做，主要是因为我可以将其隐藏在管理视图中，但也因为当我保存实例时，我不想每次都填充它）

Answer 1

由于您需要从 request 对象获取当前登录的用户，因此您无法在模型的 save 方法中获取它，但您可以覆盖模型管理员的 save_model-方法：

class MyAdmin(admin.ModelAdmin):
    def save_model(self, request, instance, form, change):
        user = request.user 
        instance = form.save(commit=False)
        if not change or not instance.created_by:
            instance.created_by = user
        instance.modified_by = user
        instance.save()
        form.save_m2m()
        return instance

Answer 2

您必须重写 admin.py 中模型 Admin 类中的 get_changeform_initial_data 方法，如下所示：

# admin.py

class ClientDetailsAdmin(admin.ModelAdmin):
    def get_changeform_initial_data(self, request):
        get_data = super(ClientDetailsAdmin, self).get_changeform_initial_data(request)
        get_data['created_by'] = request.user.pk
        return get_data

admin.site.register(ClientDetails, ClientDetailsAdmin)

通过这种方式，您可以获得自 created_by 以来最优雅的解决方案> 当您创建新记录时，字段会被归档。

Answer 3

我最近遇到了类似的问题，这是来自我的views.py文件

def CircleAdd(request):

    form = CircleAddForm(request.POST)

    if form.is_valid():

        Circle = form.save(commit=False)
        Circle.Author = request.user
        Circle = Circle.save()

然后我有一个“圆圈”模型的表单，它实际上只是一个包装器（forms.py）

class CircleAddForm(ModelForm):
    class Meta:
        model = Circle

记住在您的视图中导入表单！

编辑：甚至不确定您是否需要单独的表单，关键位是假提交，其次是真正的提交

Answer 4

基于已接受的答案，如果您希望使用基于类的视图执行此操作，您可以按照 docs，覆盖类视图上的 form_valid() 方法：

# views.py

from django.views.generic.edit import CreateView

class CreateClient(CreateView):
    def form_valid(self, form):
        form.instance.created_by = self.request.user
        return super(CreateClient, self).form_valid(form)

Answer 5

普通模型字段有一个默认参数。但据我所知，ForeignKeys 没有，所以我想你需要使用 post_save 信号。

Answer 6

我发现：

from django.db import models
from django.contrib.auth.models import User

class CurrentUserField(models.ForeignKey):
    def __init__(self, **kwargs):
        super(CurrentUserField, self).__init__(User, null=True, **kwargs)

    def contribute_to_class(self, cls, name):
        super(CurrentUserField, self).contribute_to_class(cls, name)
        registry = registration.FieldRegistry()
        registry.add_field(cls, self)

class ClientDetails(models.Model):
    created_by = CurrentUserField()
...

来自 这里。但有没有更简单的方法呢？