from atlas_um.pgdb import pgdb
from atlas_um.pgdb.queries.personnel_type import PersonnelTypeQuery


class PersonnelType(pgdb.Model):
    __tablename__ = "personnel_type"
    query_class = PersonnelTypeQuery

    id = pgdb.Column(pgdb.Integer, primary_key=True)
    name = pgdb.Column(pgdb.String(), nullable=False, unique=True)
    slug = pgdb.Column(pgdb.String(), nullable=False, unique=True)
    is_deleted = pgdb.Column(pgdb.Boolean(), nullable=False, default=False)

    # Relations
    dna_accounts = pgdb.relationship(
        "DNAAccount", back_populates="personnel_type"
    )

    def __str__(self):
        return self.name