from atlas_um.pgdb import pgdb
from atlas_um.pgdb.queries.busines_unit import BusinessUnitQuery


class BusinessUnit(pgdb.Model):
    __tablename__ = "business_unit"
    query_class = BusinessUnitQuery

    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="business_unit"
    )

    def __str__(self):
        return self.name