Python裝飾器的應用場景代碼總結
更新時間：2020年04月10日 10:34:35 作者：臨淵
這篇文章主要介紹了Python裝飾器的應用場景,文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下
裝飾器的應用場景
- 附加功能
- 數(shù)據(jù)的清理或添加:
- 函數(shù)參數(shù)類型驗證 @require_ints 類似請求前攔截
- 數(shù)據(jù)格式轉換 將函數(shù)返回字典改為 JSON/YAML 類似響應后篡改
- 為函數(shù)提供額外的數(shù)據(jù) mock.patch
- 函數(shù)注冊
- 在任務中心注冊一個任務
- 注冊一個帶信號處理器的函數(shù)
不同應用場景下裝飾器實現(xiàn)
函數(shù)注冊表
簡單注冊表
funcs = [] def register(func): funcs.append(func) return func @register def a(): return 3 @register def b(): return 5 # 訪問結果 result = [func() for func in funcs]
注冊表隔離(使用類的不同實例)
class Registry(object):
def __init__(self):
self._funcs = []
def register(self, func):
self._funcs.append(func)
def run_all(self):
return [func() for func in self._funcs]
r1 = Registry()
r2 = Registry()
@r1.register
def a():
return 3
@r2.register
def b():
return 5
@r1.register
@r2.register
執(zhí)行時封裝代碼
類型檢查
from functools import wraps
def require_ints(func):
@wraps(func) # 將func的信息復制給inner
def inner(*args, **kwargs):
for arg list(args) + list(kwargs.values()):
if not isinstance(arg, int:
raise TypeError("{} 只接受int類型參數(shù)".format(func.__name__)
return func(*args, **kwargs)
return inner
用戶驗證
from functools import wraps
class User(object):
def __init__(self, username, email):
self.username = username
self.email = email
class AnonymousUser(object):
def __init__(self):
self.username = self.email = None
def __nonzero__(self): # 將對象轉換為bool類型時調用
return False
def requires_user(func):
@wraps(func)
def inner(user, *args, **kwargs): # 由于第一個參數(shù)無法支持self, 該裝飾器不支持裝飾類
if user and isinstance(user, User):
return func(use, *args, **kwargs)
else:
raise ValueError("非合法用戶")
return inner
輸出格式化
import json
from functools import wraps
def json_output(func): # 將原本func返回的字典格式轉為返回json字符串格式
@wrap(func)
def inner(*args, **kwargs):
return json.dumps(func(*args, **kwargs))
return inner
異常捕獲
import json
from functools import wraps
class Error1(Exception):
def __init__(self, msg):
self.msg = msg
def __str__(self):
return self.msg
def json_output(func):
@wrap(func)
def inner(*args, **kwargs):
try:
result = func(*args, **kwargs)
except Error1 as ex:
result = {"status": "error", "msg": str(ex)}
return json.dumps(result)
return inner
# 使用方法
@json_ouput
def error():
raise Error1("該條異常會被捕獲并按JSON格式輸出")
日志管理
import time
import logging
from functools import wraps
def logged(func):
@wraps(func)
def inner(*args, **kwargs): # *args可以裝飾函數(shù)也可以裝飾類
start = time.time()
result = func(*args, **kwargs)
exec_time = time.time() - start
logger = logging.getLoger("func.logged")
logger.warning("{} 調用時間:{:.2} 執(zhí)行時間:{:.2}s 結果:{}".format(func.__name__, start, exec_time, result)
帶參數(shù)的裝飾器
帶參數(shù)的裝飾器相當于一個返回裝飾器的函數(shù),@deco(a=1)在調用@之前會首先執(zhí)行deco(a=1)得到一個實際的裝飾器, 帶參數(shù)的裝飾器deco(a=1)模塊導入時立即執(zhí)行
裝飾類
為類增加可排序功能(而不通過繼承子類擴充父類方法,比如多個類需要增加此功能時)
import time
from functools import wraps
def sortable_by_created(cls):
original_init = cls.__init__
@wrap(original_init)
def new_init(self, *args, **kwargs):
original_init(*args, **kwargs)
self._created = time.time()
cls.__init__ = new_init
cls.__lt__ = lambda self, other: self._created < other._created
cls.__gt__ = lambda self, other: self._created > other._created
return cls
也可定義一個SortableByCreated()類, 子類使用多重繼承其父類和SortableByCreated
類型轉換
函數(shù)被裝飾后有可能變?yōu)橐粋€類的實例,此時為了兼容函數(shù)調用,應為所返回的類提供__call__方法
class Task(object):
def __call__(self, *args, **kwargs):
return self.run(*args, **kwargs)
def run(self, *args, **kwargs):
raise NotImplementedError("子類未實現(xiàn)該接口")
def task(func):
class SubTask(Task):
def run(self, *args, **kwargs):
func(*args, **kwargs)
return SubTask()
第二章 上下文管理器
定義
包裝任意代碼
確保執(zhí)行的一致性
語法
with語句
__enter__和__exit__方法
class ContextManager(object):
def __init__(self):
self.entered = False
def __enter__(self):
self.entered = True
return self
def __exit__(self, exc_type, exc_instance, traceback):
self.entered = False
應用場景
資源清理
import pymysql
class DBConnection(object):
def __init__(self, *args, **kwargs):
self.args,self.kwargs = args, kwargs
def __enter__(self):
self.conn = pymysql.connect(*args, **kwargs)
return self.conn.cursor()
def __exit__(self, exc_type, exc_instance, trackback):
self.conn.close()
異常處理(避免重復)
傳播異常(__exit__中return False)
終止異常(__exit__中return True)
class BubleExceptions(object):
def __enter__(self):
return self
def __exit__(self, exc_type, exc_instance, trackback):
if exc_instance:
print("出現(xiàn)異常: {}".format(exc_instance)
return False # return True終止異常
處理特定的異常
class HandleValueError(object):
def __enter__(self):
return self
def __exit__(self, exc_type, exc_instance, trackback):
if not exc_type: return True
if issubclass(exc_type, ValueError):
print("處理ValueError: {}".format(exc_instance)
return False
if issubclass...語句改為if exec_type == ValueError則不處理ValueType的子類異常
也可以根據(jù)異常的屬性來判斷是否傳播或終止
更簡單的語法
import contextlib
@contextlib.contextmanager
def acceptable_error_codes(*codes):
try:
yield
except ShellException as exc_instance:
if exc_instance.code not in codes:
raise
pass
以上就是本文的全部內容，希望對大家的學習有所幫助，也希望大家多多支持腳本之家。
相關文章
這篇文章主要給大家介紹了關于刪除PyCharm解釋器的方法步驟,PyCharm解釋器是指在PyCharm集成開發(fā)環(huán)境中用于運行和調試Python代碼的解釋器,需要的朋友可以參考下2023-09-09
文章介紹了如何在PyCharm中配置VimEmulator插件,包括檢查插件是否已安裝、下載插件以及安裝IdeaVim插件的步驟2025-02-02
python實現(xiàn)的各種排序算法，包括選擇排序、冒泡排序、插入排序、歸并排序等，學習python的朋友可以參考下2013-03-03
詳解多線程Django程序耗盡數(shù)據(jù)庫連接的問題
這篇文章主要介紹了多線程Django程序耗盡數(shù)據(jù)庫連接的問題，小編覺得挺不錯的，現(xiàn)在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧2018-10-10
這篇文章主要介紹了如何使用Python監(jiān)控屏幕變化并截圖,文中通過代碼示例給大家講解的非常詳細,并對大家的學習工作有一定的幫助,需要的朋友可以參考下2024-04-04
in在Python中是操作符，具體來說是成員操作符。這篇文章主要介紹了python 中in 的 用法,需要的朋友可以參考下2019-12-12
這篇文章主要介紹了PyCharm 專業(yè)版安裝圖文教程，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2020-02-02
python-opencv 將連續(xù)圖片寫成視頻格式的方法
今天小編就為大家分享一篇python-opencv 將連續(xù)圖片寫成視頻格式的方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-01-01
對于如何結束一個Python程序或者用Python操作去結束一個進程等,Python本身給出了好幾種方法,而這些方式也存在著一些區(qū)別,對相關的幾種方法看了并實踐了下,同時也記錄下,需要的朋友可以參考下2024-02-02
這篇文章主要為大家詳細介紹了python 2.7.14安裝圖文教程，具有一定的參考價值，感興趣的小伙伴們可以參考一下2018-04-04