ce139bbac3
Step 1 of Google login integration: Alembic migration for oauth_accounts + avatar_url on users, OAuthAccount model with User relationship, UserProfile schema extended with avatar_url, get_current_user updated to include avatar_url. Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
57 lines
2.3 KiB
Python
57 lines
2.3 KiB
Python
"""Add oauth_accounts table, nullable password_hash, avatar_url to users.
|
|
|
|
Revision ID: b4c0d1e2f3a4
|
|
Revises: a3b9c0d1e2f3
|
|
Create Date: 2026-04-10 00:00:00.000000
|
|
|
|
"""
|
|
from __future__ import annotations
|
|
|
|
from typing import Sequence, Union
|
|
|
|
import sqlalchemy as sa
|
|
from alembic import op
|
|
from sqlalchemy.dialects import postgresql
|
|
|
|
|
|
# revision identifiers, used by Alembic.
|
|
revision: str = "b4c0d1e2f3a4"
|
|
down_revision: Union[str, None] = "a3b9c0d1e2f3"
|
|
branch_labels: Union[str, Sequence[str], None] = None
|
|
depends_on: Union[str, Sequence[str], None] = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
# ── users: make password_hash nullable (social users have no password) ──
|
|
op.alter_column("users", "password_hash", existing_type=sa.String(255), nullable=True)
|
|
|
|
# ── users: add avatar_url ─────────────────────────────────────────────
|
|
op.add_column("users", sa.Column("avatar_url", sa.String(2048), nullable=True))
|
|
|
|
# ── oauth_accounts ────────────────────────────────────────────────────
|
|
op.create_table(
|
|
"oauth_accounts",
|
|
sa.Column("id", postgresql.UUID(as_uuid=False), nullable=False),
|
|
sa.Column("user_id", postgresql.UUID(as_uuid=False), nullable=False),
|
|
sa.Column("provider", sa.String(50), nullable=False),
|
|
sa.Column("provider_user_id", sa.String(255), nullable=False),
|
|
sa.Column("provider_email", sa.String(255), nullable=True),
|
|
sa.Column(
|
|
"created_at",
|
|
sa.DateTime(timezone=True),
|
|
nullable=False,
|
|
server_default=sa.text("now()"),
|
|
),
|
|
sa.PrimaryKeyConstraint("id"),
|
|
sa.ForeignKeyConstraint(["user_id"], ["users.id"], ondelete="CASCADE"),
|
|
sa.UniqueConstraint("provider", "provider_user_id", name="uq_oauth_provider_user"),
|
|
)
|
|
op.create_index("ix_oauth_accounts_user_id", "oauth_accounts", ["user_id"])
|
|
|
|
|
|
def downgrade() -> None:
|
|
op.drop_index("ix_oauth_accounts_user_id", table_name="oauth_accounts")
|
|
op.drop_table("oauth_accounts")
|
|
op.drop_column("users", "avatar_url")
|
|
op.alter_column("users", "password_hash", existing_type=sa.String(255), nullable=False)
|