ihrm/onboarding/models.py

"""
models.py

This module is used to register models for onboarding app

"""
from django.db import models
from django.utils.translation import gettext_lazy as _
from recruitment.models import Recruitment, Candidate
from employee.models import Employee


class OnboardingStage(models.Model):
    """
    OnboardingStage models
    """

    stage_title = models.CharField(max_length=200)
    recruitment_id = models.ManyToManyField(Recruitment)
    employee_id = models.ManyToManyField(Employee)
    sequence = models.IntegerField(null=True)

    def __str__(self):
        return f"{self.stage_title}"


class OnboardingTask(models.Model):
    """
    OnboardingTask models
    """

    task_title = models.CharField(max_length=200)
    recruitment_id = models.ManyToManyField(Recruitment, related_name="onboarding_task")
    employee_id = models.ManyToManyField(Employee)

    def __str__(self):
        return f"{self.task_title}"


class CandidateStage(models.Model):
    """
    CandidateStage model
    """

    candidate_id = models.OneToOneField(
        Candidate, on_delete=models.CASCADE, related_name="onboarding_stage"
    )
    onboarding_stage_id = models.ForeignKey(
        OnboardingStage, on_delete=models.CASCADE, related_name="candidate"
    )

    def __str__(self):
        return f"{self.candidate_id}  |  {self.onboarding_stage_id}"


class CandidateTask(models.Model):
    """
    CandidateTask model
    """

    Choice = (
        ("", ""),
        ("todo", _("Todo")),
        ("scheduled", _("Scheduled")),
        ("ongoing", _("Ongoing")),
        ("stuck", _("Stuck")),
        ("done", _("Done")),
    )
    candidate_id = models.ForeignKey(
        Candidate, on_delete=models.CASCADE, related_name="candidate_task"
    )
    status = models.CharField(max_length=50, choices=Choice, blank=True, null=True)
    onboarding_task_id = models.ForeignKey(OnboardingTask, on_delete=models.CASCADE)

    def __str__(self):
        return f"{self.candidate_id} | {self.onboarding_task_id} | {self.status}"

    class Meta:
        """
        Meta class to add some additional options
        """

        unique_together = ("candidate_id", "onboarding_task_id")


class OnboardingPortal(models.Model):
    """
    OnboardingPortal model
    """

    candidate_id = models.OneToOneField(
        Candidate, on_delete=models.CASCADE, related_name="onboarding_portal"
    )
    token = models.CharField(max_length=200)
    used = models.BooleanField(default=False)
    count = models.IntegerField(default=0)

    def __str__(self):
        return f"{self.candidate_id} | {self.token}"