add python_context.py

Author: xianhu

README.md

@@ -37,6 +37,8 @@
 ### python_magic_methods: Python进阶: 实例讲解Python中的魔法函数(Magic Methods)
 
 ### python_restful_api.py: 利用Python和Flask快速开发RESTful API
+
+### python_context.py: With语句和上下文管理器ContextManager
 ===================================================================================================
 
 ### 您可以fork该项目, 并在修改后提交Pull request

python_context.py

@@ -0,0 +1,78 @@
+# _*_ coding: utf-8 _*_
+
+"""
+python_context.py by xianhu
+"""
+
+import contextlib
+
+
+# 自定义打开文件操作
+class MyOpen(object):
+
+ def __init__(self, file_name):
+ """初始化方法"""
+ self.file_name = file_name
+ self.file_handler = None
+ return
+
+ def __enter__(self):
+ """enter方法，返回file_handler"""
+ print("enter:", self.file_name)
+ self.file_handler = open(self.file_name, "r")
+ return self.file_handler
+
+ def __exit__(self, exc_type, exc_val, exc_tb):
+ """exit方法，关闭文件并返回True"""
+ print("exit:", exc_type, exc_val, exc_tb)
+ if self.file_handler:
+ self.file_handler.close()
+ return True
+
+# 使用实例
+with MyOpen("python_base.py") as file_in:
+ for line in file_in:
+ print(line)
+ raise ZeroDivisionError
+# 代码块中主动引发一个除零异常，但整个程序不会引发异常
+
+
+# 内置库contextlib的使用
+@contextlib.contextmanager
+def open_func(file_name):
+ # __enter__方法
+ print("open file:", file_name, "in __enter__")
+ file_handler = open(file_name, "r")
+
+ yield file_handler
+
+ # __exit__方法
+ print("close file:", file_name, "in __exit__")
+ file_handler.close()
+ return
+
+# 使用实例
+with open_func("python_base.py") as file_in:
+ for line in file_in:
+ print(line)
+ break
+
+
+# 内置库contextlib的使用
+class MyOpen2(object):
+
+ def __init__(self, file_name):
+ """初始化方法"""
+ self.file_handler = open(file_name, "r")
+ return
+
+ def close(self):
+ """关闭文件，会被自动调用"""
+ print("call close in MyOpen2")
+ if self.file_handler:
+ self.file_handler.close()
+ return
+
+# 使用实例
+with contextlib.closing(MyOpen2("python_base.py")) as file_in:
+ pass