3

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

由于我没有计划为要获取初始 ~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() 调用仅适用于“引擎”，而不适用于“会话”，因此效果很好：

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

2 回答 2

4

好吧，这很奇怪，根据文档，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。

于 2012-09-18T20:52:07.410 回答
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格式。您需要查看数据库适配器文档以了解支持的内容。

于 2012-09-18T19:52:29.797 回答