python带参数的装饰器调用原理_python装饰器原理和用法总结(示例代码)

原理

装饰器本质也是一个函数, 只不过这个函数需要遵循以下规则:

入参只能有一个，类型为函数。 被装饰的函数将入会被传入这个参数

返回值是必须是一个函数, 届时被调用的时候实际上调用的是返回出来的这个函数，所以返回的函数入参通常是

(*args, **kwargs):

以满足所有函数需要

之后通过@语法糖即可装饰到任意函数上

简单装饰器例子

# 不带参数的装饰器

def pre_do_sth(func):

def wrapper(*args, **kwargs):

print("Do sth before call one")

func(*args, **kwargs)

return wrapper

@pre_do_sth

def echo(msg):

print(msg)

echo("Hello World")

运行结果

Do sth before call one

Hello World

实际上调用的是 wrapper("Hello World") --> echo("Hello World")

带参数的装饰器例子(参数控制的是装饰器的行为)

只需要写一个返回 装饰器(入参只有一个， 返回值是一个函数)函数的函数

同样也能利用@语法糖

# 带参数的装饰器

def pre_do_sth_2(msg):

def decorator(func):

def wrapper(*args, **kwargs):

print("Do sth before call two, print:%s"%(msg))

func(*args, **kwargs)

return wrapper

return decorator

@pre_do_sth_2("Foo")

def echo(msg):

print(msg)

echo("Hello World")

实际上@后面并不是对pre_do_sth_2这个函数生效 而是对pre_do_sth_2的返回值生效

运行结果

Do sth before call two, print:Foo

Hello World

多个装饰器调用顺序

先声明的装饰器先执行, 即在最外层

# 不带参数的装饰器

def pre_do_sth(func):

def wrapper(*args, **kwargs):

print("Do sth before call one")

func(*args, **kwargs)

return wrapper

# 带参数的装饰器

def pre_do_sth_2(msg):

def decorator(func):

def wrapper(*args, **kwargs):

print("Do sth before call two, print:%s"%(msg))

func(*args, **kwargs)

return wrapper

return decorator

@pre_do_sth

@pre_do_sth_2("Foo")

def echo(msg):

print(msg)

echo("Hello World")

运行结果

Do sth before call one

Do sth before call two, print:Foo

Hello World