peewee - 如何在 Peewee 中动态设置 SQLite 数据库文件？

Question

我正在将 Peewee 用于我正在进行的一个项目，并且我正在尝试弄清楚如何动态设置数据库，以便我可以使用一个用于生产，一个用于测试。我见过的所有示例在任何类之外都有以下行:

database = SqliteDatabase(DATABASE)

我觉得很奇怪，因为我认为你会希望它位于一个类中，这样你就可以传递不同的数据库路径。对于选择一个用于生产的数据库和另一个用于测试的数据库有什么建议吗？

Answer 1

我刚刚遇到了类似的问题，以下是我解决该问题以在运行时定义数据库路径的方法:

模型文件:

import peewee

database = peewee.SqliteDatabase(None)  # Defer initialization

class SomeData(peewee.Model):
    somefield = peewee.CharField()

    class Meta:
        database = database

然后在使用数据库的类中:

from models import SomeData

class DatabaseUser:
    def __init__(self, db_path):
        database.init(db_path)