"""文章已读记录(per-user)

- article_reads.user_id       INTEGER       FK users.id    ON DELETE CASCADE
- article_reads.article_id    BIGINT        FK articles.id ON DELETE CASCADE
- article_reads.read_at       TIMESTAMPTZ   DEFAULT now()
- 复合主键 (user_id, article_id) — 天然幂等

Revision ID: 0007
Revises: 0006
Create Date: 2026-06-13
"""
from __future__ import annotations

from typing import Sequence, Union

import sqlalchemy as sa
from alembic import op

revision: str = "0007"
down_revision: Union[str, None] = "0006"
branch_labels = None
depends_on = None


def upgrade() -> None:
    op.create_table(
        "article_reads",
        sa.Column(
            "user_id",
            sa.Integer,
            sa.ForeignKey("users.id", ondelete="CASCADE"),
            nullable=False,
        ),
        sa.Column(
            "article_id",
            sa.BigInteger,
            sa.ForeignKey("articles.id", ondelete="CASCADE"),
            nullable=False,
        ),
        sa.Column(
            "read_at",
            sa.DateTime(timezone=True),
            nullable=False,
            server_default=sa.text("now()"),
        ),
        sa.PrimaryKeyConstraint("user_id", "article_id", name="pk_article_reads"),
    )
    op.create_index(
        "ix_article_reads_user_read_at",
        "article_reads",
        ["user_id", "read_at"],
    )
    op.create_index(
        "ix_article_reads_article",
        "article_reads",
        ["article_id"],
    )


def downgrade() -> None:
    op.drop_index("ix_article_reads_article", table_name="article_reads")
    op.drop_index("ix_article_reads_user_read_at", table_name="article_reads")
    op.drop_table("article_reads")