How to create a special function when a model instance is created?

I'm making a system wherein when the user creates a log he gains points from how it. How do I go about in trying to make this? I tried making a signal.py function but it's giving me an error of 'DPRLog' object has no attribute 'Points'. Am I doing it right? I just want to add points whenever I create a log so I placed it as signal.py class. Can anyone help me out? Thanks

Heres my models.py:

from django.db import models
from profiles.models import User
from django.urls import reverse



# Create your models here.

class Points(models.Model):
    user = models.OneToOneField(User, on_delete=models.CASCADE)
    points = models.IntegerField(default=0, null=False)

    def __str__(self):
        return self.user.username


class Profile(models.Model):
    user = models.OneToOneField(User, on_delete=models.CASCADE)
    image = models.ImageField(default='default.png', upload_to='profile_pics')

    def __str__(self):
        return f'{self.user.username} Profile'


class Manager(models.Model):
    manager = models.OneToOneField(User, on_delete=models.CASCADE)

    def __str__(self):
        return self.manager.full_name


class Member(models.Model):
    manager = models.ForeignKey(Manager, on_delete=models.CASCADE)
    member = models.OneToOneField(User, on_delete=models.CASCADE)
    name = models.CharField(max_length=30, null=True)

    def __str__(self):
        return self.member.full_name


class Job(models.Model):
    manager = models.ForeignKey(Manager, on_delete=models.CASCADE)
    member = models.ForeignKey(Member, on_delete=models.CASCADE)
    title = models.CharField(max_length=30, blank=False, null=False)
    description = models.TextField()
    datePosted = models.DateTimeField(auto_now=True)
    file = models.FileField(null=True, blank=True, upload_to='job_files')

    def __str__(self):
        return self.title

    def get_absolute_url(self):
        return reverse('job-detail', kwargs={'pk': self.pk})

class DPRLog(models.Model):
    STATUS_CHOICES = (

        ('PENDING', 'PENDING'),
        ('CANCELLED', 'CANCELLED'),
        ('COMPLETED', 'COMPLETED'),


        )
    TASKS_CHOICES = (
        ('TESTS EXECUTION', 'TESTS EXECUTION'),
        ('TESTS DESIGN', 'TESTS DESIGN'),
        ('MOBILE TESTING WORKSHOP', 'MOBILE TESTING WORKSHOP'),
        ('BENCH ACTIVITY', 'BENCH ACTIVITY'),
        ('DEFECT ANALYSIS','DEFECT ANALYSIS'),
    )




    testuser = models.ForeignKey(User,on_delete = models.CASCADE,unique_for_date= 'reportDate')
    status = models.CharField(max_length=30, choices=STATUS_CHOICES,null=True)
    reportDate = models.DateField(blank=False, null=False)
    login = models.TimeField(blank=False, null=False)
    logout = models.TimeField(blank=False, null=False)
    totalHours = models.DecimalField(max_digits=4,decimal_places=2,null=True)
    OTHours = models.DecimalField(max_digits=4,decimal_places=2,null=True)
    mainTasks = models.CharField(max_length=50, blank=False, choices=TASKS_CHOICES, null=True)
    remarks = models.CharField(max_length=30,null=True)

    def __str__(self):
        return f'{self.testuser.full_name} DPR Log'

    @property
    def CalculateTotalHours(self):
        self.totalHours = self.logout.timeDelta - self.login.TimeDelta
        return self.TotalHours

    @property
    def OTHours(self):
        if self.totalHours > 9.00:
            self.OTHours = self.totalHours-9.00

And Signals.py:

@receiver(post_save, sender=DPRLog)
def add_point(sender, instance, created, **kwargs):
    instance.Points+=1

Upvotes: 0

Views: 39

Answers (1)

Charnel
Charnel

Reputation: 4432

There is no Points field on DPRLog table. I guess you wanted to increase points field on Points model through DPRLog -> User > Points so, unless you define related_name in your one-to-one filed, ORM will use lowercased model name to access related object:

instance.testuser.points.points +=1
instance.testuser.points.save()

Upvotes: 1

Related Questions