python - 如何在 Django 中使用 Python 函数扩展 SQLite？

Question

有可能define new SQL functions for SQLite in Python .我如何在 Django 中执行此操作，以便函数随处可用？

一个示例用例是使用 GREATEST() and LEAST() 的查询PostgreSQL 函数，在 SQLite 中不可用。我的测试套件运行此查询，我希望能够在运行测试时使用 SQLite 作为数据库后端。

Answer 1

这是一个 Django 代码示例，它通过调用 Python 的内置 max() 和 min() 来使用 GREATEST() 和 LEAST() 方法扩展 SQLite:

from django.db.backends.signals import connection_created
from django.dispatch import receiver

@receiver(connection_created)
def extend_sqlite(connection=None, **kwargs):
    connection.connection.create_function("least", 2, min)
    connection.connection.create_function("greatest", 2, max)

我只在测试中需要它，所以我把它放在我的 test_settings.py 中。如果您在代码的其他地方有它，您可能需要测试 connection.vendor == "sqlite"。