如何禁用peewee的自动连接?

艾伦·斯塔克

考虑以下代码

import peewee

SQL_CONN = peewee.MySQLDatabase(database=SQL_DATA,
                                host=SQL_HOST,
                                port=SQL_PORT,
                                user=SQL_USER,
                                passwd=SQL_PASS)

class User(peewee.Model):
    name = peewee.CharField(max_length=100, primary_key=True)

    born = peewee.DateTimeField()

print(SQL_CONN.is_closed()) # True

print(User.select().where(User.name == "Jack").execute()) # Silently opens a connection w/o letting me know

print(SQL_CONN.is_closed()) # False

这将SQL_CONN.connect()在后台自动执行

我如何禁用此功能-peewee如果未连接数据库则强制抛出异常-而不是在不通知我的情况下自动连接到该数据库。

鞘翅目

首先,您的示例是假的和错误的。当您仅创建查询对象时,Peewee不会打开连接。看看这个:

In [1]: from peewee import *                                                                    

In [2]: db = SqliteDatabase(':memory:')                                                         

In [3]: class User(Model): 
   ...:     username = TextField() 
   ...:     class Meta: 
   ...:         database = db 
   ...:                                                                                         

In [4]: db.is_closed()  # Should be True.                                                       
Out[4]: True

In [5]: query = User.select().where(User.username == 'charlie')                                 

In [6]: db.is_closed()  # Still TRUE, we haven't evaluated anything yet!
Out[6]: True

因此,首先,您的示例甚至都不正确。您必须评估查询才能执行。

要回答此问题的其余部分:

Peewee没有提供禁止隐式连接的机制。如果尝试执行查询,则Peewee将打开连接(如果该连接不存在)。

我建议您在执行查询时(因此在需要打开连接时)应该非常清楚。如果这还不够,请子类化并重写该Database.cursor()方法。


编辑:

即使以前没有人要求这样做,实现起来也是如此简单,我已经将此功能添加到了Peewee中。将包含在下一个版本中。

本文收集自互联网,转载请注明来源。

如有侵权,请联系 [email protected] 删除。

编辑于
0

我来说两句

0 条评论
登录 后参与评论

相关文章