Alembic - 使用複合主鍵在MySQL中導致表定義不正確的問題

Question

我有多個使用複合主鍵的「versioned」資料庫SQLAlchemy模型，透過組合自增整型欄位（「id」）和日期時間欄位（「record_valid_from」）來實現。我正在嘗試在本地的Docker容器中運行這個模型對MySQL資料庫進行設定。

模型定義大致如下：

from sqlalchemy.orm import (DeclarativeBase, Mapped)

class classA(DeclarativeBase):
    id: Mapped[int] = mapped_column(primary_key=True, index=True, autoincrement=True)
    record_valid_from: Mapped[datetime] = mapped_column(DateTime,
        primary_key=True,
        default=get_current_timestamp # this is a python method returning datetime.now()
    )
    active: Mapped[bool] = mapped_column(Boolean, default=True,
        comment="TRUE if latest version, FALSE otherwise"
    )
    ... # some more fields and logic

其他模型看起來類似，它們之間有各種不同的關係。

當使用Alembic自動產生遷移腳本（alembic revision --autogenerate -m "init database"）時，產生的Python程式碼似乎會產生無效的SQL語句。

更具體地說，我遇到了：

(pymysql.err.OperationalError) (1075, 'Incorrect table definition; there can be only one auto column and it must be defined as a key')< ;/pre>
<p>以下是遷移程式碼（注意：我對其進行了簡化）：</p>
<pre class="brush:php;toolbar:false;">def upgrade() -> None:
    op.create_table('classA',
        sa.Column('name', sa.String(length=100), nullable=False),
        sa.Column('record_valid_from', sa.DateTime(), nullable=False),
        sa.Column('active', sa.Boolean(), nullable=False),
        sa.Column('id', sa.Integer(), autoincrement=True, nullable=False),
        sa.PrimaryKeyConstraint('record_valid_from', 'id')
    )
    op.create_index(op.f('ix_classA_id'), 'classA', ['id'], unique=False)

有人經歷過類似的情況嗎？或知道如何解決這個問題嗎？

我試過以下方法：

• 在建立表格後呼叫op.create_primary_key（請參閱：https://alembic.sqlalchemy.org/en/latest/ops.html#alembic.operations.Operations.create_primary_key）。結果：sqlalchemy.exc.OperationalError: (pymysql.err.OperationalError) (1068, '定義了多個主鍵')。
• 移除sa.PrimaryKeyConstraint然後直接呼叫op.create_primary_key。結果：
  • 遷移成功正常運作。
  • 嘗試建立一個新的ORM模型導致了以下錯誤：sqlalchemy.exc.OperationalError: (pymysql.err.OperationalError) (1364, "字段'id'沒有預設值")。

Answer 1

我花了幾個小時解決了這個問題，並且自己修復了它。對於遇到類似問題的人，這是答案：

實際上，主鍵欄位在PrimaryKeyConstraint中的順序是有影響的。我的問題是透過改變順序來解決的，而不是使用sa.PrimaryKeyConstraint('record_valid_from', 'id')，我改成了sa.PrimaryKeyConstraint("id", "record_valid_from")。

希望這可以幫到你。