将Python DataFrame写入SQL Server后无法设置主键及非空约束

问题原因分析 #

你的代码存在两个关键问题，导致修改表结构的操作无效果：

1. SQL语法不兼容：SQL Server不支持用反引号`包裹列名，正确的写法是使用方括号[]。
2. 事务未提交：SQLAlchemy的engine.connect()上下文默认不会自动提交事务，执行的ALTER语句会被自动回滚。

修正后的代码 #

import sqlalchemy as sa
from sqlalchemy import text

engine = sa.create_engine("mssql+pyodbc:///?odbc_connect={}".format(params))

# 写入DataFrame到SQL Server
df.to_sql('table_name', con=engine, if_exists='replace', index=False)

# 使用begin()上下文自动提交事务，同时修正SQL语法
with engine.begin() as con:
    # 先设置列非空约束（需确保DataFrame中该列无空值）
    con.execute(text('ALTER TABLE table_name ALTER COLUMN [column] INTEGER NOT NULL;'))
    # 添加主键约束，用方括号包裹列名适配SQL Server语法
    con.execute(text('ALTER TABLE table_name ADD PRIMARY KEY ([column]);'))

额外注意事项 #

• 必须确保DataFrame中column列没有空值（NaN），否则执行ALTER COLUMN ... NOT NULL会直接报错，导致操作失败。可以提前在Python中处理空值：df['column'] = df['column'].fillna(默认值) 或过滤含空值的行。
• 如果列名是SQL Server的关键字（如USER、DATE），必须用方括号包裹，否则会触发语法错误。

内容的提问来源于stack exchange，提问作者Rosey18