from .base import pgdb


dna_account_claim_table = pgdb.Table(
    "dna_account_claim",
    pgdb.Model.metadata,
    pgdb.Column("id", pgdb.Integer, primary_key=True),
    pgdb.Column(
        "dna_account_id",
        pgdb.Integer,
        pgdb.ForeignKey("dna_account.id"),
        nullable=False,
    ),
    pgdb.Column(
        "claim_value_id",
        pgdb.Integer,
        pgdb.ForeignKey("claim_value.id"),
        nullable=False,
    ),
    # explicit claim name for global claims
    pgdb.Column(
        "claim_name_id",
        pgdb.Integer,
        pgdb.ForeignKey("claim_name.id"),
        nullable=True,
    ),
    pgdb.Column("created_at", pgdb.DateTime, nullable=False),
    pgdb.Column("updated_at", pgdb.DateTime, nullable=False),
    pgdb.Column("is_disabled", pgdb.Boolean),
)

dna_account_tag_table = pgdb.Table(
    "dna_account_tag",
    pgdb.Model.metadata,
    pgdb.Column("id", pgdb.Integer, primary_key=True),
    pgdb.Column(
        "dna_account_id",
        pgdb.Integer,
        pgdb.ForeignKey("dna_account.id"),
        nullable=False,
    ),
    pgdb.Column(
        "tag_id",
        pgdb.Integer,
        pgdb.ForeignKey("tag.id"),
        nullable=False,
    ),
    pgdb.Column("created_at", pgdb.DateTime, nullable=False),
    pgdb.Column("updated_at", pgdb.DateTime, nullable=False),
)