from atlas_um.pgdb import pgdb
from atlas_um.pgdb.queries.internal_user_domain import InternalUserDomainQuery


class InternalUserDomain(pgdb.Model):
    __tablename__ = "internal_user_domain"
    query_class = InternalUserDomainQuery

    id = pgdb.Column(pgdb.Integer, primary_key=True)
    domain = pgdb.Column(pgdb.String(), nullable=False, unique=True)

    def __str__(self):
        return self.domain