CODEWITHSUNDEEP
djangodjango-modelsdjango-usersdjango-request
Theophile
Theophile

Reputation: 300

get current user for 'created_by' field in model class

I'm currently working on a Django app, and I'm trying to set the current user as default on a model, but it doesn't work. 

created_by = models.ForeignKey(User, default=request.user, null=True, blank=True, on_delete=models.DO_NOTHING, related_name='created_by')

I tried to override the save() method but it doesn't work either, anyone has any experience on this matter ?

Thanks a lot in advance for your help

Upvotes: 5

Views: 12807

Answers (3)

Samar Jyoti Nath
Samar Jyoti Nath

Reputation: 96

If your model is like below model.py

class Post(models.Model):
   title = models.CharField(max_length=500)
   created_by = models.ForeignKey(User, on_delete=models.SET_NULL, null= True)

In model admin class in admin.py add the following method

class PostAdmin(admin.ModelAdmin):
    readonly_fields = ('created_by',)
    list_display = ('title', 'created_by')

    def save_model(self, request, obj, form, change):
        if obj.id == None:
           obj.created_by = request.user
           super().save_model(request, obj, form, change)

        else:

           super().save_model(request, obj, form, change)


admin.site.register(Post, PostAdmin)

Upvotes: 3

Theophile
Theophile

Reputation: 300

If anyone encounters this problem I followed the advice of a_k_v and did it in views. Here is how I did it :

I added two fields in my class :

class Class(models.Model):
    created_by = models.ForeignKey(User, on_delete=models.DO_NOTHING, blank=True, null=True, related_name='create')
    updated_by = models.ForeignKey(User, on_delete=models.DO_NOTHING, blank=True, null=True, related_name='update')

then created a function :

def created_updated(model, request):
    obj = model.objects.latest('pk')
    if obj.created_by is None:
        obj.created_by = request.user
    obj.updated_by = request.user
    obj.save()

to get it into my views.py : 

    if request.method == 'POST':
    form = AddVacation(request.POST)
    if form.is_valid:
        form.save()
        created_updated(Vacation, request)

Upvotes: 2

a_k_v
a_k_v

Reputation: 1608

Refer official doc. It explained it pretty well. An example is also there

from django.views.generic.edit import CreateView
from myapp.models import Author

class AuthorCreate(CreateView):
    model = Author
    fields = ['name']

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

https://docs.djangoproject.com/en/2.1/topics/class-based-views/generic-editing/#models-and-request-user

Upvotes: 7

Related Questions