python with語句上下文管理的兩種實現方法

在編程中會常常碰到這種狀況：有一個特殊的語句塊，在執行這個語句塊以前須要先執行一些準備動做；當語句塊執行完成後，須要繼續執行一些收尾動做。例如，文件讀寫後須要關閉，數據庫讀寫完畢須要關閉鏈接，資源的加鎖和解鎖等狀況。
對於這種狀況python提供了上下文管理器（Context Manager）的概念，能夠經過上下文管理器來定義/控制代碼塊執行前的準備動做，以及執行後的收尾動做。

1、爲什麼使用上下文管理器

一、不使用上下文管理器的狀況
經過try...finally語句執行異常處理和關閉句柄的動做。

logger = open("log.txt", "w")
try:
　　logger.write('Hello ')
　　logger.write('World')
finally:
　　logger.close()

print logger.closed

 

 

二、使用上下文管理器
默認文件Python的內置file類型是支持上下文管理協議的。
使用上下文管理器with使得依據精簡了不少。

with open("log.txt", "w") as logger:
　　logger.write('Hello ')
　　logger.write('World')

print logger.closed

 

 

2、實現上下文管理器實現上下文管理器有兩種方式實現。方法一：類實現__enter__和__exit__方法。方法二：contextlib模塊裝飾器和生成器實現。
下面咱們經過兩種方法分別實現一個自定義的上下文管理器。
一、方法一：經過類實現__enter__和__exit__方法

class File(object):
    def __init__(self, file_name, method):
        self.file_obj = open(file_name, method)
    def __enter__(self):
        return self.file_obj
    def __exit__(self, type, value, traceback):
        self.file_obj.close()

with File('demo.txt', 'w') as opened_file:
    opened_file.write('Hola!')

 

實現__enter__和__exit__方法後，就能經過with語句進行上下文管理。

a、底層都發生了什麼？

1、with語句先暫存了File類的__exit__方法，而後它調用File類的__enter__方法。
二、__enter__方法打開文件並返回給with語句，打開的文件句柄被傳遞給opened_file參數。
三、with語句調用以前暫存的__exit__方法，__exit__方法關閉了文件。

 

b、異常處理

關於異常處理，with語句會採起哪些步驟。
1. 它把異常的type,value和traceback傳遞給__exit__方法
2. 它讓__exit__方法來處理異常 
3. 若是__exit__返回的是True，那麼這個異常就被忽略。
4. 若是__exit__返回的是True之外的任何東西，那麼這個異常將被with語句拋出。

 

異常拋出

#異常拋出，_exit__返回的是True之外的任何東西，那麼這個異常將被with語句拋出
class File(object):
    def __init__(self, file_name, method):
        self.file_obj = open(file_name, method)
    def __enter__(self):
        return self.file_obj
    def __exit__(self, type, value, traceback):
        self.file_obj.close()
        print "type:",type
        print "value:",value
        print "traceback:",traceback


with File('demo.txt', 'w') as opened_file:
    opened_file.undefined_function('Hola!')

#output================================================

# type: <type 'exceptions.AttributeError'>
# value: 'file' object has no attribute 'undefined_function'
# traceback: <traceback object at 0x000000000262D9C8>

#     opened_file.undefined_function('Hola!')
# AttributeError: 'file' object has no attribute 'undefined_function'

 

異常忽略：

#異常忽略，__exit__返回的是True，那麼這個異常就被忽略。
class File(object):
    def __init__(self, file_name, method):
        self.file_obj = open(file_name, method)
    def __enter__(self):
        return self.file_obj
    def __exit__(self, exception_type, exception_value, traceback):
        print("Exception has been handled")
        self.file_obj.close()
        return True


with File('demo.txt', 'w') as opened_file:
    opened_file.undefined_function('Hola!')
    
# output==================================
# Exception has been handled

 

 

二、方法二：contextlib模塊裝飾器和生成器實現

這種方式實現更優雅，我我的更喜歡這種方式。

yield以前的代碼由__enter__方法執行，yield以後的代碼由__exit__方法執行。本質上仍是__enter__和__exit__方法。

# coding:utf-8
import contextlib


@contextlib.contextmanager
def myopen(filename, mode):
    f = open(filename, mode)
    try:
        yield f.readlines()
    except Exception as e:
        print e

    finally:
        f.close()

if __name__ == '__main__':
    with myopen(r'c:\ip2.txt', 'r') as f:
        for line in f:
            print line

 

 三、with語句上多個下文關聯

直接經過一個with語句打開多個上下文，便可同時使用多個上下文變量，而沒必要需嵌套使用with語句。

class File(object):
    def __init__(self, file_name, method):
        self.file_obj = open(file_name, method)
    def __enter__(self):
        return self.file_obj
    def __exit__(self, exception_type, exception_value, traceback):
        self.file_obj.close()
        return True


with File('demo.txt', 'w') as f1,File('demo.txt','w') as f2:
    print f1,f2
    
# Output============================# <open file 'demo.txt', mode 'w' at 0x000000000263D150> <open file 'demo.txt', mode 'w' at 0x000000000263D1E0>