Django模型用户自动保存为当前用户|极客教程

Django模型用户自动保存为当前用户

在Django中，我们经常会遇到需要将当前登录用户自动保存为某个字段的情况，比如在创建某个对象时，我们希望将当前登录用户保存为对象的创建者。本文将详细介绍如何实现这一功能。

1. 创建一个自定义的模型管理器

首先，我们需要为我们的模型创建一个自定义的模型管理器。在这个自定义的模型管理器中，我们可以重写create方法，在对象创建时自动保存当前登录用户。

from django.db import models

class CustomManager(models.Manager):
    def create(self, **kwargs):
        # 获取当前登录用户
        current_user = self.request.user
        kwargs['created_by'] = current_user
        return super().create(**kwargs)

2. 将自定义的模型管理器应用到模型中

接下来，我们需要将自定义的模型管理器应用到我们的模型中。假设我们有一个名为Post的模型，我们可以这样修改它：

from django.db import models

class Post(models.Model):
    title = models.CharField(max_length=100)
    content = models.TextField()
    created_by = models.ForeignKey('auth.User', on_delete=models.SET_NULL, null=True)

    objects = CustomManager()

在这个修改后的模型中，我们将objects属性指定为我们刚刚创建的CustomManager对象。

3. 设置请求中的当前用户

为了能够获取到当前登录的用户，我们需要在视图中设置请求中的当前用户。可以通过在视图中将当前用户保存为请求对象的一个属性来实现：

from django.db import models
from django.http import HttpRequest

def my_view(request: HttpRequest):
    request.user = request.user
    # 其他视图逻辑

4. 测试功能

最后，我们来测试一下我们的功能是否正常工作。我们可以创建一个视图来创建一个Post对象：

from django.shortcuts import render
from .models import Post

def create_post(request):
    if request.method == 'POST':
        title = request.POST.get('title')
        content = request.POST.get('content')

        post = Post.objects.create(title=title, content=content)

        return render(request, 'post_created.html')

    return render(request, 'create_post.html')

在这个视图中，我们调用Post.objects.create来创建一个Post对象，并且理应自动保存当前登录用户。请确保在请求中正确设置了当前用户。