fonrey/apps/org/models/staff.py

from django.conf import settings
from django.db import models

from core.enums import StaffGender, StaffIdType, StaffRole, StaffStatus
from core.models.base import SoftDeleteModel


class Staff(SoftDeleteModel):
    org_unit = models.ForeignKey(
        "org.OrgUnit",
        on_delete=models.RESTRICT,
        related_name="staff_members",
        db_index=True,
    )
    user = models.OneToOneField(
        settings.AUTH_USER_MODEL,
        null=True,
        blank=True,
        on_delete=models.SET_NULL,
        related_name="staff_profile",
    )
    name = models.CharField(max_length=50)
    nickname = models.CharField(max_length=50, blank=True, default="")
    employee_no = models.CharField(max_length=30, null=True, blank=True, unique=True)
    role = models.CharField(max_length=30, choices=StaffRole.choices)
    job_title = models.CharField(max_length=100, blank=True, default="")
    job_category = models.CharField(
        max_length=50,
        blank=True,
        default="",
        help_text="Job classification (e.g. '置业顾问' = agent qualification flag).",
    )
    job_level = models.SmallIntegerField(null=True, blank=True)
    supervisor = models.ForeignKey(
        "self",
        null=True,
        blank=True,
        on_delete=models.SET_NULL,
        related_name="subordinates",
    )
    status = models.CharField(
        max_length=20,
        choices=StaffStatus.choices,
        default=StaffStatus.ACTIVE,
    )
    phone_enc = models.BinaryField(
        null=True,
        blank=True,
        help_text="AES-256-GCM encrypted phone (DATA_MODEL_ORG §3.2).",
    )
    phone_hash = models.CharField(max_length=64, null=True, blank=True, db_index=True)
    phone_hide = models.BooleanField(default=False)
    email = models.EmailField(max_length=255, blank=True, default="")
    extension = models.CharField(max_length=20, blank=True, default="")
    avatar_key = models.TextField(blank=True, default="")
    is_active = models.BooleanField(default=True)
    is_system_admin = models.BooleanField(default=False)
    first_joined_at = models.DateField(null=True, blank=True)
    rejoined_at = models.DateField(null=True, blank=True)
    resigned_at = models.DateField(null=True, blank=True)
    joined_count = models.SmallIntegerField(default=1)
    industry_exp_years = models.SmallIntegerField(null=True, blank=True)
    mentor = models.ForeignKey(
        "self",
        null=True,
        blank=True,
        on_delete=models.SET_NULL,
        related_name="mentees",
    )
    business_type = models.CharField(max_length=50, blank=True, default="")
    bank_name = models.CharField(max_length=100, blank=True, default="")
    bank_account = models.CharField(max_length=50, blank=True, default="")
    partner_no = models.CharField(max_length=50, blank=True, default="")
    recruit_by = models.ForeignKey(
        "self",
        null=True,
        blank=True,
        on_delete=models.SET_NULL,
        related_name="recruited_staff",
    )
    recruit_source = models.CharField(max_length=50, blank=True, default="")
    referrer = models.ForeignKey(
        "self",
        null=True,
        blank=True,
        on_delete=models.SET_NULL,
        related_name="referred_staff",
    )

    class Meta:
        db_table = "staff"
        indexes = [
            models.Index(fields=["org_unit"], name="idx_staff_org_unit"),
            models.Index(fields=["supervisor"], name="idx_staff_supervisor"),
            models.Index(fields=["status"], name="idx_staff_status"),
        ]

    def __str__(self) -> str:
        return self.name


class StaffPersonalInfo(models.Model):
    staff = models.OneToOneField(
        "org.Staff",
        on_delete=models.CASCADE,
        related_name="personal_info",
        primary_key=True,
    )
    gender = models.CharField(max_length=10, choices=StaffGender.choices, blank=True, default="")
    id_type = models.CharField(max_length=20, choices=StaffIdType.choices, blank=True, default="")
    id_number_enc = models.BinaryField(null=True, blank=True)
    id_number_hash = models.CharField(max_length=64, null=True, blank=True, db_index=True)
    id_verified = models.BooleanField(default=False)
    id_verified_at = models.DateTimeField(null=True, blank=True)
    birthdate = models.DateField(null=True, blank=True)
    native_place = models.CharField(max_length=100, blank=True, default="")
    domicile_type = models.CharField(max_length=20, blank=True, default="")
    marital_status = models.CharField(max_length=20, blank=True, default="")
    political_status = models.CharField(max_length=20, blank=True, default="")
    has_children = models.BooleanField(null=True, blank=True)
    education_level = models.CharField(max_length=20, blank=True, default="")
    ethnicity = models.CharField(max_length=20, blank=True, default="")
    domicile_address = models.CharField(max_length=200, blank=True, default="")
    residence_address = models.CharField(max_length=200, blank=True, default="")
    work_start_date = models.DateField(null=True, blank=True)
    emergency_contact = models.CharField(max_length=50, blank=True, default="")
    emergency_phone_enc = models.BinaryField(null=True, blank=True)
    updated_at = models.DateTimeField(auto_now=True)
    updated_by = models.ForeignKey(
        "org.Staff",
        null=True,
        blank=True,
        on_delete=models.SET_NULL,
        related_name="updated_personal_info",
    )

    class Meta:
        db_table = "staff_personal_info"