Newer
Older
gnexus-tasks / backend / alembic / versions / e1f0a2b3c4d5_garden_coins_items.py
"""Сад-сцена: журнал монет (coin_events) и элементы сцены (garden_items).

Монеты начисляются за закрытые задачи (floor(xp/2)) задним числом по всем
существующим xp_events; бонусы уровней и траты — в сервисе сада.

Revision ID: e1f0a2b3c4d5
Revises: d9e8f7c6b5a4
Create Date: 2026-09-20
"""

from collections.abc import Sequence

import sqlalchemy as sa

from alembic import op

revision: str = "e1f0a2b3c4d5"
down_revision: str | Sequence[str] | None = "d9e8f7c6b5a4"
branch_labels: str | Sequence[str] | None = None
depends_on: str | Sequence[str] | None = None


def upgrade() -> None:
    op.create_table(
        "coin_events",
        sa.Column("id", sa.Integer(), primary_key=True),
        sa.Column(
            "task_id", sa.Integer(), sa.ForeignKey("tasks.id", ondelete="SET NULL"), nullable=True
        ),
        sa.Column("level", sa.Integer(), nullable=True),
        sa.Column("source", sa.String(length=16), nullable=False, server_default="task"),
        sa.Column("amount", sa.Integer(), nullable=False),
        sa.Column("item_key", sa.String(length=50), nullable=True),
        sa.Column("created_at", sa.DateTime(timezone=True), nullable=False),
    )
    op.create_table(
        "garden_items",
        sa.Column("id", sa.Integer(), primary_key=True),
        sa.Column("kind", sa.String(length=16), nullable=False),
        sa.Column(
            "ref_id", sa.Integer(), sa.ForeignKey("xp_events.id", ondelete="CASCADE"), nullable=True
        ),
        sa.Column("item_key", sa.String(length=50), nullable=True),
        sa.Column("x", sa.Integer(), nullable=False, server_default="0"),
        sa.Column("y", sa.Integer(), nullable=False, server_default="0"),
        sa.Column("stage", sa.Integer(), nullable=False, server_default="0"),
        sa.Column("created_at", sa.DateTime(timezone=True), nullable=False),
        sa.UniqueConstraint("kind", "ref_id"),
    )
    # Монеты за уже закрытые задачи задним числом
    op.get_bind().execute(
        sa.text(
            "INSERT INTO coin_events (task_id, source, amount, created_at) "
            "SELECT task_id, 'task', amount / 2, created_at FROM xp_events"
        )
    )


def downgrade() -> None:
    op.drop_table("garden_items")
    op.drop_table("coin_events")