针对不同数据库后端的 peewee 数据库的运行时初始化

Question

我正在尝试编写一个（在某种程度上）独立于数据库的模块。我想定义我的 peewee 模型，然后使用init数据库对象的方法在运行时连接到数据库。

当我传递 Sqlite 连接字符串时，它按预期工作，例如

>>> app.db.init('sqlite://mydb.sqlite')

连接到数据库，一切都按预期进行。但是当我尝试使用 postgres 连接字符串进行相同操作时，我收到错误；

>>> app.db.init('postgresql://username:password@localhost/mydb')
...
peewee.OperationalError: FATAL:  database "postgresql://username:password@localhost/mydb" does not exist 

如果我使用单独的参数，我可以获得init连接的方法；

>>> app.db.init('mydb', username='username', password='password')

但这在不同的数据库后端之间不能很好地转换。

init谁能指出我使用连接 URI的正确方向？

Answer 1

在与 Peewee 作者进行一些互动后，答案似乎是使用该Proxy对象 - http://docs.peewee-orm.com/en/latest/peewee/database.html#dynamically-defining-a-database

在我的模型模块中我这样做；

database_proxy = pw.Proxy()
...
class Blah(pw.Model):
    column_one = pw.CharField()

    class Meta:
        database = database_proxy

然后在运行时连接；

>>> from playhouse.db_url import connect
>>> import models
>>> db = connect('postgresql://username:password@localhost/mydb')
>>> models.database_proxy.initialize(db)

然后我就可以正常与我的模型对象交互。这样我就可以在不同的数据库后端之间切换，并且只需在我的应用程序配置中使用连接 URL 字符串即可在它们之间进行切换。