This commit is contained in:
Kenneth Kehl
2025-08-28 12:01:54 -07:00
parent ee9b3afe11
commit 887eff7d81

View File

@@ -6,19 +6,10 @@ Create Date: 2025-08-28 12:34:32.857422
""" """
from contextlib import contextmanager
from enum import Enum
from re import I
from typing import Iterator, TypedDict
import sqlalchemy as sa import sqlalchemy as sa
from alembic import op from alembic import op
from sqlalchemy.dialects import postgresql from sqlalchemy.dialects import postgresql
from app.enums import (
AuthType,
)
revision = "0418_user_state_enum" revision = "0418_user_state_enum"
down_revision = "0417_change_total_message_limit" down_revision = "0417_change_total_message_limit"
@@ -29,23 +20,25 @@ user_state_enum = postgresql.ENUM(
def upgrade(): def upgrade():
user_state_enum.create(op.get_bind(), checkfirst=True) # user_state_enum.create(op.get_bind(), checkfirst=True)
op.alter_column( # op.alter_column(
"user", # "user",
"state", # "state",
existing_type=sa.String(), # existing_type=sa.String(),
type_=user_state_enum, # type_=user_state_enum,
existing_nullable=False, # existing_nullable=False,
) # )
pass
def downgrade(): def downgrade():
op.alter_column( # op.alter_column(
"user", # "user",
"state", # "state",
existing_type=user_state_enum, # existing_type=user_state_enum,
type_=sa.String, # type_=sa.String,
existing_nullable=False, # existing_nullable=False,
) # )
user_state_enum.drop(op.get_bind(), checkfirst=True) # user_state_enum.drop(op.get_bind(), checkfirst=True)
pass