sqalchemy update bindparam Первичный ключ - PullRequest
Новая
       24

sqalchemy update bindparam Первичный ключ

0 голосов
/ 02 ноября 2019

Следующий код выдает «sqlalchemy.exc.CompileError: Неполные имена столбцов: _id». 

User = Table('users', metadata,
 Column('id', Integer, primary_key=True),
 Column('score', Integer)
)

values = [
    {'score': 2, '_id': 1},
    {'score': 3, '_id': 3}
]
query = User.update().where(User.c.id == bindparam('_id')).values(score=bindparam('score'))
await db.execute_many(query, values)

db является экземпляром баз данных . Database. Обратите внимание, что у меня есть имя «_id», потому что SQLalchemy говорит, что «id» зарезервировано.

Есть ли какое-либо решение, кроме обновления каждой строки individullay?

1 Ответ

1 голос
/ 02 ноября 2019

Database.execute_many() вызывает Connection.execute_many(), который разбивает ваш запрос на отдельные отдельные запросы (по одному на элемент в values), вот метод ( source ): 

    async def execute_many(
        self, query: typing.Union[ClauseElement, str], values: list
    ) -> None:
        queries = [self._build_query(query, values_set) for values_set in values]
        async with self._query_lock:
            await self._connection.execute_many(queries)

Обратите внимание, что он вызывает метод _build_query() ( source ): 

    @staticmethod
    def _build_query(
        query: typing.Union[ClauseElement, str], values: dict = None
    ) -> ClauseElement:
        if isinstance(query, str):
            query = text(query)

            return query.bindparams(**values) if values is not None else query
        elif values:
            return query.values(**values)

        return query

Поскольку вы не передаете запрос str и передаете значения, элемент управления вводит elif values: обработка условий, при которой индивидуальный набор значений распаковывается в метод .values() в вашем запросе (то есть Update.values()). По сути, это делает запрос, который он пытается скомпилировать следующим образом: 

query = (
    User.update()
    .where(User.c.id == bindparam("_id"))
    .values(score=bindparam("score"))
    .values(score=2, _id=1)
)

Это второе предложение values ​​приводит к новому обновлению с новыми параметрами связывания, которые пытаются установить значения как для score, так и _id. Это приводит к ошибке компиляции запроса, поскольку в таблице нет столбца _id.

Таким образом, MCVE для воспроизведения ошибки действительно выглядит так: 

from sqlalchemy.dialects import postgresql

User.update().values(score=2, _id=1).compile(dialect=postgresql.dialect())

Что вызывает: 

Traceback (most recent call last):
  File ".\main.py", line 31, in <module>
    User.update().values(score=2, _id=1).compile(dialect=postgresql.dialect())
  File "<string>", line 1, in <lambda>
  File "C:\Users\peter\Documents\git\stackoverflow\58668615-sqalchemy-update-bindparam-primary-key\.venv\lib\site-packages\sqlalchemy\sql\elements.py", line 462, in compile
    return self._compiler(dialect, bind=bind, **kw)
  File "C:\Users\peter\Documents\git\stackoverflow\58668615-sqalchemy-update-bindparam-primary-key\.venv\lib\site-packages\sqlalchemy\sql\elements.py", line 468, in _compiler
    return dialect.statement_compiler(dialect, self, **kw)
  File "C:\Users\peter\Documents\git\stackoverflow\58668615-sqalchemy-update-bindparam-primary-key\.venv\lib\site-packages\sqlalchemy\sql\compiler.py", line 571, in __init__
    Compiled.__init__(self, dialect, statement, **kwargs)
  File "C:\Users\peter\Documents\git\stackoverflow\58668615-sqalchemy-update-bindparam-primary-key\.venv\lib\site-packages\sqlalchemy\sql\compiler.py", line 319, in __init__
    self.string = self.process(self.statement, **compile_kwargs)
  File "C:\Users\peter\Documents\git\stackoverflow\58668615-sqalchemy-update-bindparam-primary-key\.venv\lib\site-packages\sqlalchemy\sql\compiler.py", line 350, in process
    return obj._compiler_dispatch(self, **kwargs)
  File "C:\Users\peter\Documents\git\stackoverflow\58668615-sqalchemy-update-bindparam-primary-key\.venv\lib\site-packages\sqlalchemy\sql\visitors.py", line 92, in _compiler_dispatch
    return meth(self, **kw)
  File "C:\Users\peter\Documents\git\stackoverflow\58668615-sqalchemy-update-bindparam-primary-key\.venv\lib\site-packages\sqlalchemy\sql\compiler.py", line 2569, in visit_update
    self, update_stmt, crud.ISUPDATE, **kw
  File "C:\Users\peter\Documents\git\stackoverflow\58668615-sqalchemy-update-bindparam-primary-key\.venv\lib\site-packages\sqlalchemy\sql\crud.py", line 62, in _setup_crud_params
    return _get_crud_params(compiler, stmt, **kw)
  File "C:\Users\peter\Documents\git\stackoverflow\58668615-sqalchemy-update-bindparam-primary-key\.venv\lib\site-packages\sqlalchemy\sql\crud.py", line 177, in _get_crud_params
    % (", ".join("%s" % c for c in check))
sqlalchemy.exc.CompileError: Unconsumed column names: _id

Чтобы подвести итог проблемы, вы строите запрос с параметрами связывания, передаваемыми как Update.where(), так и Update.values(). Затем вы передаете этот запрос и ваши значения в Database.execute_many(), где они распаковывают отдельные элементы вашего списка значений во второй вызов Update.values() в вашем запросе, который заменяет ваш запрос на тот, который пытается установить значение для _idстолбец, который не существует.

Есть ли какое-либо решение, кроме обновления каждой строки individullay?

Что ж, запрос отлично работает как при использовании механизма sqlalchemy, так и запроса: 

# using a sqlalchemy engine
engine.execute(query, values)

В противном случае, должна работать отправка запроса в виде строки в Database.execute_many(), поскольку это будет означать, что запрос обрабатывается в if isinstance(query, str): части метода _build_query(), что позволит избежатьвторой .values() вызов по запросу: 

db.execute_many(str(query), values)
...