Question

我有一个sqlalchemy核心批量更新查询，我需要以编程方式传递要更新的列的名称。

该函数如下所示，每个变量都有注释：

def update_columns(table_name, pids, column_to_update):
    '''
    1. table_name: a string denoting the name of the table to be updated
    2. pid: a list of primary ids
    3. column_to_update: a string representing the name of the column that will be flagged. Sometimes the name can be is_processed or is_active and several more other columns. I thus need to pass the name as a parameter.
    '''
    for pid in pids:
        COL_DICT_UPDATE = {}
        COL_DICT_UPDATE['b_id'] = pid
        COL_DICT_UPDATE['b_column_to_update'] = True
        COL_LIST_UPDATE.append(COL_DICT_UPDATE)

    tbl = Table(table_name, meta, autoload=True, autoload_with=Engine)
    trans = CONN.begin()
    stmt = tbl.update().where(tbl.c.id == bindparam('b_id')).values(tbl.c.column_to_update==bindparam('b_column_to_update'))
    trans.commit()

table参数被接受并且正常工作。

作为参数传递时，column_to_update不起作用。它失败并显示错误raise AttributeError(key) AttributeError: column_to_mark。但是，如果我对列名进行硬编码，则运行查询。

如何为SQLAlchemy传递column_to_update的名称以识别它？

编辑：最终剧本

感谢@Paulo，最终的脚本如下所示：

def update_columns(table_name, pids, column_to_update):
    for pid in pids:
        COL_DICT_UPDATE = {}
        COL_DICT_UPDATE['b_id'] = pid
        COL_DICT_UPDATE['b_column_to_update'] = True
        COL_LIST_UPDATE.append(COL_DICT_UPDATE)

    tbl = Table(table_name, meta, autoload=True, autoload_with=Engine)         
    trans = CONN.begin()
    stmt = tbl.update().where(
                              tbl.c.id == bindparam('b_id')
                              ).values(**{column_to_update: bindparam('b_column_to_update')})
    CONN.execute(stmt, COL_LIST_UPDATE)
    trans.commit()

Answer 1

我不确定我是否理解你想要的东西，你的代码看起来与我认为惯用的sqlalchemy（我不批评，只是评论我们可能使用正交代码样式）非常不同。

如果要将文字列作为参数传递，请使用：

from sqlalchemy.sql import literal_column
...
tbl.update().where(
    tbl.c.id == bindparam('b_id')
).values(
    tbl.c.column_to_update == literal_column('b_column_to_update')
)

如果要动态设置表达式的右侧，请使用：

tbl.update().where(
    tbl.c.id == bindparam('b_id')
).values(
    getattr(tbl.c, 'column_to_update') == bindparam('b_column_to_update')
)

如果这不是您想要的，请对答案进行评论或改进您的问题，我会尽力提供帮助。

[更新]

values方法使用.values(column_to_update=value)之类的命名参数，其中column_to_update是实际的列名，而不是包含列名的变量。例如：

stmt = users.update().\
        where(users.c.id==5).\
        values(id=-5)

请注意，where使用比较运算符==而values使用归因运算符= - 前者使用布尔表达式中的列对象，后者使用列名作为关键字参数绑定。

如果您需要它是动态的，请使用**kwargs表示法：.values(**{'column_to_update': value})

但您可能希望使用values参数而不是values方法。

Answer 2

还有另一种简单的方法： tbl.c[column_name_here]

如何在SQLAlchemy Core中将列的名称作为参数传递？

2 个答案: