如何在Peewee的Api类中实现批量保存所有修改模型实例的方法？

当然可行！Peewee提供了事务支持和实例变更追踪的能力，我们可以轻松实现批量提交所有修改的功能。先帮你指出代码里的两个小问题：

1. Api.get_foo2_by_id方法里的查询条件写错了，应该是foo2.id == id而不是foo1.id == id；
2. 修改foo2实例时，原模型定义的字段是bar2，你写的foo2.bar是笔误，这两处先修正避免后续问题。

下面给你两种实用的实现方案：

方案一：利用事务包裹批量保存（简单直接） #

如果你希望手动管理要保存的实例，可以让save_changes接收实例列表，然后在事务中批量执行保存操作，确保原子性（要么全部成功，要么全部回滚）：

from peewee import SqliteDatabase  # 替换成你实际使用的数据库类型

# 初始化你的数据库连接
db = SqliteDatabase('your_database.db')

class foo1(Model):
    id = PrimaryKeyField()
    bar = TextField(null=True)
    
    class Meta:
        database = db

class foo2(Model):
    id = PrimaryKeyField()
    bar2 = TextField(null=True)
    
    class Meta:
        database = db

class Api:
    def get_foo1_by_id(self, id):
        return foo1.get(foo1.id == id)
    
    def get_foo2_by_id(self, id):
        return foo2.get(foo2.id == id)  # 修正查询条件
    
    def save_changes(self, instances):
        with db.atomic():  # 开启事务保证原子性
            for instance in instances:
                if instance.dirty:  # 只保存有变更的实例，避免多余操作
                    instance.save()

使用方式：

api = Api()
foo1_inst = api.get_foo1_by_id(1)
foo2_inst = api.get_foo2_by_id(1)

foo1_inst.bar = 'str'
foo2_inst.bar2 = 'str'  # 修正字段名

api.save_changes([foo1_inst, foo2_inst])

方案二：自动追踪Api加载的实例（更便捷） #

如果想让Api自动追踪所有通过它获取的实例，无需手动传入，可以给Api添加一个实例集合来维护加载的实例，然后在save_changes中自动处理变更：

class Api:
    def __init__(self):
        self._tracked_instances = set()  # 用集合避免重复追踪同一个实例
    
    def get_foo1_by_id(self, id):
        inst = foo1.get(foo1.id == id)
        self._tracked_instances.add(inst)
        return inst
    
    def get_foo2_by_id(self, id):
        inst = foo2.get(foo2.id == id)
        self._tracked_instances.add(inst)
        return inst
    
    def save_changes(self):
        with db.atomic():
            # 转成列表避免遍历过程中修改集合导致的异常
            for instance in list(self._tracked_instances):
                if instance.dirty:
                    instance.save()
                    # 可选：保存后从追踪集合移除，避免重复保存已提交的变更
                    self._tracked_instances.remove(instance)

使用方式更简洁：

api = Api()
foo1_inst = api.get_foo1_by_id(1)
foo2_inst = api.get_foo2_by_id(1)

foo1_inst.bar = 'str'
foo2_inst.bar2 = 'str'

api.save_changes()  # 无需传入实例，自动处理所有追踪到的变更实例

补充说明 #

• instance.dirty是Peewee Model实例的内置属性，用来判断实例是否有未保存的变更，用它可以过滤掉无需保存的实例；
• db.atomic()上下文管理器会自动处理事务的提交和回滚，如果任何一个保存操作失败，所有变更都会回滚，保证数据一致性；
• 确保你的Model类在Meta中正确指定了数据库连接，否则事务可能无法正常生效。

内容的提问来源于stack exchange，提问作者TorSen