我有一个(递归)函数,我想接受一个字符串或一个打开的文件对象。如果参数是一个字符串,那么该函数将打开一个文件并使用该文件对象。当我从函数返回时,似乎最好明确地关闭这个打开的文件对象,但前提是传入一个字符串。(想象一下当用户传入一个打开的文件对象并发现他们的文件对象在某处关闭时的惊讶).这是我目前正在使用的:
def read_file(f, param):
do_close = isinstance(f,basestring)
f = open(f, 'rb') if do_close else f
try:
info = f.read(4)
#check info here
if info == Info_I_Want(param):
return f.read(get_data(info))
else:
f.seek(goto_new_position(info))
return read_file(f,param)
except IKnowThisError:
return None
finally:
if do_close:
f.close()
您可以假设如果我没有找到我想要的信息,IKnowThisError 会在某个时候被引发。
这感觉很糟糕。有没有更好的办法?
请您参考如下方法:
为什么不用包装器包装递归函数以避免开销?
def read_file(f, param):
if isinstance(f, basestring):
with open(f, 'rb') as real_f:
return read_file2(real_f, param)
else:
return read_file2(real_f, param)
def read_file2(f, param):
# Now f should be a file object
...
