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

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 而失败。但是，如果我对列名称进行硬编码，查询就会运行。

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

编辑:最终脚本

感谢@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 使用属性运算符 = -前者在 bool 表达式中使用列对象，后者使用列名作为关键字参数绑定(bind)。

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

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