在 Django Admin 后台中自动设置当前用户为文章作者

在 Django Admin 后台中自动设置当前用户为文章作者，最推荐的方法是重写 ModelAdmin 的 save_model 方法（即方法一）。以下是详细分析和推荐理由：

推荐方法：重写 save_model（方法一）

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

class ArticleAdmin(admin.ModelAdmin):
    def save_model(self, request, obj, form, change):
        if not change:  # 仅对新增对象生效
            obj.author = request.user  # 自动设置当前用户为作者
        super().save_model(request, obj, form, change)

admin.site.register(Article, ArticleAdmin)

推荐理由

1. 专为 Admin 设计

   • 直接集成在 ModelAdmin 中，与 Django Admin 的逻辑紧密结合，无需修改模型或信号。
   • 仅影响通过 Admin 后台创建/编辑的对象，不会干扰其他方式（如 API、表单）的创建逻辑。

2. 简洁高效

   • 代码量少，逻辑清晰，易于维护。
   • 无需处理信号的注册/注销或模型方法的额外调用。

3. 安全可控

   • 通过 if not change 确保仅对新增对象设置作者，避免编辑时意外修改作者。
   • 直接使用 request.user 获取当前用户，安全可靠。

4. 性能最优

   • 在保存时直接赋值，无额外数据库查询或信号开销。

其他方法的对比

方法二：在模型中重写 save 方法

# models.py
class Article(models.Model):
    # ...字段定义...
    def save(self, *args, **kwargs):
        if not self.pk:  # 新增对象
            self.author = kwargs.pop('author', None)  # 需要从外部传入作者
        super().save(*args, **kwargs)

缺点：

• 需要手动在 Admin 的 save_model 中调用 obj.save(author=request.user)，逻辑分散。
• 如果其他地方调用 save() 未传入 author，会导致作者为空。

方法三：使用 pre_save 信号

# signals.py
from django.db.models.signals import pre_save
from django.contrib.auth.models import User

@receiver(pre_save, sender=Article)
def set_author(sender, instance, **kwargs):
    if not instance.pk and not instance.author:  # 需额外处理获取当前用户
        instance.author = get_current_user()  # 实现复杂，可能不可靠

缺点：

• 信号是全局的，难以精确控制仅在 Admin 后台生效。
• 获取当前用户（request.user）在信号中较困难，需借助线程局部变量或中间件，增加复杂性。

最终建议

• 优先选择方法一（重写 save_model），除非有特殊需求（如需要在模型层面强制设置作者）。
• 如果项目已使用信号或需要在多个地方复用逻辑，可考虑方法二或三，但需谨慎处理边界条件。

完整示例（推荐实现）

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

class Article(models.Model):
    title = models.CharField(max_length=200)
    content = models.TextField()
    author = models.ForeignKey(User, on_delete=models.CASCADE)
    created_at = models.DateTimeField(auto_now_add=True)

    def __str__(self):
        return self.title

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

class ArticleAdmin(admin.ModelAdmin):
    list_display = ('title', 'author', 'created_at')
    
    def save_model(self, request, obj, form, change):
        if not change:
            obj.author = request.user  # 自动设置作者
        super().save_model(request, obj, form, change)

admin.site.register(Article, ArticleAdmin)

这样实现后，管理员在后台添加文章时，作者字段会自动填充为当前登录用户，无需手动选择。