我需要能够确定 Django 运行时使用的数据库类型。
MYSQL = False
if <something goes here>:
MYSQL = True
Run Code Online (Sandbox Code Playgroud)
我看到了相关的 Django 票证,但并没有完全找到我的解决方案。(https://code.djangoproject.com/ticket/18332)
如果有一个解决方案,我可以确定它是否是 MySQL、PostgreSQL、SQLite、Oracle...等等,那就太好了。
我尝试过挖掘django.db,但看不到任何有帮助的东西。
我将详细说明我正在尝试做的事情,因为它似乎被误解了。这是一个独立于任何 Django 项目的测试实用程序项目。我的实用程序需要了解正在使用哪个数据库来防止非法活动。
例如,我的包支持 PostgreSQL hstore 字段,但在 SQLite 中不存在这样的字段。因此,如果他们使用 SQLite,我不想允许特定功能。
我可以访问他们与该实用程序一起使用的 Django 模型。
from my_app.models import Foo
testing_utility(Foo, **kwargs)
Run Code Online (Sandbox Code Playgroud)
我的testing_utility签名是:
def testing_utility(klass, **kwargs):
instance = klass(**kwargs)
Run Code Online (Sandbox Code Playgroud)
hstore现在,如果不是 PostgreSQL 数据库,我不想在此类上执行特定功能。这是否更清楚一些?
编辑1:
谢谢 KevinDTimm 提供的链接。看起来 Django 模型._state.db在实例化时确实有。当然,我的实例有一个None类型。:(
>> print(instance._state.db)
None
Run Code Online (Sandbox Code Playgroud)
KevinDTimm 引导我迈出了这个解决方案的第一步。谢谢!
/sf/answers/1284074361/建议使用模型的._state.db参数来确定用于实例化的数据库。
但是,如果正在使用的数据库是,settings.DATABASES['default']则为。._state.dbNone
因此,我能够通过以下方式解决我的问题:
db_name = 'default'
if instance._state.db is not None:
db_name = instance._state.db
db_backend = settings.DATABASES[db_name]['ENGINE'].split('.')[-1]
MYSQL = db_backend == 'mysql'
Run Code Online (Sandbox Code Playgroud)
| 归档时间: |
|
| 查看次数: |
2461 次 |
| 最近记录: |