Sqlite / SQLAlchemy: как применять внешние ключи?

Question

В новой версии SQLite предусмотрена возможность ограничения внешнего ключа, но для обеспечения обратной совместимости его необходимо включать для каждого соединения с базой данных отдельно!

sqlite> PRAGMA foreign_keys = ON;

Я использую SQLAlchemy - как я могу убедиться, что это всегда включено? Я попробовал вот что:

engine = sqlalchemy.create_engine('sqlite:///:memory:', echo=True)
engine.execute('pragma foreign_keys=on')

... но это не работает! ... Что мне не хватает?

EDIT: Я думаю, что моя настоящая проблема заключается в том, что у меня установлено более одной версии SQLite, а Python не использует последнюю версию!

>>> import sqlite3
>>> print sqlite3.sqlite_version
3.3.4

Но я только что скачал 3.6.23 и поместил exe в каталог моего проекта! Как я могу выяснить, какой .exe он использует, и изменить его?

Answer 1

Для последних версий (SQLAlchemy ~ 0.7) домашняя страница SQLAlchemy сообщает:

PoolListener устарел.Пожалуйста, обратитесь к PoolEvents .

Тогда пример CarlS становится:

engine = create_engine(database_url)

def _fk_pragma_on_connect(dbapi_con, con_record):
    dbapi_con.execute('pragma foreign_keys=ON')

from sqlalchemy import event
event.listen(engine, 'connect', _fk_pragma_on_connect)

Answer 2

Опираясь на ответы от conny и shadowmatter, вот код, который проверит, используете ли вы SQLite3, прежде чем выдавать оператор PRAGMA:

from sqlalchemy import event
from sqlalchemy.engine import Engine
from sqlite3 import Connection as SQLite3Connection

@event.listens_for(Engine, "connect")
def _set_sqlite_pragma(dbapi_connection, connection_record):
    if isinstance(dbapi_connection, SQLite3Connection):
        cursor = dbapi_connection.cursor()
        cursor.execute("PRAGMA foreign_keys=ON;")
        cursor.close()

Answer 3

У меня теперь это работает:

Загрузите последние сборки sqlite и pysqlite2, как описано выше: убедитесь, что Python использует правильные версии во время выполнения.

import sqlite3   
import pysqlite2 
print sqlite3.sqlite_version   # should be 3.6.23.1
print pysqlite2.__path__       # eg C:\\Python26\\lib\\site-packages\\pysqlite2

Далее добавьте PoolListener:

from sqlalchemy.interfaces import PoolListener
class ForeignKeysListener(PoolListener):
    def connect(self, dbapi_con, con_record):
        db_cursor = dbapi_con.execute('pragma foreign_keys=ON')

engine = create_engine(database_url, listeners=[ForeignKeysListener()])

Тогда будьте осторожны, как вы проверяете, работают ли внешние ключи: у меня была некоторая путаница здесь. При использовании sqlalchemy ORM для добавления () вещей мой код импорта неявно обрабатывал связи с отношениями, поэтому никогда не мог завершиться неудачей. В этом мне помогло добавление nullable = False к некоторым операторам ForeignKey ().

Способ проверки поддержки внешнего ключа sqlalchemy sqlite - выполнить ручную вставку из декларативного класса ORM:

# example
ins = Coverage.__table__.insert().values(id = 99,
                                    description = 'Wrong',
                                    area = 42.0,
                                    wall_id = 99,  # invalid fkey id
                                    type_id = 99)  # invalid fkey_id
session.execute(ins) 

Здесь 'wall_id' и 'type_id' - это ForeignKey () и sqlite теперь правильно генерируют исключение, если пытаются подключить недействительные fkeys. Так что это работает! Если вы удалите слушателя, то sqlalchemy с радостью добавит неверные записи.

Я полагаю, что основная проблема может заключаться в множественных файлах sqlite3.dll (или .so).

Answer 4

Со страницы SQLite :

SQLite поддерживает синтаксис FOREIGN KEY при создании операторов CREATE для таблиц, однако по умолчанию эти ограничения не влияют на работу таблицы.

Проверка ограничений в SQLite имеет три предварительных условия:

• Должна использоваться как минимум версия 3.6.19 SQLite
• Библиотека SQLite должна быть скомпилирована без включенных символов SQLITE_OMIT_FOREIGN_KEY или SQLITE_OMIT_TRIGGER.
• Оператор PRAGMA foreign_keys = ON должен передаваться во все соединения перед использованием.

SQLAlchemy позволяет автоматически выдавать оператор PRAGMA для новых соединений с использованием событий:

from sqlalchemy.engine import Engine
from sqlalchemy import event

@event.listens_for(Engine, "connect")
def set_sqlite_pragma(dbapi_connection, connection_record):
    cursor = dbapi_connection.cursor()
    cursor.execute("PRAGMA foreign_keys=ON")
    cursor.close()

Answer 5

В качестве более простого подхода, если создание сеанса централизовано за вспомогательной функцией Python (вместо непосредственного предоставления механизма SQLA), вы можете просто выполнить "session.execute ('pragma foreign_keys = on')" перед возвратом вновь созданного сессия.

Подход с использованием прослушивателя пула необходим только в том случае, если произвольные части вашего приложения могут создавать сеансы SQLA для базы данных.

Answer 6

Раньше у меня была такая же проблема (скрипты с ограничениями внешних ключей проходили, но ограничения фактически не применялись движком sqlite); получил решение:

1. загрузка, сборка и установка последней версии sqlite здесь: sqlite-sqlite-amalgamation ; до этого у меня был sqlite 3.6.16 на моей машине с Ubuntu; который еще не поддерживал внешние ключи; должно быть 3.6.19 или выше, чтобы они работали.

2. установка последней версии pysqlite отсюда: pysqlite-2.6.0

после этого я начал получать исключения, когда ограничение внешнего ключа не удалось

надеюсь, это поможет, с уважением

Answer 7

Если вам нужно выполнить что-то для настройки на каждом соединении, используйте PoolListener .