python装饰器---------@Decorator(二)

1.定义处理任意函数、任意参数和任意默认值设置的装饰器

  • 解决方法是使用 *args 和 **kwargs传递内部包装函数的参数
def timer(func):
    import time
    @wraps(func)
    def f(*args,**kwargs):
        start = time.time()
        result = func(*args,**kwargs)
        end = time.time()
        print("{}execute time is {} s" .format(func.__name__,end-start))
        return result
    return f
 

2.创建装饰器时保留函数元信息(如__name__等不因装饰器的修饰而改变)

  •     问题:你写了一个装饰器作用在某个函数上,但是这个函数的重要的元信息比如名字、文档字符串、注解和参数签名都丢失了。
  • 解决方案--------任何时候你定义装饰器的时候,都应该使用 functools 库中的 @wraps 装饰器来注解底层包装函数。

code6

from functools import wraps


'''计时函数'''
def timer(func):
    import time
    @wraps(func)
    def f(x,y):
        start = time.time()
        result = func(x,y)
        end = time.time()
        print "{}({},{}) execute time is {} s" .format(func.__name__,x,y,end-start) 
        return result
    return f
 
 
'''@name表示用name装饰器修饰下一行的函数'''
 
@timer
def add(x,y):
    result = x + y
    return x + y
@timer
def sub(x,y):
   return x-y
@timer
def multi(x,y):
    return x*y
 
 
'''测试'''
print add.__name__
print sub.__name__

输出结果:

   有兴趣的同学输出装饰器(一)这篇博客中code5的 add.__name和sub.__name__看看和上面的代码输出结果有什么比不同。

3.定义带参数的装饰器

带参数的装饰器如下:

@decorator(x, y, z)
def func(a, b):
    pass

这个装饰器处理过程跟下面的调用是等效的:

def func(a, b):
    pass
func = decorator(x, y, z)(func)

decorator(x, y, z) 的返回结果必须是一个可调用对象,它接受一个函数作为参数并包装它。

下面我们通过一个具体的例子来看一下带参数的装饰器定义过程:

 假设你想写一个装饰器,给函数添加日志功能,同时允许用户指定日志的级别和其他的选项。 下面是这个装饰器的定义和使用示例:

from functools import wraps
import logging

def logged(level, name=None, message=None):
    """
    Add logging to a function. level is the logging
    level, name is the logger name, and message is the
    log message. If name and message aren't specified,
    they default to the function's module and name.
    """
    def decorate(func):
        logname = name if name else func.__module__
        log = logging.basicConfig(level = level)
        logmsg = message if message else func.__name__

        @wraps(func)
        def wrapper(*args, **kwargs):
            logging.log(level, logmsg)
            return func(*args, **kwargs)
        return wrapper
    return decorate

# Example use
@logged(logging.DEBUG)
def add(x, y):
    return x + y

@logged(logging.CRITICAL, 'example')
def spam():
    print('Spam!')
  
    
add(2,14)    
spam()

       初看起来,这种实现看上去很复杂,但是核心思想很简单。 最外层的函数 logged() 接受参数并将它们作用在内部的装饰器函数上面。 内层的函数 decorate() 接受一个函数作为参数,然后在函数上面放置一个包装器。 这里的关键点是包装器是可以使用传递给 logged() 的参数的。

4.解除装饰器

  • 问题-----------------一个装饰器已经作用在一个函数上,你想撤销它,直接访问原始的未包装的那个函数
  • 解决方案---------假设装饰器是通过 @wraps (参考9.2小节)来实现的,那么你可以通过访问 __wrapped__ 属性来访问原始函数

下面的示例中核心代码为:orign_add = add.__wrapped__   

但是注意,python3.3以上版本中才有__wrapped__属性,因此本次我们用python3.7版本测试

from functools import wraps


'''计时函数'''
def timer(func):
    import time
    @wraps(func)
    def f(x,y):
        start = time.time()
        result = func(x,y)
        end = time.time()
        print("{}({},{}) execute time is {} s" .format(func.__name__,x,y,end-start))
        return result
    return f
 
 
'''@name表示用name装饰器修饰下一行的函数'''
 
@timer
def add(x,y):
    result = x + y
    return x + y


orign_add = add.__wrapped__ 
 
'''测试'''
print(add(3,2))
print(orign_add(3,2))

输出结果:


注意:

  • 该方法尽在python3.3以上版本使用
  • 直接访问未包装的原始函数在调试、内省和其他函数操作时是很有用的。 但是我们这里的方案仅仅适用于在包装器中正确使用了 @wraps 或者直接设置了 __wrapped__ 属性的情况。

  • 如果有多个包装器,那么访问 __wrapped__ 属性的行为是不可预知的,应该避免这样做

  • 并不是所有的装饰器都使用了 @wraps ,因此这里的方案并不全部适用。 特别的,内置的装饰器 @staticmethod 和 @classmethod 就没有遵循这个约定 (它们把原始函数存储在属性 __func__ 中)

参考资料:

资料一:python中解除装饰器

资料二:创建装饰器时保留函数元信息

资料三:*args和**kwargs

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
装饰器模式是一种结构型设计模式,它可以动态地给对象添加额外的职责,同时又不需要修改原有的代码。在电商领域中,我们可以使用装饰器模式来实现优惠活动价格计算。 首先,我们需要定义一个抽象组件类,代表商品或服务: ```python from abc import ABC, abstractmethod class Component(ABC): @abstractmethod def get_price(self): pass ``` 然后,我们定义具体的商品类: ```python class Product(Component): def __init__(self, name, price): self.name = name self.price = price def get_price(self): return self.price ``` 接下来,我们定义装饰器类: ```python class Decorator(Component): def __init__(self, component): self.component = component def get_price(self): return self.component.get_price() ``` 其中,`Decorator` 类继承自 `Component` 类,它包含一个对抽象组件的引用,代表要装饰的组件。 然后,我们定义具体的装饰器类,用来实现各种优惠活动: ```python class Discount(Decorator): def __init__(self, component, discount_rate): super().__init__(component) self.discount_rate = discount_rate def get_price(self): return self.component.get_price() * self.discount_rate ``` `Discount` 类继承自 `Decorator` 类,它包含一个对抽象组件的引用,同时还有一个折扣率属性 `discount_rate`。`get_price` 方法将原始价格与折扣率相乘,得到折扣后的价格。 我们还可以定义其他的装饰器类,如满减、满赠等等。 最后,我们可以使用这些装饰器类来计算商品的最终价格: ```python product = Product('iPhone', 10000) discounted_product = Discount(product, 0.8) # 打八折 print(discounted_product.get_price()) # 输出 8000 ``` 这样,我们就通过装饰器模式实现了电商的优惠活动价格计算。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值