如何禁用peewee的自动连接?

时间:2019-06-21 04:20:26

标签: python peewee flask-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仅引发异常-而不是在不通知我的情况下自动连接到数据库。

1 个答案:

答案 0 :(得分:1)

首先,您的示例是假的和错误的。当您仅创建查询对象时,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中。将包含在下一个版本中。