如何在SQLAlChemy中加载SQLite3扩展?

我构建了一个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()或以其他方式成功加载扩展?


解决方案

经过一番搜索和测试,基于各种来源,这个方法适合我。关于代码质量,我只能说这么多。我把它贴出来只是因为它可能会对某人有帮助。如果发现问题,请毫不犹豫地发表意见。

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完全任意。

相关文章