python - SQLalchemy 0.7 的任何示例，更新使用 from_statement()

Question

我正在编写一个快速的一次性迁移脚本，用于更新具有 50 万行的表中的单个字段。

由于我没有计划为要获取初始 ~25000 行数据的连接写出完整模型，所以我一直在尝试弄清楚如何使用 from_statement() 执行 UPDATE 语句调用并使用我自己的原始 sql，但找不到任何示例。

与此同时，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() 调用仅适用于“引擎”，不适用于“ session ”，因此效果很好:

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

Answer 1

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

Execute the given SELECT statement and return results.

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

您可以只使用 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 相比。