如何在python中用相同的包装器包装许多函数

Question

我是 Python 新手，还没有找到明确的方法。

我有几个函数（可能有数百个），我想一遍又一遍地包装在相同的代码中。

try:
    OneOfTheOneHundredFunctions()

except MY_ERROR as error:
    print('Error: I caught an error')

我是否必须为每个OneOfTheOneHundredFunctions()指定包装器？ 在 C++ 中，我会用macro来做到这一点，python 中是否有类似的东西？

Answer 1

您可以通过它们的名称引用函数，因此您可以将它们收集在一些可迭代的

for index, fn in enumerate((  
    function1,
    function2,
    function3,
)):
    try:
        fn()
    except Exception as ex:
        print(f"ERROR: function {index} raised {repr(ex)}")

enumerate在这里只是方便获取元组中函数的索引，但是您可以将名称放入任何可迭代对象中，例如dict并命名函数（带一些注释？）

functions_dict = {"check A": fn1, "check B": fn2, ...}
for comment, function in functions_dict.items():
    ...

Answer 2

您基本上可以使用包装器创建一个函数，并传入一个指向您想要的其他函数的指针。 在包装器内，您将调用指向的函数。

def catch_error(my_function):
    try:
        my_function()
    except MY_ERROR as error:
        print('Error: I caught an error')

list_of_functions = [OneOfTheHundredFunctions, TwoOfTheHundredFunctions, ...]
for function in list_of_functions:
    catch_error(function)


# Calling functions out of order
catch_error(TwoOfTheHundredFunctions)
# Some more logic
catch_error(OneOfTheHundredFunctions)

Answer 3

函数和异常是 Python 中的一等成员，因此您可以简单地将它们传递给另一个执行 try/except 块的函数：

def try_to_run(function, error_to_catch, *args, **kwargs):
    try:
        return function(*args, **kwargs)
    except error_to_catch:
        print('Error: I caught an error')

def OneOfTheOneHundredFunctions():
    pass

MY_ERROR = ValueError

result = try_to_run(
    function=OneOfTheOneHundredFunctions, 
    error_to_catch=MY_ERROR,
)

请注意，传递给try_to_run任何其他参数/​​关键字参数都将传递给您包装 try/except 的实际函数（这就是*args和**kwargs的用途）。

Answer 4

如果您在单独的文件中有函数，或者我认为您可以将其分开。 然后您可以使用它从该模块/文件中获取函数列表：
如何列出 Python 模块中的所有函数？ 另一个答案相同，但我认为有一些描述。

 from inspect import getmembers, isfunction from my_project import my_module list_of_functions = getmembers(my_module, isfunction)

现在您可以遵循 OP 的想法。

就像@nanotek 说的：

def catch_error(my_function):
    try:
        my_function()
    except MY_ERROR as error:
        print('Error: I caught an error')

for function in list_of_functions:
    catch_error(function)