在django中使用sqlite启用完整性检查

Question

在我的django项目中，我使用mysql db进行生产，使用sqlite进行测试。

问题是，我的一些代码依赖于模型完整性检查。它在mysql中工作得很好，但在测试中执行相同的代码时不会抛出完整性错误。

我知道必须在sqlite中激活外键检查：

PRAGMA foreign_keys = 1;

然而，我不知道在哪里做这个激活(same question here)是最好的方式。

此外，以下代码将无法工作：

def test_method(self):
    from django.db import connection
    cursor = connection.cursor()
    cursor.execute('PRAGMA foreign_keys = ON')
    c = cursor.execute('PRAGMA foreign_keys')
    print c.fetchone()
    >>> (0,)

有什么想法吗？

Answer 1

所以，如果最终找到了正确的答案。我所要做的就是在我安装的一个应用程序的__init__.py文件中添加以下代码：

from django.db.backends.signals import connection_created


def activate_foreign_keys(sender, connection, **kwargs):
    """Enable integrity constraint with sqlite."""
    if connection.vendor == 'sqlite':
        cursor = connection.cursor()
        cursor.execute('PRAGMA foreign_keys = ON;')

connection_created.connect(activate_foreign_keys)

Answer 2

你可以使用django signals，监听post_syncdb。

from django.db.models.signals import post_syncdb

def set_pragma_on(sender, **kwargs):
    "your code here"

post_syncdb.connect(set_pragma_on)

这确保了无论何时运行同步数据库(在创建测试数据库时运行同步数据库)，您的SQLite数据库都已将“杂注”设置为“开”。你应该在上面的方法'set_pragma_on‘中检查你正在使用的数据库。