如何在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
完全任意。
相关文章