Python 中的 @wraps 到底是个啥东西?

开发 前端
更具体地说,@wraps(something)​ 用 something 的元数据覆盖了函数的元数据。这样,我们原来函数的元数据就不会丢失了。

只要你读得很快

你可能在随意的 Python 代码中见过这个 @wraps 的东西,你可能想知道这到底是什么?

函数有元数据

元数据指的是函数本身的数据。

def apple():
  '''a function that prints apple'''
  print('apple')

print(apple.__name__)  # apple
print(apple.__doc__)   # 打印apple的函数

例子包括函数名 .__name__ 或函数 docstring .__doc__

装饰器如何工作

装饰器用于改变函数的行为方式,而无需修改任何源代码。

def greet(name):
    return 'hello ' + name

print(greet('tom'))  # hello tom

在这里,我们有一个普通的 greet 功能

def add_exclamation(func):
    def wrapper(name):
        return func(name) + '!'
    return wrapper

@add_exclamation
def greet(name):
    return 'hello ' + name

print(greet('tom'))  # hello tom!

我们通过在 greet() 上添加 @add_exclamation 来用 add_exclamation() 来装饰 greet()。这里,add_exclamation 是装饰器,greet 是被装饰的函数。

请注意,greet() 的行为已经改变,而我们根本没有编辑 greet() 的源代码。这是装饰器的功劳。

装饰语法魔术

def add_exclamation(func):
    def wrapper(name):
        return func(name) + '!'
    return wrapper

@add_exclamation
def greet(name):
    return 'hello ' + name

print(greet('tom'))  # hello tom!

这是完全相同的:

def add_exclamation(func):
    def wrapper(name):
        return func(name) + '!'
    return wrapper

def greet(name):
    return 'hello ' + name

greet = add_exclamation(greet)

print(greet('tom'))  # hello tom!

注意 “greet = add_exclamation(greet) ”一行。

装饰会导致元数据丢失

# 没有装饰
def greet(name):
  '''says hello to someone'''
  return 'hello ' + name

print(greet.__name__)  # greet
print(greet.__doc__)   # says hello to someone

在这里,我们可以顺利打印 greet() 的元数据。

# 加了装饰
def add_exclamation(func):
    def wrapper(name):
        return func(name) + '!'
    return wrapper

@add_exclamation
def greet(name):
  '''says hello to someone'''
  return 'hello ' + name

print(greet.__name__)  # wrapper
print(greet.__doc__)   # None

用 add_exclamation 装饰 greet 后,请注意元数据发生了变化。__name__ 变成了 “wrapper”,而 __doc__ 变成了 wrapper 的 docstring。

这是因为当我们装饰 greet 时,我们实际上是在做这件事:

greet = add_exclamation(greet)

我们正在将 greet 重新分配给一个由 add_exclamation 返回的函数--wrapper。

这就是为什么当我们尝试打印 greet.__name__ 和 greet.__doc__ 时,会打印 wrapper 的元数据的原因。

@wraps 防止元数据在装饰过程中丢失

from functools import wraps

def add_exclamation(func):
    @wraps(func)
    def wrapper(name):
        return func(name) + '!'
    return wrapper

@add_exclamation
def greet(name):
  '''says hello to someone'''
  return 'hello ' + name

print(greet.__name__)  # greet
print(greet.__doc__)   # says hello to someone

请注意,尽管使用了 ad_exclamation 装饰,greet 的元数据还是回到了正常状态。

更具体地说,@wraps(something) 用 something 的元数据覆盖了函数的元数据。这样,我们原来函数的元数据就不会丢失了。

责任编辑:武晓燕 来源: 数据STUDIO
相关推荐

2022-04-10 19:26:07

TypeScript类型语法

2024-08-26 14:23:56

2024-02-07 12:35:00

React并发模式concurrent

2021-01-08 09:48:18

Pythonname变量

2021-12-16 15:11:59

Facebook天秤币加密货币

2022-09-06 21:38:45

数字人数字孪生

2021-05-11 07:30:58

JNIJavaAPI

2022-05-04 08:38:32

Netty网络框架

2021-01-28 17:41:32

Github网站Pull Reques

2020-10-25 20:05:29

Pythonyield开发

2021-12-26 00:01:51

Log4Shell漏洞服务器

2024-08-01 17:34:56

Promiseaxios请求

2020-03-07 09:47:48

AVL树算法场景

2024-02-01 20:15:37

2013-05-29 10:17:56

Hadoop分布式文件系统

2012-07-25 09:09:46

GNOME OS桌面

2011-12-28 09:53:18

HTML 5

2021-02-01 06:41:47

流水线计算机队列

2021-12-16 21:13:38

通信网管平台

2021-05-19 10:44:42

数据库架构技术
点赞
收藏

51CTO技术栈公众号