SQLalchemy 0.7的使用示例：使用from_statement()进行UPDATE操作

Question

SQLalchemy 0.7的使用示例：使用from_statement()进行UPDATE操作

3

我正在编写一个快速的迁移脚本，用于更新一张包含500000行数据的表中的单个字段。

由于我没有计划为连接查询编写完整的模型来获取最初的大约25000行数据，因此我一直在尝试使用from_statement()调用和使用自己的原始SQL来进行UPDATE语句，但是我找不到任何示例。

此外，SQLalchemy抛出了一个错误。以下是我的调用和错误示例：

mydb = self.session()
mydb.query().from_statement(
    """
    UPDATE my_table
    SET settings=mysettings
    WHERE user_id=myuserid AND setting_id=123
    """).params(mysettings=new_settings, myuserid=user_id).all()

我遇到的错误：

Traceback (most recent call last):
  File "./sample_script.py", line 111, in <module>
    main()
  File "./sample_script.py", line 108, in main
    migrate.set_migration_data()
  File "./sample_script.py", line 100, in set_migration_data
    """).params(mysettings=new_settings, myuserid=user_id).all()
  File "/usr/lib/pymodules/python2.6/sqlalchemy/orm/query.py", line 1267, in all
    return list(self)
  File "/usr/lib/pymodules/python2.6/sqlalchemy/orm/query.py", line 1361, in __iter__
    return self._execute_and_instances(context)
  File "/usr/lib/pymodules/python2.6/sqlalchemy/orm/query.py", line 1364, in _execute_and_instances
    result = self.session.execute(querycontext.statement, params=self._params, mapper=self._mapper_zero_or_none())
  File "/usr/lib/pymodules/python2.6/sqlalchemy/orm/query.py", line 251, in _mapper_zero_or_none
    if not getattr(self._entities[0], 'primary_entity', False):
IndexError: list index out of range

更新

我正在使用MySQL。

根据Samy的建议，我尝试了以下操作：

mydb.execute(
    "UPDATE mytable SET settings=:mysettings WHERE user_id=:userid AND setting_id=123",
    {'userid': user_id, 'mysettings': new_settings}
    )

这没有任何效果。我没有收到任何错误消息，但是语句似乎并没有真正执行，因为行没有更改。如果我手动剪切和粘贴来自echo=True选项的记录查询，行在数据库中更新得很好。

更新-已解决

Samy的建议是正确的，但.execute()调用仅适用于'engine'，而不是'session'，因此这可以很好地工作：

self.engine.execute(
    "UPDATE mytable SET settings=:mysettings WHERE user_id=:userid AND setting_id=123",
    {'userid': user_id, 'mysettings': new_settings}
    )

- iandouglas

2个回答

3

您需要使用正确的参数语法；其格式完全取决于您的数据库适配器。例如，一些适配器支持:name参数，在这种情况下，您的查询中缺少这些冒号：

mydb.query().from_statement(
    """
    UPDATE my_table
    SET settings=:mysettings
    WHERE user_id=:myuserid AND setting_id=123
    """).params(mysettings=new_settings, myuserid=user_id).all()

DBAPI 2.0规范支持多种格式，包括使用?和%s占位符的位置参数，以及上述形式的命名参数和%(name)s格式。您需要查看数据库适配器文档以确定支持哪种格式。

- Martijn Pieters

谢谢Martijn。添加冒号没有帮助。我已经在我的问题中标记了MySQL，但应该在问题的文本中指定它。我会尝试找到我需要的文档，但可能会切换到原始的mysql驱动程序以更快地完成此操作。 - iandouglas

mysqldb 使用 %s 参数（位置参数）。 - Martijn Pieters

谢谢，我也试过了，错误消失了，但实际上没有写入数据库。 - iandouglas

你提交了交易吗？ - Martijn Pieters

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- Samy Vilar · Accepted Answer

这很奇怪，根据文档，from_statement 用于 SELECT 语句。

执行给定的 SELECT 语句并返回结果。

我可能在看错函数，或者可能可以使用其他类型的语句，我不太确定。

您可以只使用 execute，因为它可以执行任何类型的语句，以下是一个快速示例。

from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker

session = sessionmaker(bind = create_engine('sqlite://'), autocommit = True)()

_ = session.execute('CREATE TABLE my_table (user_id int, setting_id int, settings string)')
for id in xrange(200):
    _ = session.execute('INSERT INTO my_table (user_id, setting_id) VALUES (:user_id, :setting_id)',
        {'user_id':id, 'setting_id':id})

_ = session.execute(
"""
    UPDATE my_table
    SET settings = :mysettings
    WHERE user_id = :user_id AND setting_id = 123
""", {'user_id':123, 'mysettings':'test'})

r = session.execute('SELECT * FROM my_table WHERE user_id = :user_id', {'user_id':123}).fetchall()
print r

[(123, 123, u'test')]

请注意，这并不是使用 sqlalchemy 的最佳方式。它旨在创建一个干净的环境，与特定的数据库后端解耦。但是，您可能有使用原始 SQL 而不是 ORM 的原因。