python - 使用 klepto 通过函数名和参数进行缓存

Question

我正在尝试使用 klepto 作为缓存，它会散列 args 和函数名称，这可能吗？

例如，所以使用 dir_cache 我将能够

@inf_cache(cache=dir_archive(cached=False))
def func1(x, y):
    return x + y

@inf_cache(cache=dir_archive(cached=False))
def func2(x, y):
    return x - y

并且对 func1(1, 2) 和 func2(1, 2) 的调用都会在 dir_archive 中产生单独的键

我错过了什么吗？

Answer 1

尽管可能不是一个完全强大的解决方案，但这似乎在某种程度上可以满足您的需求。

>>> import klepto 
>>> @klepto.inf_cache(cache=klepto.archives.dir_archive(cached=False))
... def func1(x, y, name='func1'):
...   return x+y
... 
>>> @klepto.inf_cache(cache=klepto.archives.dir_archive(cached=False))
... def func2(x, y, name='func2'):
...   return x-y
... 
>>> 
>>> func1(1,2)
3
>>> func1(1,3)
4
>>> func2(2,4)
-2
>>> func2(1,2)
-1
>>> func1.__cache__()
dir_archive('memo', {-8532897055064740991: 4, -8532897055063328358: 3, -960590732667544693: -1, -3289964007849195004: -2}, cached=False)
>>> 
>>> func1(1,2)        
3
>>> func1(1,2)
3
>>> func2(1,2)
-1
>>> 
>>> func1.__cache__() == func2.__cache__()
True

您会注意到它们dir_archive是相同的，并且这两个函数似乎都使用单独的缓存。一个问题是，您可以传入 a'name'来覆盖默认值，并且很容易搞砸。我猜如果需要使用另一个装饰器来防止用户更改'name'关键字，您可以做一些更强大的事情。