Question

我正在编写一个按顺序执行一组任务的程序。每个任务都是一个输出新文件的函数，但如果文件名已经存在，则不应执行任何给定的任务。我发现自己一遍又一遍地写这种代码：

task1_fname = task1()
# come up with filename for next task
task2_fname = "task2.txt"
if not os.path.isfile(task2_fname):
  # task2 might depend on task1's output, so it gets task1's filename
  task2(task1_fname)
task3_fname = "task3.txt"
if not os.path.isfile(task3_fname):
  task3(...)

基本思想是，如果文件存在（理想情况下非空），则不应执行生成此文件的任务。

什么是最好的Pythonic方式表达这一点，而不必每次都写os.path.isfile次电话？装饰者可以更简洁地表达这一点吗？或者类似的东西：

with task2(task1_fname):
  # continue to next task

任何想法？

Answer 1

你在找这样的东西吗？

def preserve_output(f):
    def wrap(input, output):
        if not os.path.isfile(output):
            f(input, output)
    return wrap

@preserve_output
def task1(input, output):
    ...

@preserve_output
def task2(input, output):
    ...

task1('input', 'output_1')
task2('output_1', 'output_2')
task3('output_2', 'output_3') etc

装饰器在Python中创建一系列连续的任务？

1 个答案: