假设我想使用以下语法打开一个文本文件进行阅读：
with open(fname,'r') as f:
# do something
pass
但如果我发现它以.gz结尾,我会调用gzip.open().
if fname.endswith('.gz'):
with gzip.open(fname,'rt') as f:
# do something
pass
else:
with open(fname,'r') as f:
# do something
pass
如果“做某事”部分很长并且不方便在函数中写入(例如,它会创建一个无法序列化的嵌套函数),使用gzip.open或基于返回的打开调用的最短方法是什么？ fname.endswith( ‘广州’)？
解决方法
上下文管理器帮助关闭对象.
但是,您不必创建用作上下文管理器的对象,同时使用它来输入上下文. open()和gzip.open()调用返回一个恰好是上下文管理器的新对象,您可以在输入上下文之前创建它们：
if fname.endswith('.gz'):
f = gzip.open(fname,'rt')
else:
f = open(fname,'r')
with f:
# do something
在这两种情况下,对象在进入上下文时返回self,因此这里不需要使用f作为f.
此外,函数是一等公民,因此您还可以使用变量来存储函数,然后在with语句中调用它来创建上下文管理器和文件对象：
if fname.endswith('.gz'):
opener = gzip.open
else:
opener = open
with opener(fname,'rt') as f: # yes,both open and gzip.open support mode='rt'
# do something
这并没有真正为你带来任何其他方法,但你可以使用字典将扩展名映射到callables,如果你愿意的话.
底线是calls context-manager hook methods,没有更少,仅此而已.应用之后的表达式应该提供这样的管理器,但是创建该对象不受上下文管理协议的约束.