Bật tính năng kiểm tra tính toàn vẹn với sqlite trong django

Question

Trong dự án django của tôi, tôi sử dụng db mysql cho sản xuất và sqlite để kiểm tra.

Vấn đề là, một số mã của tôi dựa vào kiểm tra tính toàn vẹn của mô hình. Nó hoạt động tốt với mysql, nhưng lỗi toàn vẹn không được ném khi cùng một mã được thực thi trong các thử nghiệm.

Tôi biết rằng các phím nước ngoài kiểm tra phải được kích hoạt trong sqlite:

PRAGMA foreign_keys = 1; 

Tuy nhiên, tôi không biết đâu là cách tốt nhất để làm kích hoạt này (same question here).

Hơn nữa, đoạn code sau sẽ không làm việc:

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,) 

Bất kỳ ý tưởng?

Answer 1

Vì vậy, nếu cuối cùng đã tìm thấy câu trả lời đúng. Tất cả những gì tôi phải làm là thêm mã này vào tệp __init__.py trong một trong các ứng dụng đã cài đặt của tôi:

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

Bạn có thể sử dụng tín hiệu django, nghe 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) 

Điều này đảm bảo rằng bất cứ khi nào syncdb chạy (syncdb được chạy, khi tạo cơ sở dữ liệu thử nghiệm), cơ sở dữ liệu SQLite của bạn đã đặt 'pragma' thành 'bật'. Bạn nên kiểm tra xem bạn đang sử dụng cơ sở dữ liệu nào trong phương thức 'set_pragma_on' ở trên.