У меня есть sqlalchemy core bulk update query, который мне нужно программно передать имя столбца, который должен быть обновлен.Как передать имя столбца в качестве параметра в SQLAlchemy Core?
Функция выглядит, как показано ниже с комментариями по каждой переменной:
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, чтобы его распознать?
EDIT: Окончательный скрипт
Благодаря @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()
Как мне передать запрос 'bindparam (b_column_to_mark)' в запрос, поскольку для массового обновления требуется этот параметр привязки? – lukik
Второй вариант - это больше того, что я хочу, поскольку я видел, что 'getattr' выбирает имя столбца. Тем не менее, когда я запускаю его, я получаю сообщение об ошибке «AttributeError: объект« BinaryExpression »и объект« Comparator »имеют атрибут« items » ' – lukik
Метод 'values' использует именованные аргументы типа'.values (column_to_update = value) 'где' column_to_update' - это фактическое имя столбца, а не переменная, содержащая имя столбца. Если вам нужно, чтобы он был динамическим, используйте нотацию '** kwargs':' .values (** {'column_to_update': значение}) ' –