在 Django 中将固定装置加载到数据库后如何延迟评估 ORM 调用？

Question

我有一个模块 pagetypes.py 从数据库中提取几个常量（我不应该在这里真正使用单词常量）以供以后重用：

def _get_page_type_(type):
    return PageType.objects.get(type=type)

PAGE_TYPE_MAIN = _get_page_type_('Main')
PAGE_TYPE_OTHER = _get_page_type_('Other')

然后在视图中的某个地方我所做的：

import pagetypes
...
print pagetypes.PAGE_TYPE_MAIN #simplified

一切正常当数据库有这些记录并且我确保它有时就很好......除非此代码正在测试中。在这种情况下，我想通过固定装置将这些记录加载到数据库中。问题是，在导入 pagetypes 模块时，固定装置尚未加载（甚至 syncdb 未运行），导致 _get_page_type_ 调用失败，并显示：

psycopg2.ProgrammingError: relation "pagetype" does not exist

测试运行程序始终尝试导入 pagetypes 模块，因为它是由正在测试的视图导入的。

我该如何解决这个问题？

我正在考虑延迟加载页面类型常量PAGE_TYPE_MAIN、PAGE_TYPE_OTHER，但是如果这些记录不在数据库中（或者在测试时的固定装置中），我希望它尽早失败，所以我真的不知道如何实现这个。

我也在考虑对象级缓存，只要使用/调用常量就调用 PageType.objects.get(type=type) ，但这不是一种矫枉过正吗？在没有缓存的情况下调用 orm 会导致过多的数据库调用，这是我想防止的。

这一定是非常简单的事情，但我无法解决。 ;-)

Answer 1

我将使用函数而不是常量，但会记住它们：

_cache = {}

def get_page_type(type_name):
    if type_name not in _cache:
        _cache[type_name] = PageType.objects.get(type=type_name)
    return _cache[type_name]

所以现在您可以在必要时直接调用 get_page_type('Main') 。