使用Peewee库进行批量更新

时间:2018-08-30 11:55:55

标签: python mysql sql-update peewee

我正在尝试使用 Peewee 库更新表中的许多记录。在for循环中,我获取了一条记录,然后对其进行了更新,但这在性能方面听起来很糟糕,因此需要批量进行更新。当前代码如下:

usernames_to_update = get_target_usernames()
for username in usernames_to_update:
    user = User.get(User.username == username) # username is primary key
    if user.type == 'type_1':
        user.some_attr_1 = some_value_1
    elif user.type == 'type_2':
        user.some_attr_2 = some_value_2
    # elif ....
    user.save()

documentation中,有insert_many函数,但没有类似update_many的函数。我四处搜寻,提出了以下解决方案:

  1. 使用CASELink
  2. 执行原始查询
  3. 使用replace_manyLink
  4. 使用updateLink

但是我找不到如何使用第二种或第三种解决方案的任何示例。有人可以澄清情况2和3的用法吗?

2 个答案:

答案 0 :(得分:2)

您需要.update()方法:

query = User.update(validated=True).where(User.username.in_(usernames_to_update))
query.execute()

编辑:因此您希望在更新期间有条件地设置该值。您可以使用Case帮助器。未经测试:

some_value_1 = 'foo'
some_value_2 = 'bar'
case_stmt = Case(User.type, [
    ('type_1', some_value_1),
    ('type_2', some_value_2)])
query = User.update(some_field=case_stmt).where(User.username.in_(list_of_usernames))
query.execute()

文档可在此处找到:http://docs.peewee-orm.com/en/latest/peewee/api.html#Case

答案 1 :(得分:1)

新的最佳答案是使用here中发现的bulk_update()方法:

with database.atomic():
    User.bulk_update(user_list, fields=['username'], batch_size=50)