python3函数读取文件写入文件默认覆盖文件

Question

我想创建一个函数，读取txt文件，删除每行的前导空格和尾随空格，然后写入文件，默认情况下会覆盖我读取的文件，但可以选择写入新文件。 这是我的代码。

def cleanfile(inputfile, outputfile = inputfile):
    file1 = open(inputfile,'r')
    file2 = open(outputfile, 'w')
    lines = list(file1)
    newlines = map(lambda x: x.strip(), lines)
    newlines = list(newlines)
    for i in range(len(newlines)):
        file2.write(newlines[i] + '\n')
    file1.close()
    file2.close()    
cleanfile('hw.txt',)
cleanfile('hw.txt','hw_2.txt')

但这给了我错误。 NameError：名称“ inputfile”未定义

请问如何解决这个问题并实现我的目标？ 非常感谢你。

Answer 1

Python中的标准约定是使用None作为默认值并进行检查。

def cleanfile(inputfile, outputfile = None):
    if outputfile is None:
        outputfile = inputfile
    file1 = open(inputfile,'r')
    file2 = open(outputfile, 'w')
    lines = list(file1)
    newlines = map(lambda x: x.strip(), lines)
    newlines = list(newlines)
    for i in range(len(newlines)):
        file2.write(newlines[i] + '\n')
    file1.close()
    file2.close()    
cleanfile('hw.txt',)
cleanfile('hw.txt','hw_2.txt')

Answer 2

您不能将outputfile = inputfile设置为默认参数。 这是Python的局限性-指定默认参数后，“ inputfile”作为变量不存在。

您可以使用哨兵值：

sentinel = object()
def func(argA, argB=sentinel):
    if argB is sentinel:
       argB = argA
    print (argA, argB)

func("bar")           # Prints 'bar bar'
func("bar", None)     # Prints 'bar None'