python3 with語句和裝飾器使用舉例

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

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡書系信息發(fā)布平臺,僅提供信息存儲服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容