python-3.x - 如何使用 @functools.lru_cache 装饰器绕过函数

Question

希望你一切顺利！

我正在使用一个利用 functools 库的 lru_cache 的函数。例如：

@functools.lru_cache(maxsize=pow(2,13))
def get_id(word):
    # retrieving id, using cache if possible

在某些情况下，我想绕过缓存，从而直接从源获取 ID，但我不想创建运行完全相同代码的 2 个单独的函数（一个有缓存，另一个没有）。在 docs.python 上阅读 functools 的文档，我了解到可以绕过缓存：

原始底层函数可以通过Wrapped 属性访问。这对于自省、绕过缓存或使用不同的缓存重新包装函数很有用。

我尝试使用包装函数这样做，但由于内部函数仅在外部函数运行时存在，因此每次调用时都会重置缓存。

我将不胜感激有关此事的任何帮助。

标签： python-3.xcaching

Answer 1

文档告诉您的是，您可以通过这种方式直接访问包装函数（绕过缓存）：

get_id.__wrapped__(word="hello")

---

您可以添加一个带有标志的附加层：

from functools import lru_cache

@lru_cache(maxsize=pow(2, 13))
def get_cached(word):
    return get_raw(word)

def get_raw(word):
    # your logic goes here...
    pass

def get_id(word, cached=True):
    if cached:
        return get_cached(word)
    else:
        return get_raw(word)