python - 在peewee中设置严格的sql模式

Question

我想在 mySQL 数据库中强制使用 ENUM 约束。为此，建议我使用“严格”sql 模式。在交互式提示中，可以使用以下之一进行设置:

SET GLOBAL sql_mode = 'STRICT_ALL_TABLES';
SET SESSION sql_mode = 'STRICT_ALL_TABLES';

或者在服务器启动时使用:

 --sql-mode="STRICT_ALL_TABLES"

或者在 my.cnf 中为:

sql-mode="STRICT_ALL_TABLES"

有没有办法在 peewee 中做这样的事情？也许通过扩展Database 类？或者在查询中注入(inject)原始sql？

Answer 1

您最好的选择可能是子类化 MySQLDatabase 并重写 _connect() 方法，例如

class StrictMySQLDatabase(MySQLDatabase):
    def _connect(self, database, **kwargs):
        conn = super(StrictMySQLDatabase, self)._connect(database, **kwargs)
        cursor = conn.cursor()
        cursor.execute("SET SESSION sql_mode = 'STRICT_ALL_TABLES';")
        return conn