with语句足够优雅,特别使用语一些上下文管理。常用的打开文件对象,建立连接,Lock等操作。原理是通过 enter , exit 两个魔术方法来实现。写了一个简单的例子如下:
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另一个优雅语法的例子,装饰器的优势和缺点,不作赘述。通过一个执行异常,进行重试的业务,举例几个常用的装饰器类型的写法:不带参数装饰器,带默认参数的装饰器,类的实例方法的装饰器,类装饰器。
python
import time
from functools import wraps
# 不带参数的装饰器
def retry_func(func):
@wraps(func)
def warp_call(*args, **kwargs):
print('执行 retry_func')
return func(*args, **kwargs)
return warp_call
# 带默认参数装饰器,出现异常后进行重试,总计运行amount次
def retry(amount=3):
def warp_rsp(func):
@wraps(func)
def warp_call(*args, **kwargs):
"""执行异常,重复执行2次,执行失败返回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
# 带参数的用于装饰实例方法的装饰器
def retry_class(amount=3):
def warp_rsp(func):
@wraps(func)
# 调用函数传入一个self参数,用于绑定实例对象
def warp_call(self, *args, **kwargs):
"""执行异常,重复执行,执行失败返回None"""
for i in range(amount):
try:
rsp = func(self, *args, **kwargs)
break
except Exception as e:
# 可以直接调用实例方法
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}/{b}')
return a / b
# 类装饰器,通过类属性传递默认参数
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 #修改重试次数,全局性的
@class_decorator
def func(a, b):
print(f'{a}/{b}')
return a / b