CODEWITHSUNDEEP
djangodjango-modelsdjango-formsdjango-views
Thameem
Thameem

Reputation: 3636

django override model save() method

I want to get request as a parameter in save() method.

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()

Instead of the value 2 i want to pass request.user. How can i do it. If question is not correct somebody please correct the question.

Upvotes: 2

Views: 5160

Answers (3)

ruddra
ruddra

Reputation: 51968

One way is to use crequest, like this:

# 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)

Upvotes: 5

Daniel Roseman
Daniel Roseman

Reputation: 599490

Don't do that in the model. Do it in the view.

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

Upvotes: 7

juree
juree

Reputation: 273

You can use kwargs. For example:

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

In you case, pass request as keyword argument, and then use it in models save() method.

Upvotes: 1

Related Questions