如何使用 cachetools ttl_cache 忽略函数参数

Question

我正在利用 cachetools @ttl_cache 装饰器（不是 @cached）。我需要忽略缓存键中的一些参数。例如，。

@ttl_cache(maxsize=1024, ttl=600)
def my_func(foo, ignore_bar, ignore_baz):
    # do stuff

这样工作，我明白了：

>>> my_func("foo", "ignore_bar", "ignore_baz")         # cache miss
>>> my_func("foo", "ignore_bar", "ignore_baz")         # cache hit
>>> my_func("foo", "ignore_bar_bar", "ignore_baz_baz") # cache miss!

我需要什么：

>>> my_func("foo", "ignore_bar", "ignore_baz")         # cache miss
>>> my_func("foo", "ignore_bar", "ignore_baz")         # cache hit
>>> my_func("foo", "ignore_bar_bar", "ignore_baz_baz") # cache hit!!!!!

有没有办法使用@ttl_cache 获得它？

Answer 1

我没有使用过缓存工具，但出于兴趣，我查看了文档。显然，没有内置的方法。如果您真的需要此功能，我可以建议如下 hack：

class PackedArgs(tuple):
    def __hash__(self):
        return hash(self[0])

    def __eq__(self, other):
        if isinstance(other, self.__class__):
            return self[0] == other[0]
        else:
            return NotImplemented

def pack_args(func):
    def inner(packed_args):
        return func(*packed_args)
    return inner

def unpack_args(func):
    def inner(*args):
        return func(PackedArgs(args))
    return inner

@unpack_args
@ttl_cache(maxsize=1024, ttl=600)
@pack_args
def my_func(foo, ignore_bar, ignore_baz):
    # do stuff

本质上：将函数的参数“打包”到一个类似元组的对象中，该对象作为第 0 个元素进行哈希处理，并导致 @ttl_cache 的行为符合您的需要。然后，“解包”它们以恢复正常的界面，并避免在调用时实际上必须传递这一大参数。

请注意，这只是一个（非常老套的）概念，我根本没有测试过这段代码。它可能不会按原样工作。

顺便说一句，您的要求听起来很有趣。您可以将其作为功能请求提交给 cachetools 并链接此线程。我可以想象它被实现为类似于 key= kwarg lambda 的东西，类似于内置的 sort's。

Answer 2

嗯，这正是@cached 及其自定义key 参数的用途，所以我想知道您为什么要为此使用@ttl_cache：

from cachetools import TTLCache, cached
from cachetools.keys import hashkey

def mykey(foo, ignore_bar, ignore_baz):
    return hashkey(foo)

@cached(cache=TTLCache(maxsize=1024, ttl=600), key=mykey)
def my_func(foo, ignore_bar, ignore_baz):
    # do stuff