如何在DataSaver类内部为PeeWee的BaseModel传入数据库实例？

你遇到的核心问题是：Python的类定义是在模块加载阶段执行的，而self只有当类被实例化后才会存在。所以当你在DataSaver内部定义BaseModel时，它的Meta类试图引用self.db，此时这个self还没被创建出来，自然会报错。

下面提供两种可行的解决方案，分别适用于不同的场景：

方案一：动态创建模型类（支持多独立数据库实例） #

这种方法会在DataSaver实例化时，动态生成绑定当前数据库的模型类。每个DataSaver实例都有自己独立的模型，适合需要同时操作多个数据库的场景。

from peewee import *
import datetime

class DataSaver:
    def __init__(self, database_save_path):
        self.database_save_path = database_save_path
        self.db = SqliteDatabase(database_save_path)
        
        # 动态创建绑定当前数据库的BaseModel
        class BaseModel(Model):
            class Meta:
                database = self.db
        
        # 基于动态BaseModel创建具体业务模型
        self.User = type('User', (BaseModel,), {
            'name': CharField(unique=True),
            '__module__': __name__  # 必须设置，避免序列化/ORM操作异常
        })
        
        self.Chat = type('Chat', (BaseModel,), {
            'name': CharField(unique=True),
            '__module__': __name__
        })
        
        self.Message = type('Message', (BaseModel,), {
            'chat': ForeignKeyField(self.Chat, backref='messages'),
            'content': TextField(),
            'created_at': DateTimeField(default=datetime.datetime.now),
            '__module__': __name__
        })
        
        # 连接数据库并创建表（safe=True避免重复创建报错）
        self.db.connect()
        self.db.create_tables([self.User, self.Chat, self.Message], safe=True)
    
    def close(self):
        """关闭数据库连接"""
        if not self.db.is_closed():
            self.db.close()

# 使用示例
saver = DataSaver('my_chat_app.db')
# 操作模型：saver.User.create(name='Alice')
saver.close()

方案二：后期绑定数据库（单数据库场景更简洁） #

如果你的应用只需要操作一个数据库实例，可以先定义好模型结构，再在__init__中动态绑定数据库。这种写法更接近Peewee官方的风格，但要注意多实例会覆盖数据库绑定。

from peewee import *
import datetime

class DataSaver:
    # 先定义模型结构，数据库留空
    class BaseModel(Model):
        class Meta:
            database = None
    
    class User(BaseModel):
        name = CharField(unique=True)
    
    class Chat(BaseModel):
        name = CharField(unique=True)
    
    class Message(BaseModel):
        chat = ForeignKeyField(Chat, backref='messages')
        content = TextField()
        created_at = DateTimeField(default=datetime.datetime.now)
    
    def __init__(self, database_save_path):
        self.database_save_path = database_save_path
        self.db = SqliteDatabase(database_save_path)
        
        # 给所有BaseModel子类绑定当前数据库
        self.BaseModel._meta.database = self.db
        
        # 连接并创建表
        self.db.connect()
        self.db.create_tables([self.User, self.Chat, self.Message], safe=True)
    
    def close(self):
        if not self.db.is_closed():
            self.db.close()

关键注意事项 #

• 无论哪种方案，都要记得在使用完数据库后调用close()方法，或者用上下文管理器管理连接（可以给DataSaver添加__enter__和__exit__方法实现）。
• 方案一中的__module__属性必须设置，否则模型在序列化、迁移等场景下可能出现模块识别异常。
• 方案二不适合多数据库实例场景，因为所有DataSaver实例会共享同一个模型类，后面的实例会覆盖前面的数据库绑定。

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