输入/输出装饰器到泡菜函数结果

Question

给定一个带有参数a 和其他两个参数（pickle_from、pickle_to）的函数，我想：

• 如果pickle_from 不是None，则加载并返回位于pickle_from 的腌制对象。如果是None，计算a的某个函数并返回。
• 如果pickle_to 不是None，则将上述结果转储到pickle_to。

只有一个函数，这很简单。如果pickle_from 不为空，该函数只会加载腌制结果并返回它。否则，它会使用a 执行一些耗时的计算，然后将其转储到pickle_to，并返回计算结果。

try:
   import cPickle as pickle
except:
   import pickle

def somefunc(a, pickle_from=None, pickle_to=None):

    if pickle_from:
        with open(pickle_from + '.pickle', 'rb') as f
            res = pickle.load(f)

    else:
        # Re-calcualte some time-intensive func call
        res = a ** 2

    if pickle_to:
        # Update pickled data with newly calculated `res`
        with open(pickle_to + '.pickle', 'wb') as f:
            pickle.dump(res, f)

    return res

我的问题是关于如何构建一个装饰器，以便这个过程可以围绕多个类似于somefunc的函数形成一个外壳，从而减少过程中的源代码。

我希望能够写出类似的东西：

@pickle_option
def somefunc(a, pickle_from=None, pickle_to=None)  
    # or do params need to be in the decorator call?
    # remember, "the files are in the computer"
    res = a ** 2
    return res

这可能吗？关于装饰器的一些事情让我头疼，所以我会礼貌地拒绝在这里发布“我尝试过的东西”。

Answer 1

这个装饰器需要一点内省。具体来说，我使用inspect.Signature 来提取pickle_from 和pickle_to 参数。

除此之外，它是一个非常简单的装饰器：它保留对被装饰函数的引用，并在必要时调用它。

import inspect
from functools import wraps

def pickle_option(func):
    sig = inspect.signature(func)

    @wraps(func)
    def wrapper(*args, **kwargs):
        # get the value of the pickle_from and pickle_to parameters
        # introspection magic, don't worry about it or read the docs
        bound_args = sig.bind(*args, **kwargs)
        pickle_from = bound_args.arguments.get('pickle_from', \
                             sig.parameters['pickle_from'].default)
        pickle_to = bound_args.arguments.get('pickle_to', \
                             sig.parameters['pickle_to'].default)

        if pickle_from:
            with open(pickle_from + '.pickle', 'rb') as f:
                result = pickle.load(f)
        else:
            result = func(*args, **kwargs)

        if pickle_to:
            with open(pickle_to + '.pickle', 'wb') as f:
                pickle.dump(result, f)

        return result

    return wrapper

Answer 2

鉴于您的用例，我认为只使用通用包装器会更清楚：

def pickle_call(fun, *args, pickle_from=None, pickle_to=None, **kwargs):
    if pickle_from:
        with open(pickle_from + '.pickle', 'rb') as f
            res = pickle.load(f)
    else:
        res = fun(*args, **kwargs)
    if pickle_to:
        # Update pickled data with newly calculated `res`
        with open(pickle_to + '.pickle', 'wb') as f:
            pickle.dump(res, f)
    return res

然后你会像这样使用它：

res = pickle_call(somefunc, a, pickle_from="from", pickle_to="to")

这避免了必须在您想使用此功能的任何地方添加装饰器，并且实际上适用于任何可调用（不仅仅是函数），来自您的代码或其他。