一尘不染

重复密钥更新时的SQLAlchemy

mysql

INSERT ... ON DUPLICATE KEY UPDATE在SQLAlchemy中有一种优雅的方法吗?我的意思是语法类似于inserter.insert().execute(list_of_dictionaries)


阅读 342

收藏
2020-05-17

共1个答案

一尘不染

ON DUPLICATE KEY UPDATE 发布适用于MySQL的1.2版

现在,仅在MySQL的SQLAlchemy中内置了此功能。somada141的以下答案是最佳解决方案:https
://stackoverflow.com/a/48373874/319066

ON DUPLICATE KEY UPDATE 在SQL语句中

如果希望生成的SQL实际上包含ON DUPLICATE KEY UPDATE,则最简单的方法涉及使用@compiles装饰器。

的代码(来自关于该主题的很好的螺纹连接的上书签交易)中的示例,可以发现在github

from sqlalchemy.ext.compiler import compiles
from sqlalchemy.sql.expression import Insert

@compiles(Insert)
def append_string(insert, compiler, **kw):
    s = compiler.visit_insert(insert, **kw)
    if 'append_string' in insert.kwargs:
        return s + " " + insert.kwargs['append_string']
    return s


my_connection.execute(my_table.insert(append_string = 'ON DUPLICATE KEY UPDATE foo=foo'), my_values)

但是请注意,在这种方法中,您必须手动创建append_string。您可能会更改append_string函数,以便它自动将插入字符串更改为带有“ ON
DUPLICATE KEY UPDATE”字符串的插入,但是由于懒惰,我在这里不打算这样做。

ON DUPLICATE KEY UPDATE ORM中的功能

SQLAlchemy的不提供到接口ON DUPLICATE KEY UPDATEMERGE或在其ORM层的任何其他类似的功能。但是,它具有
仅在所讨论的键是主键时session.merge()才能复制功能的功能。


session.merge(ModelObject)首先通过发送SELECT查询(或在本地查找)来检查是否存在具有相同主键值的行。如果是这样,它将在某处设置一个标志,指示ModelObject已经在数据库中,并且SQLAlchemy应该使用UPDATE查询。请注意,合并比这要复杂得多,但是它可以使用主键很好地复制功能。

但是,如果您想要ON DUPLICATE KEY UPDATE具有非主键(例如,另一个唯一键)的功能怎么办?不幸的是,SQLAlchemy没有任何此类功能。相反,您必须创建类似于Django的内容get_or_create()另一个StackOverflow答案涵盖了,为了方便起见,我将在此处粘贴其修改后的工作版本。

def get_or_create(session, model, defaults=None, **kwargs):
    instance = session.query(model).filter_by(**kwargs).first()
    if instance:
        return instance
    else:
        params = dict((k, v) for k, v in kwargs.iteritems() if not isinstance(v, ClauseElement))
        if defaults:
            params.update(defaults)
        instance = model(**params)
        return instance
2020-05-17