我正在尝试编写一个(某种程度上)与数据库无关的模块。我想定义我的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吗?
答案 0 :(得分: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字符串,即可在它们之间切换。
答案 1 :(得分:0)
我想我已经想出办法来做到这一点。它有点笨重,所以如果有人有更好的建议,我会全力以赴。我正在创建一个connect
对象,然后将其属性传递给db.init
。喜欢这个
>>> from playhouse.db_url import connect
>>> import models
>>> db = connect('postgresql://username:password@localhost/mydb')
>>> models.db.init(db.database, **db.connect_kwargs)
这似乎适用于任何有效的Peewee后端。