Merge branch 'main' of https://github.com/fourblocks2020/portfolio-hub

migrations/versions/20260413_0010_portfolio_owner_user_id.py

@@ -0,0 +1,97 @@
+"""add portfolio owner user id
+
+Revision ID: 20260413_0010
+Revises: 20260410_0009
+Create Date: 2026-04-13 01:00:00
+"""
+
+from __future__ import annotations
+
+import sqlalchemy as sa
+from alembic import op
+from sqlalchemy import inspect
+
+
+revision = "20260413_0010"
+down_revision = "20260410_0009"
+branch_labels = None
+depends_on = None
+
+
+def _column_exists(inspector, table_name: str, column_name: str) -> bool:
+    return any(column["name"] == column_name for column in inspector.get_columns(table_name))
+
+
+def _index_exists(inspector, table_name: str, index_name: str) -> bool:
+    return any(index["name"] == index_name for index in inspector.get_indexes(table_name))
+
+
+def _fk_exists(inspector, table_name: str, fk_name: str) -> bool:
+    return any(fk.get("name") == fk_name for fk in inspector.get_foreign_keys(table_name))
+
+
+def upgrade() -> None:
+    bind = op.get_bind()
+    inspector = inspect(bind)
+
+    if not _column_exists(inspector, "portfolios", "owner_user_id"):
+        op.add_column("portfolios", sa.Column("owner_user_id", sa.Text(), nullable=True))
+
+    bind.execute(
+        sa.text(
+            """
+            WITH fallback_owner AS (
+                SELECT id
+                FROM users
+                ORDER BY created_at ASC, id ASC
+                LIMIT 1
+            )
+            UPDATE portfolios p
+            SET owner_user_id = f.id
+            FROM fallback_owner f
+            WHERE p.owner_user_id IS NULL
+            """
+        )
+    )
+
+    unresolved = bind.execute(sa.text("SELECT COUNT(*) FROM portfolios WHERE owner_user_id IS NULL")).scalar()
+    if unresolved and int(unresolved) > 0:
+        raise RuntimeError("owner_user_id backfill failed: users 테이블에 fallback 소유주가 필요합니다")
+
+    op.alter_column("portfolios", "owner_user_id", existing_type=sa.Text(), nullable=False)
+
+    inspector = inspect(bind)
+    if not _index_exists(inspector, "portfolios", "idx_portfolios_owner_updated"):
+        op.create_index(
+            "idx_portfolios_owner_updated",
+            "portfolios",
+            ["owner_user_id", sa.text("updated_at DESC")],
+            unique=False,
+        )
+
+    inspector = inspect(bind)
+    if not _fk_exists(inspector, "portfolios", "fk_portfolios_owner_user"):
+        op.create_foreign_key(
+            "fk_portfolios_owner_user",
+            "portfolios",
+            "users",
+            ["owner_user_id"],
+            ["id"],
+            ondelete="RESTRICT",
+        )
+
+
+def downgrade() -> None:
+    bind = op.get_bind()
+    inspector = inspect(bind)
+
+    if _fk_exists(inspector, "portfolios", "fk_portfolios_owner_user"):
+        op.drop_constraint("fk_portfolios_owner_user", "portfolios", type_="foreignkey")
+
+    inspector = inspect(bind)
+    if _index_exists(inspector, "portfolios", "idx_portfolios_owner_updated"):
+        op.drop_index("idx_portfolios_owner_updated", table_name="portfolios")
+
+    inspector = inspect(bind)
+    if _column_exists(inspector, "portfolios", "owner_user_id"):
+        op.drop_column("portfolios", "owner_user_id")

migrations/versions/20260414_0011_bot_principal_owner_user_id.py

@@ -0,0 +1,97 @@
+"""restore bot principal owner user id
+
+Revision ID: 20260414_0011
+Revises: 20260413_0010
+Create Date: 2026-04-14 10:30:00
+"""
+
+from __future__ import annotations
+
+import sqlalchemy as sa
+from alembic import op
+from sqlalchemy import inspect
+
+
+revision = "20260414_0011"
+down_revision = "20260413_0010"
+branch_labels = None
+depends_on = None
+
+
+def _column_exists(inspector, table_name: str, column_name: str) -> bool:
+    return any(column["name"] == column_name for column in inspector.get_columns(table_name))
+
+
+def _index_exists(inspector, table_name: str, index_name: str) -> bool:
+    return any(index["name"] == index_name for index in inspector.get_indexes(table_name))
+
+
+def _fk_exists(inspector, table_name: str, fk_name: str) -> bool:
+    return any(fk.get("name") == fk_name for fk in inspector.get_foreign_keys(table_name))
+
+
+def upgrade() -> None:
+    bind = op.get_bind()
+    inspector = inspect(bind)
+
+    if not _column_exists(inspector, "bot_principals", "owner_user_id"):
+        op.add_column("bot_principals", sa.Column("owner_user_id", sa.Text(), nullable=True))
+
+    bind.execute(
+        sa.text(
+            """
+            WITH fallback_owner AS (
+                SELECT id
+                FROM users
+                ORDER BY created_at ASC, id ASC
+                LIMIT 1
+            )
+            UPDATE bot_principals b
+            SET owner_user_id = f.id
+            FROM fallback_owner f
+            WHERE b.owner_user_id IS NULL
+            """
+        )
+    )
+
+    unresolved = bind.execute(sa.text("SELECT COUNT(*) FROM bot_principals WHERE owner_user_id IS NULL")).scalar()
+    if unresolved and int(unresolved) > 0:
+        raise RuntimeError("bot_principals.owner_user_id backfill failed: users 테이블에 fallback 소유주가 필요합니다")
+
+    op.alter_column("bot_principals", "owner_user_id", existing_type=sa.Text(), nullable=False)
+
+    inspector = inspect(bind)
+    if not _index_exists(inspector, "bot_principals", "idx_bot_principals_owner_active_created"):
+        op.create_index(
+            "idx_bot_principals_owner_active_created",
+            "bot_principals",
+            ["owner_user_id", "is_active", sa.text("created_at DESC")],
+            unique=False,
+        )
+
+    inspector = inspect(bind)
+    if not _fk_exists(inspector, "bot_principals", "fk_bot_principals_owner_user"):
+        op.create_foreign_key(
+            "fk_bot_principals_owner_user",
+            "bot_principals",
+            "users",
+            ["owner_user_id"],
+            ["id"],
+            ondelete="RESTRICT",
+        )
+
+
+def downgrade() -> None:
+    bind = op.get_bind()
+    inspector = inspect(bind)
+
+    if _fk_exists(inspector, "bot_principals", "fk_bot_principals_owner_user"):
+        op.drop_constraint("fk_bot_principals_owner_user", "bot_principals", type_="foreignkey")
+
+    inspector = inspect(bind)
+    if _index_exists(inspector, "bot_principals", "idx_bot_principals_owner_active_created"):
+        op.drop_index("idx_bot_principals_owner_active_created", table_name="bot_principals")
+
+    inspector = inspect(bind)
+    if _column_exists(inspector, "bot_principals", "owner_user_id"):
+        op.drop_column("bot_principals", "owner_user_id")

migrations/versions/20260414_0012_portfolio_tags.py

@@ -0,0 +1,79 @@
+"""add portfolio tags table
+
+Revision ID: 20260414_0012
+Revises: 20260414_0011
+Create Date: 2026-04-14 16:10:00
+"""
+
+from __future__ import annotations
+
+import sqlalchemy as sa
+from alembic import op
+from sqlalchemy import inspect
+
+
+revision = "20260414_0012"
+down_revision = "20260414_0011"
+branch_labels = None
+depends_on = None
+
+
+def _table_exists(inspector, table_name: str) -> bool:
+    return table_name in inspector.get_table_names()
+
+
+def _index_exists(inspector, table_name: str, index_name: str) -> bool:
+    return any(index["name"] == index_name for index in inspector.get_indexes(table_name))
+
+
+def _fk_exists(inspector, table_name: str, fk_name: str) -> bool:
+    return any(fk.get("name") == fk_name for fk in inspector.get_foreign_keys(table_name))
+
+
+def upgrade() -> None:
+    bind = op.get_bind()
+    inspector = inspect(bind)
+
+    if not _table_exists(inspector, "portfolio_tags"):
+        op.create_table(
+            "portfolio_tags",
+            sa.Column("portfolio_id", sa.Text(), nullable=False),
+            sa.Column("tag_value", sa.Text(), nullable=False),
+            sa.Column("created_at", sa.TIMESTAMP(timezone=True), nullable=False, server_default=sa.text("NOW()")),
+            sa.PrimaryKeyConstraint("portfolio_id", "tag_value", name="pk_portfolio_tags"),
+        )
+
+    inspector = inspect(bind)
+    if not _fk_exists(inspector, "portfolio_tags", "fk_portfolio_tags_portfolio"):
+        op.create_foreign_key(
+            "fk_portfolio_tags_portfolio",
+            "portfolio_tags",
+            "portfolios",
+            ["portfolio_id"],
+            ["portfolio_id"],
+            ondelete="CASCADE",
+        )
+
+    inspector = inspect(bind)
+    if not _index_exists(inspector, "portfolio_tags", "idx_portfolio_tags_tag_portfolio"):
+        op.create_index(
+            "idx_portfolio_tags_tag_portfolio",
+            "portfolio_tags",
+            ["tag_value", "portfolio_id"],
+            unique=False,
+        )
+
+
+def downgrade() -> None:
+    bind = op.get_bind()
+    inspector = inspect(bind)
+
+    if _table_exists(inspector, "portfolio_tags"):
+        if _fk_exists(inspector, "portfolio_tags", "fk_portfolio_tags_portfolio"):
+            op.drop_constraint("fk_portfolio_tags_portfolio", "portfolio_tags", type_="foreignkey")
+
+        inspector = inspect(bind)
+        if _index_exists(inspector, "portfolio_tags", "idx_portfolio_tags_tag_portfolio"):
+            op.drop_index("idx_portfolio_tags_tag_portfolio", table_name="portfolio_tags")
+
+        op.drop_table("portfolio_tags")