概述
作为一个 Java 为母语的程序员来讲,学习起其他新的语言就难免任何事都与 Java 进行横向对比。Java 7 引入了能省去许多重复代码的 try-with-resources 特性,不用每回 try/finally 来释放资源(不便之处有局部变量必须声明在 try 之前,finally 里还要嵌套 try/catch 来处理异常)。比如下面的 Java 代码
try(InputStream inputStream = new FileInputStream("abc.txt")) { System.out.println(inputStream.read()); } catch (Exception ex) { }
它相应的不使用 try-with-resources 语法的代码就是
InputStream inputStream = null; try { inputStream = new FileInputStream("abc.txt"); } catch (Exception ex) { } finally { if(inputStream != null) { try { inputStream.close(); } catch (Exception ex) { } } }
类似的 Python 也有自己的 try-with-resources 写法,就是 with 关键字,它的概念叫做上下文管理器(Context Manager)。
with 关键字的使用
with open('some_file','w') as opened_file: opened_file.write('Hola!')
以上的代码相当于
opened_file = open('some_file','w') try: opened_file.write('Hola!') finally: opened_file.close()
也就是 with 关键字打开的资源会在 with 语句块结束后自动调用相应的方法自动释放(无论 with 中操作是否有异常)。
with 用起来是很方便的,但是什么样的资源可以用 with 关键字?Python 是怎么知道要调用哪个方法来关闭资源的?进而如何实现自己的支持上下文管理器的 Python 类。
再次回顾 Java 的 try-with-resources 语法, try(...) 括号支持的类必须是实现了 AutoCloseable 接口,它的接口方法是
进群:960410445 即可获取数十套PDF的!
public void close() throws IOException
也就是 Java 的 try-with-resources 语法会自动调用以上方法来释放资源,要实现可被自动释放的 Java 就只须遵照这一规则就行。
而在 Python 中,能被 with 的类有两种实现方式
一个 Python 类要能被用于 with 上下文,必须实现至少 __enter__ 和 __exit__ 方法。这两个方法的意思好理解,一个是创建资源后,后者是退出 with 语句块后。请看下面的例子
class File(object): def __init__(self,file_name,method): self.file_obj = open(file_name,method)def enter(self):
print("---enter")
return self.file_objdef exit(self,type,value,traceback):
print("---exit")
self.file_obj.close()with File('data.txt','r') as data_file:
print(data_file.read())
def enter(self):
print("---enter")
return self.file_obj
def exit(self,type,value,traceback):
print("---exit")
self.file_obj.close()
with File('data.txt','r') as data_file:
print(data_file.read())
def enter(self):
print("---enter")
return self.file_obj
def exit(self,type,value,traceback):
print("---exit")
self.file_obj.close()
with File('data.txt','r') as data_file:
print(data_file.read())
hello world
那么以上程序执行后的输出就是
--enter hello world ---exit
利用生成器(Generator) 创建支持上下文管理器的方法
此种方式比较简单,不过逻辑控制上没有这么强。
from contextlib import contextmanager@contextmanager
def open_file(name,method):
f = open(name,method)
yield f
f.close()
@contextmanager
def open_file(name,method):
f = open(name,method)
yield f
f.close()
@contextmanager
def open_file(name,method):
f = open(name,method)
yield f
f.close()
使用 f 的执行代码将被放置在 yield f 所处的位置, with 使用以上方法。 yield 后的 f 变量将是 with...as 后的变量值
with open_file('some_file','w') as file_object: file_object.write('hola!')
这里也要注意异常处理的情况,比如把上面代码打开文件的模式换作 r,仍然试图去写文件,这样在 open_file 方法的 yield f 位置将产生异常,会造成 f.close() 得不到执行,不能正确释放该资源。
欲更具防御性,前面的 yield f 可以扩展也如下的形式
try: yield f except Exception as ex: pass #处理异常,或继续向外抛 finally: f.close()
总结
以上是编程之家为你收集整理的Python 版的 try-with-resources——with 上下文管理器!全部内容,希望文章能够帮你解决Python 版的 try-with-resources——with 上下文管理器!所遇到的程序开发问题。
如果您也喜欢它,动动您的小指点个赞吧