python - Flask-cache 为 int 和 long 参数生成不同的键

Question

Flask-cache 使用函数参数来生成缓存键，但是它为 long 和 int 类型的参数生成不同的键：

@cache.memoize(3600)
def foo(a):
    return a

foo(1)并且foo(1L)会生成不同的缓存键，我该怎么做才能将它们的返回值分配给相同的缓存键？

Answer 1

您可以按子类将整数转换为长整数。

例如。

class CustomCache(Cache):
    def _memoize_kwargs_to_args(self, f, *args, **kwargs):
        keyargs, keykwargs = super(CardCache, self) \
            ._memoize_kwargs_to_args(f, *args, **kwargs)

        new_args = []
        for arg in keyargs:
            if isinstance(arg, numbers.Integral):
                arg = long(arg)
            new_args.append(arg)

        return tuple(new_args), keykwargs