9.1在函数上添加包装器.pdfVIP

9.1 在函数上添加包装器 问题 你想在函数上添加⼀个包装器，增加额外的操作处理(⽐如⽇志、计时等) 。 解决⽅案 如 你想使⽤额外的代码包装⼀个函数，可以定义⼀个装饰器函数，例如： import time from functools import wraps def timethis(func): Decorator that reports the execution time. @wraps(func) def wrapper(*args, **kwargs): start = time.time() result = func(*args, **kwargs) end = time.time() print(func. name , end-start) return result return wrapper 下⾯是使⽤装饰器的例⼦： @timethis ... def countdown(n): ... ... Counts down ... ... while n 0: ... n -= 1 ... countdown(100000) countdown 0.008917808532714844 countdown countdown 0.87188299392912 讨论 ⼀个装饰器就是⼀个函数，它接受⼀个函数作为参数并返回⼀个新的函数。当你像下 ⾯这样写： @timethis def countdown(n): pass 跟像下⾯这样写其实效 是⼀样的： def countdown(n): pass countdown = timethis(countdown) 顺便说⼀下，内置的装饰器⽐如 @staticmethod, @classmethod,@property 原理也是⼀样的。例如，下⾯这两个代码⽚段是等价的： class A : @classmethod def method(cls): pass class B: # Equivalent definition of a class method def method(cls): pass method = classmethod(method) 在上⾯的 wrapper() 函数中，装饰器内部定义了⼀个使⽤ *args 和 **kwargs 来 接受任意参数的函数。在这个函数⾥⾯调⽤了原始函数并将其结 返回，不过你还可 以添加其他额外的代码(⽐如计时) 。然后这个新的函数包装器被作为结 返回来代替 原始函数。 需要强调的是装饰器并不会修改原始函数的参数签名以及返回值。使⽤ *args 和 **kwargs ⽬的就是确保任何参数都能适⽤。⽽返回结 值基本都是调⽤原始函数 func(*args, **kwargs) 的返回结 ，其中func就是原始函数。 刚开始学习装饰器的时候，会使⽤⼀些简单的例⼦来说明，⽐如上⾯演⽰的这个。不 过实际场景使⽤时，还是有⼀些细节问题要注意的。⽐如上⾯使⽤ @wraps(func) 注解是很重要的，它能保留原始函数的元数据(下⼀⼩节会讲到) ，新⼿经常会忽略这 个细节。接下来的⼏个⼩节我们会更加深⼊的讲解装饰器函数的细节问题，如 你想 构造你⾃⼰的装饰器函数，需要认真看⼀下。 W3Cschool （ ）最⼤的技术知识分享与学习 台 此篇内容来⾃于 ⽹站⽤户上传并发布。