python - 如何在 SQLAlchemy 中加载 SQLite3 扩展？

Question

我构建了一个 SQLite 扩展(即 .so 库)，我想使用 SQLAlchemy 在我的应用程序中使用它。这是一个 Flask 应用程序，但我认为 Flask 在这里不起作用。

该扩展可以从 CLI 加载并且似乎可以工作:

$ sqlite3

SQLite version 3.20.1 2017-08-24 16:21:36
Enter ".help" for usage hints.
Connected to a transient in-memory database.
Use ".open FILENAME" to reopen on a persistent database.
sqlite> .load ./libSqliteIcu.so

但我需要在我的应用程序中做到这一点。 Python 文档中有一个示例:

import sqlite3

con = sqlite3.connect(":memory:")

# enable extension loading
con.enable_load_extension(True)

# Load the fulltext search extension
con.execute("select load_extension('./fts3.so')")

但是在我的应用程序中，我必须使用 db 访问数据库:

db = flask_sqlalchemy.SQLAlchemy()

我可以将最后一条语句改写为:

db.session.execute('select load_extension("./libsqliteicu.so")')

但它因“未授权”错误而失败。

怎么打电话 enable_load_extension()或以其他方式成功加载扩展？

Answer 1

经过一些搜索和测试，并基于各种来源，这对我有用。关于代码质量，我只能说这么多。我发布它只是因为它可能对某人有所帮助。如果您发现问题，请不要犹豫发表评论。

from sqlalchemy.event import listen

# initialization routine
# app: this Flask application
# db: the database, see the question 
db_collate = 'sk_SK.UTF-8'   # Slovak language for example
def load_extension(dbapi_conn, unused):
    dbapi_conn.enable_load_extension(True)
    dbapi_conn.load_extension('/path/to/libSqliteIcu.so')
    dbapi_conn.enable_load_extension(False)
    dbapi_conn.execute("SELECT icu_load_collation(?, 'ICU_EXT_1')", (db_collate,))
with app.app_context():
    listen(db.engine, 'connect', load_extension)

和用法:

from sqlalchemy.sql.expression import collate

...query.order_by(collate(Table.column, 'ICU_EXT_1'))

姓名 ICU_EXT_1是完全任意的。