with語句足夠優(yōu)雅,特別使用語一些上下文管理。常用的打開文件對象,建立連接,Lock等操作。原理是通過 enter , exit 兩個魔術(shù)方法來實現(xiàn)。寫了一個簡單的例子如下:
class request():
"""with語句支持的上下文管理類"""
def __init__(self, *args, **kwargs):
self.args = args
self.kwargs = kwargs
def __enter__(self):
print(f'do someing:{self.args},{self.kwargs}')
return self
def __exit__(self, exc_type, exc_val, exc_tb):
print(f'trace:{exc_val}')
return True
def get(self, url):
print(f'url:{url}')
return 'result'
if __name__ == "__main__":
# with語句的使用舉例
with request('13131', a=2) as req:
print(req.get('http://www.baidu.com'))
裝飾器算是python另一個優(yōu)雅語法的例子,裝飾器的優(yōu)勢和缺點(diǎn),不作贅述。通過一個執(zhí)行異常,進(jìn)行重試的業(yè)務(wù),舉例幾個常用的裝飾器類型的寫法:不帶參數(shù)裝飾器,帶默認(rèn)參數(shù)的裝飾器,類的實例方法的裝飾器,類裝飾器。
python
import time
from functools import wraps
# 不帶參數(shù)的裝飾器
def retry_func(func):
@wraps(func)
def warp_call(*args, **kwargs):
print('執(zhí)行 retry_func')
return func(*args, **kwargs)
return warp_call
# 帶默認(rèn)參數(shù)裝飾器,出現(xiàn)異常后進(jìn)行重試,總計運(yùn)行amount次
def retry(amount=3):
def warp_rsp(func):
@wraps(func)
def warp_call(*args, **kwargs):
"""執(zhí)行異常,重復(fù)執(zhí)行2次,執(zhí)行失敗返回None"""
for i in range(amount):
try:
rsp = func(*args, **kwargs)
break
except Exception as e:
print(f'exception :{e}')
else:
return None
return rsp
return warp_call
return warp_rsp
# 帶參數(shù)的用于裝飾實例方法的裝飾器
def retry_class(amount=3):
def warp_rsp(func):
@wraps(func)
# 調(diào)用函數(shù)傳入一個self參數(shù),用于綁定實例對象
def warp_call(self, *args, **kwargs):
"""執(zhí)行異常,重復(fù)執(zhí)行,執(zhí)行失敗返回None"""
for i in range(amount):
try:
rsp = func(self, *args, **kwargs)
break
except Exception as e:
# 可以直接調(diào)用實例方法
self.print_error()
print(f'exception :{e}')
else:
return None
return rsp
return warp_call
return warp_rsp
class class_method():
def print_error(self):
print('print_error')
# 在實例方法中使用裝飾器
@retry_class(amount=5)
def func(self, a, b):
print(f'{a}/')
return a / b
# 類裝飾器,通過類屬性傳遞默認(rèn)參數(shù)
class class_decorator():
amount = 3
def __init__(self, func):
self.func = func
print(f'retry:{self.__class__.amount}')
def __call__(self, *args, **kwargs):
for i in range(self.__class__.amount):
try:
rsp = self.func(*args, **kwargs)
break
except Exception as e:
time.sleep(1)
else:
return None
return rsp
if __name__ == "__main__":
# 類實例方法的使用舉例
obj = class_method()
obj.func(1, 0)
# 類裝飾器的使用舉例
class_decorator.amount = 5 #修改重試次數(shù),全局性的
@class_decorator
def func(a, b):
print(f'{a}/')
return a / b