49 lines
1.8 KiB
Python
49 lines
1.8 KiB
Python
"""create_initial_schema
|
||
|
||
从 Base.metadata 创建所有表（新库部署用）。
|
||
对已有数据库：若已 stamp 0001_baseline，本迁移会跳过已存在的表（create_all checkfirst）。
|
||
|
||
Revision ID: 0002_schema
|
||
Revises: 0001_baseline
|
||
Create Date: 2026-03-18
|
||
|
||
"""
|
||
from typing import Sequence, Union
|
||
|
||
from alembic import op
|
||
from sqlalchemy import text
|
||
|
||
revision: str = "0002_schema"
|
||
down_revision: Union[str, None] = "0001_baseline"
|
||
branch_labels: Union[str, Sequence[str], None] = None
|
||
depends_on: Union[str, Sequence[str], None] = None
|
||
|
||
|
||
def upgrade() -> None:
|
||
conn = op.get_bind()
|
||
conn.execute(text("CREATE EXTENSION IF NOT EXISTS vector"))
|
||
|
||
# 导入所有 model 以注册到 Base.metadata
|
||
from app.core.db import Base
|
||
from app.features.auth import models as _auth_models # noqa: F401
|
||
from app.features.conversation import models as _conv_models # noqa: F401
|
||
from app.features.memory import models as _memory_models # noqa: F401
|
||
from app.features.memoir import models as _memoir_models # noqa: F401
|
||
from app.features.payment import models as _payment_models # noqa: F401
|
||
from app.features.user import models as _user_models # noqa: F401
|
||
|
||
Base.metadata.create_all(bind=conn)
|
||
|
||
|
||
def downgrade() -> None:
|
||
conn = op.get_bind()
|
||
from app.core.db import Base
|
||
from app.features.auth import models as _auth_models # noqa: F401
|
||
from app.features.conversation import models as _conv_models # noqa: F401
|
||
from app.features.memory import models as _memory_models # noqa: F401
|
||
from app.features.memoir import models as _memoir_models # noqa: F401
|
||
from app.features.payment import models as _payment_models # noqa: F401
|
||
from app.features.user import models as _user_models # noqa: F401
|
||
|
||
Base.metadata.drop_all(bind=conn)
|