使用Python pymssql导入.bak数据库提示找不到_mssql模块如何解决

导入错误根因 #

pymssql 2.1及以上版本已经不再对外暴露底层_mssql模块的直接导入路径，你使用的2.2.2版本不再支持import _mssql的写法，这是抛出No module named '_mssql'错误的直接原因。

现有操作的潜在问题 #

• 执行RESTORE DATABASE、ALTER DATABASE这类操作不允许在事务中执行，必须开启自动提交，否则即使修复导入问题也会执行失败。
• 代码中字符串路径'C:\mydb.bak'的反斜杠没有转义，会被Python识别为转义字符，实际传入SQL Server的路径会出错。
• 重命名已有数据库前没有断开活跃连接，一旦有其他进程连接到目标库，ALTER DATABASE操作会直接报错。
• 没有指定备份文件的数据、日志文件映射路径，如果备份文件的原始存储路径和你本地SQL Server的存储路径不一致，还原操作会失败。
• 初始化连接时指定了自定义database参数，还原库的操作需要在master库上下文执行，指定其他库可能导致操作被锁。

正确实现步骤 #

第一步：确认依赖安装 #

执行命令确认pymssql安装正常：
pip install pymssql==2.2.2

第二步：获取备份文件的逻辑文件名 #

你可以先在SQL Server管理工具中执行如下SQL，获取bak文件中包含的数据、日志文件逻辑名，后续还原需要用到：

RESTORE FILELISTONLY FROM DISK='C:\mydb.bak'

返回结果中LogicalName列就是逻辑名，Type列D对应数据文件（mdf）、L对应日志文件（ldf）。

第三步：执行还原代码 #

import pymssql

# 连接到master库，开启自动提交
conn = pymssql.connect(
    server='你的SQL Server地址/实例名，本地默认实例填.或(local)',
    user='SQL Server登录账号',
    password='登录密码',
    database='master',
    autocommit=True # *必须开启，库级别操作不支持事务执行*
)
cursor = conn.cursor()

# 重命名已有同名库，先强制断开所有活跃连接
cursor.execute("""
IF EXISTS (SELECT 1 FROM sys.databases WHERE name = 'mydb')
BEGIN
    ALTER DATABASE mydb SET SINGLE_USER WITH ROLLBACK IMMEDIATE
    ALTER DATABASE mydb MODIFY NAME = mydb_old
END
""")

# 执行还原，替换MOVE后的逻辑名和本地存储路径
cursor.execute("""
RESTORE DATABASE mydb
FROM DISK='C:\\mydb.bak'
WITH 
    MOVE '你查询到的mdf逻辑名' TO 'C:\\SQLServer存储路径\\mydb.mdf',
    MOVE '你查询到的ldf逻辑名' TO 'C:\\SQLServer存储路径\\mydb_log.ldf',
    REPLACE
""")

cursor.close()
conn.close()

额外注意事项 #

• 如果你使用Windows身份验证登录SQL Server，不需要传user和password参数，添加trusted_connection=True即可。
• SQL Server运行身份需要有bak文件的读取权限、目标存储路径的写入权限，否则会报权限错误。
• 如果你不需要保留旧的mydb库，可以删除重命名逻辑，直接在RESTORE命令中加REPLACE覆盖已有库。

内容的提问来源于stack exchange，提问作者FluidMechanics Potential Flows