python上下文管理的使用場景實例講解

Posted on by

1、上下文管理的使用場景

凡是要在代碼塊前後插入代碼的場景，這點和裝飾器類似。

資源管理類：申請和回收，包括打開文件、網絡連接、數據庫連接等；

權限驗證。

2、實例

>>> with Context():
...   raise Exception # 直接拋出異常
...
enter context
exit context
Traceback (most recent call last):
 File "/usr/local/python3/lib/python3.6/site-packages/IPython/core/interactiveshell.py", line 2862, in run_code
  exec(code_obj, self.user_global_ns, self.user_ns)
 File "<ipython-input-4-63ba5aff5acc>", line 2, in <module>
  raise Exception
Exception

知識點擴展：

python上下文管理器異常問題解決方法

異常實例

如果我們需要對異常做特殊處理，就可以在這個方法中實現自定義邏輯。

之所以 with 能夠自動關閉文件資源，就是因為內置的文件對象實現瞭上下文管理器協議，這個文件對象的 __enter__ 方法返回瞭文件句柄，並且在 __exit__ 中實現瞭文件資源的關閉，另外，當 with 語法塊內有異常發生時，會拋出異常給調用者。

class File:
 def __enter__(self):
 return file_obj
 def __exit__(self, exc_type, exc_value, exc_tb):
 # with 退出時釋放文件資源
 file_obj.close()
 # 如果 with 內有異常發生 拋出異常
 if exc_type is not None:
  raise exception

在__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):
 print("Exception has been handled")
 self.file_obj.close()
 return True
 
with File('demo.txt', 'w') as opened_file:
 opened_file.undefined_function()
 
# Output: Exception has been handled

到此這篇關於python上下文管理的使用場景實例講解的文章就介紹到這瞭,更多相關python上下文管理的使用場景內容請搜索WalkonNet以前的文章或繼續瀏覽下面的相關文章希望大傢以後多多支持WalkonNet！

推薦閱讀：