LearnPython/python_context.py

79 lines
1.8 KiB
Python
Raw Blame History

This file contains ambiguous Unicode characters!

This file contains ambiguous Unicode characters that may be confused with others in your current locale. If your use case is intentional and legitimate, you can safely ignore this warning. Use the Escape button to highlight these characters.

# _*_ 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