Untitled

from django.db import models

class Student(models.Model):
    name = models.CharField(max_length=255)
    class_name = models.CharField(max_length=10)
    email = models.EmailField()

class Staff(models.Model):
    name = models.CharField(max_length=255)
    department = models.CharField(max_length=255)
    email = models.EmailField()

class PivotPresence(models.Model):
    points = models.IntegerField()
    school_name = models.CharField(max_length=255)
    actor_id = models.PositiveIntegerField()
    actor_type = models.CharField(max_length=50)

    def get_actor(self):
        if self.actor_type == 'Student':
            return Student.objects.get(id=self.actor_id)
        elif self.actor_type == 'Staff':
            return Staff.objects.get(id=self.actor_id)
        else:
            return None