在 SQL Server 2008 上使用 sqlalchemy 等待数据库恢复完成

Question

我正在尝试在开发过程中自动执行数据库恢复，在 SQL Server 2008 上使用 TSQL，使用 sqlalchemy 和 pyodbc 作为传输。

我正在执行的命令是：

"""CREATE DATABASE dbname

Restore database dbname FROM DISK='C:\Backups\dbname.bak' WITH REPLACE,MOVE 'dbname_data' TO 'C:\Databases\dbname_data.mdf',MOVE 'dbname_log' TO 'C:\Databases\dbname_log.ldf'"""

不幸的是，在 SQL Management Studio 中，代码运行后，我看到数据库仍处于“正在恢复...”状态。

如果我通过管理工作室恢复，它就可以工作。如果我使用子进程调用“sqlcmd”，它就可以工作。 pymssql 在身份验证方面存在问题，甚至还没有做到这一点。

可能出了什么问题？

Answer 1

BACKUP 和 RESTORE 语句异步运行，因此它们在继续执行其余代码之前不会终止。

使用 while 语句，如 http://ryepup.unwashedmeme.com/blog/2010/08/26/making-sql-server-backups-using-python-and-pyodbc/ 为我解决了这个问题：

# setup your DB connection, cursor, etc
cur.execute('BACKUP DATABASE ? TO DISK=?', 
            ['test', r'd:\temp\test.bak'])
while cur.nextset():
    pass

Answer 2

无法重现直接从 pyodbc（没有 sqlalchemy）恢复的问题，执行以下操作：

connection = pyodbc.connect(connection_string) # ensure autocommit is set to `True` in connection string
cursor = connection.cursor()
affected = cursor.execute("""CREATE DATABASE test
RESTORE DATABASE test FROM DISK = 'D:\\test.bak' WITH REPLACE, MOVE 'test_data' TO 'D:\\test_data.mdf', MOVE 'test_log' to 'D:\\test_log.ldf' """)
while cursor.nextset():
    pass

一些需要澄清的问题：

• 使用 sqlalchemy 进行恢复的代码是什么？
• 使用的 SQL Server ODBC 驱动程序版本是什么？
• SQL Server 日志中是否有与还原相关的消息？

感谢 geographika 提供的Cursor.nextset() 示例！

Answer 3

对于 SQL Alchemy 用户，感谢 geographika 的回答：我最终使用了“原始”DBAPI 连接 来自连接池。

它与 geographika 的解决方案完全相同，但有一些额外的部分：

import sqlalchemy as sa
driver = 'SQL+Server'
name = 'servername'
sql_engine_str = 'mssql+pyodbc://'\
                     + name\
                     + '/'\
                     + 'master'\
                     + '?driver='\
                     + driver
engine = sa.create_engine(sql_engine_str, connect_args={'autocommit': True})

connection = engine.raw_connection()
try:
  cursor = connection.cursor()
  sql_cmd = """
        RESTORE DATABASE [test]
        FROM DISK = N'...\\test.bak'
        WITH FILE = 1,
         MOVE N'test'
         TO N'...\\test_Primary.mdf',
         MOVE N'test_log'
         TO N'...\\test_log.ldf',
         RECOVERY,
         NOUNLOAD,
         STATS = 5,
         REPLACE
        """
  cursor.execute(sql_cmd)
  while cursor.nextset():
       pass
except Exception as e:
  logger.error(str(e), exc_info=True)

Answer 4

五件事解决了我的问题，具有相同的症状。

1. 发现我的 test.bak 文件包含错误的 mdf 和 ldf 文件：

   <前><代码>>>> cursor.execute(r"从磁盘恢复文件列表 = 'test.bak'").fetchall()
   [(u'WRONGNAME', u'C:\\Program Files\\Microsoft SQL ...),
   (u'WRONGNAME_log', u'C:\\Program Files\\Microsoft SQL ...)]

2. 创建了一个新的 bak 文件并确保设置仅复制备份选项< /p>

3. 为我的连接设置自动提交选项。

   connection = pyodbc.connect(connection_string, autocommit=True)
   

4. 仅将connection.cursor用于单个RESTORE命令，仅用于单个RESTORE命令，

5. 将test_data MOVE更正为在我的 RESTORE 命令中进行测试（由 @beargle 提供）。

   affected=cursor.execute("""从磁盘恢复数据库测试='test.bak'并替换，将'test'移动到'C:\\test.mdf'，将'test_log'移动到' C:\\test_log.ldf'""")