基于Peewee ORM的Python程序数据库OO抽象实现方法咨询

嘿，这个问题问到点子上了！在Peewee里用面向对象思路把数据库和业务代码解耦，同时把grow()、die()这类业务方法整合到模型里，其实有几个实用的方案，我给你一步步说：

首先，你原来的写法里每个Model类都要在Meta里指定database = db，这会导致数据库连接和模型高度耦合。最简单的抽象方式是定义一个基础模型基类，所有业务模型都继承它，把数据库配置统一放在基类里：

import peewee

# 先单独初始化数据库连接（可以放在配置文件里，这里简化）
db = peewee.SqliteDatabase('my_app.db')

# 定义基础模型，所有业务模型继承它
class BaseModel(peewee.Model):
    class Meta:
        database = db  # 统一指定数据库
        # 还可以加其他全局配置，比如表前缀、排序规则等

# 业务模型继承BaseModel，不用再重复写数据库配置
class Person(BaseModel):
    name = peewee.TextField()
    height = peewee.DecimalField()

    # 直接把业务方法写在模型里（Active Record模式）
    def grow(self, size):
        self.height += size
        self.save()  # Peewee自带的save方法，直接更新数据库

    def die(self):
        self.delete_instance()  # 调用Peewee的删除方法，移除这条记录

这种方式的好处是：如果后续要切换数据库（比如从SQLite换成PostgreSQL），只需要修改BaseModel里的数据库配置，所有业务模型自动生效；同时业务方法直接和数据模型绑定，符合你想要的面向对象设计。

如果你的业务逻辑比较复杂，不想把数据库查询、CRUD操作和业务方法混在一起，可以用Repository模式，把数据访问逻辑单独封装，让Model类专注于业务逻辑：

# 还是先定义基础模型和Person模型
class BaseModel(peewee.Model):
    class Meta:
        database = db

class Person(BaseModel):
    name = peewee.TextField()
    height = peewee.DecimalField()

    # 这里只放纯业务逻辑，不直接操作数据库
    def grow(self, size):
        self.height += size
        # 注意：这里不直接save，交给Repository处理

# 封装数据访问的Repository类
class PersonRepository:
    @staticmethod
    def save_person(person):
        person.save()

    @staticmethod
    def delete_person(person):
        person.delete_instance()

    @staticmethod
    def get_by_name(name):
        return Person.get_or_none(Person.name == name)

    @staticmethod
    def get_tall_people(min_height):
        return Person.select().where(Person.height >= min_height)

# 使用示例
person = Person(name="Bob", height=1.75)
PersonRepository.save_person(person)

person.grow(0.05)
PersonRepository.save_person(person)

PersonRepository.delete_person(person)

这种模式的优势是：数据访问逻辑集中在Repository里，便于统一维护和测试；Model类只负责封装业务状态和行为，关注点更清晰，适合中大型项目。

如果你的项目需要多环境切换（比如测试用内存数据库，生产用真实数据库），可以用依赖注入的方式动态给基类注入数据库连接：

class BaseModel(peewee.Model):
    class Meta:
        database = None  # 先留空，后续动态注入

class Person(BaseModel):
    name = peewee.TextField()
    height = peewee.DecimalField()

    def grow(self, size):
        self.height += size
        self.save()

# 初始化数据库的函数
def init_database(db_instance):
    BaseModel._meta.database = db_instance
    # 自动创建所有表（如果不存在）
    BaseModel.create_tables([Person], safe=True)

# 测试环境用内存数据库
test_db = peewee.SqliteDatabase(':memory:')
init_database(test_db)

# 生产环境用PostgreSQL
prod_db = peewee.PostgresqlDatabase(
    'my_prod_db',
    user='admin',
    password='secret',
    host='db.example.com'
)
init_database(prod_db)

这种方式让数据库配置完全和业务代码分离，你可以根据不同环境轻松切换数据库，非常适合需要自动化测试的项目。

• 小型项目/快速原型：直接用基类+Active Record模式，简单直观，代码量少；
• 中大型项目：用Repository模式分离数据访问和业务逻辑，代码更易维护和扩展；
• 需要多环境切换：加上依赖注入，让数据库配置更灵活。

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