python3 with语句和装饰器使用举例

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