在django中启用sqlite完整性检查

Question

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

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

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

PRAGMA foreign_keys = 1;

然而，我不知道在哪里最好的方式来做这个激活(同样的问题在这里)。

而且，下面的代码不能工作:

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信号，监听postrongyncdb。

from django.db.models.signals import post_syncdb
def set_pragma_on(sender, **kwargs):
    "your code here"
post_syncdb.connect(set_pragma_on)

这确保无论何时syncdb运行(syncdb运行时，创建测试数据库)，您的SQLite数据库已将'pragma'设置为'on'。您应该检查在上面的方法'set_pragma_on'中使用的是哪个数据库。