python - python 中的上下文管理器和辅助函数

Question

我有一些使用上下文管理器的函数:

def f1():
    with open("test.txt","r+") as f:
        f.write("common Line")
        f.write("f1 Line")

def f2():
    with open("test.txt","r+") as f:
        f.write("common Line")
        f.write("f2 Line")

def f3():
    with open("test.txt","r+") as f:
        f.write("common Line")
        f.write("f3 Line")

这些函数有一些共同点。所以我想添加一个辅助功能。像这样的事情

def helperF():
    with open("test.txt","r+") as f:
        f.write("common Line")

然后以某种方式从我的 f1、f2、f3 函数中调用它，使代码变得干燥。

但我不太确定在这种情况下如何处理上下文管理器。以下内容将不起作用，因为在调用函数时 f 已经关闭:

def f1():
    commonHelper()
    f.write("f1 Line")

def f2():
    commonHelper()
    f.write("f2 Line")

def f3():
    commonHelper()
    f.write("f3 Line")

Answer 1

如果这三个函数各自向文件写入相当多的内容，我建议进行重构，以便它们返回要写入的字符串列表，而不是使用多个直接写入文件的函数。

def write_with_common_header(lines):
    with open("test.txt", "r+") as f:
        f.write("common Line")
        for line in lines:
            f.write(line)

def f1():
    return ["f1 Line"]

def f2():
    return ["f2 Line"]

def f3():
    return ["f3 Line"]

# usage example:
write_with_common_header(f2())

如果每个函数返回的列表始终相同，那么它们甚至不需要是函数；您可以将它们声明为列表。

<小时/>

在更一般的情况下，上下文管理器不一定是文件，并且单独的函数不仅仅是调用单个方法，那么我们不能只将它们作为数据传递，但相同的技术可以应用:使 write_with_common_header 函数接受参数，以便可以参数化其行为。为了完全通用，参数应该是一个接受托管资源 f 引用的函数。

def common_helper(callback):
    with open("test.txt", "r+") as f:
        f.write("common Line")
        callback(f)

def f1(f):
    f.write("f1 Line")

def f2(f):
    f.write("f2 Line")

def f3(f):
    f.write("f3 Line")

# usage example: 
common_helper(f2)