django 覆盖模型 save() 方法

Question

我想获取 request 作为 save() 方法中的参数。

models.py

class Profile(models.Model):
    uuid = UUIDField(auto=True)
    user = models.ForeignKey(User)
    name = models.CharField(max_length=128)
    dob = models.DateField()

    class Meta:
        db_table = 'profile'

    def save(self,*args,**kwargs):
        if not self.pk:
            self.user_id = 2  #here i need request.user instead of 2
        super(Profile,self).save(*args,**kwargs)

forms.py

class ProfileForm(forms.ModelForm):

    class Meta:
        model = Profile
        exclude = ['uuid','user']

views.py

def add(request):
    profile_form = ProfileForm(request.POST)
    profile_form.save()

我想传递 request.user 而不是值 2。我该怎么做。如果问题不正确，请有人更正问题。

Answer 1

您可以使用 kwargs。例如：

views.py

profile_form.save(your_arg = request)

models.py

your_var = kwargs.pop("your_arg", None) #Second argument of kwargs.pop is default value

在你的情况下，将请求作为关键字参数传递，然后在模型 save() 方法中使用它。

Answer 2

不要在模型中这样做。在视图中执行。

profile_form = ProfileForm(request.POST)
if profile_form.is_valid():
    profile = profile_form.save(commit=False)
    profile.user = request.user
    profile.save()

Answer 3

一种方法是使用 crequest，像这样：

# model
from crequest.middleware import CrequestMiddleware

class Profile(models.Model):
    # Some fields
    def save(self, *args, **kwargs):
        crequest = CrequestMiddleware.get_request() #its the current request
        if crequest.user.pk is 1:
            # Do stuff
           pass
    super(Profile,self).save(*args,**kwargs)

django 覆盖模型 save() 方法

django override model save() method

django

django-models

django-forms

django-views