手记

python 装饰器:动态扩展函数的功能

装饰器:动态扩展函数本身的功能
函数本身是一个对象 拥有一个name属性

  1. 不带参数的装饰器模式
#!/usr/bin/env python
# -*- coding:utf-8 -*-

import functools
def log(func):
    @functools.wraps(func)
    def wrapper(*args,**kw):
        print ('print first')
        func(*args,**kw)
        print ('print last')
    return wrapper

@log # @ 相当于调用了 log(show)
def show(a,b):
    print(a+b)

# 扩展了show函数本身的功能
show(1,8)
print(show.__name__)

"""
print first
9
print last
show
"""
  1. 带参数的装饰器模式
#!/usr/bin/env python
# -*- coding:utf-8 -*-

import functools
def log(text):
    def decor(func):
        @functools.wraps(func)
        def wrapper(*args,**kw):
            print('-----',text,'---')
            func(*args,**kw)
        return wrapper
    return decor

@log('我是装饰器参数')
def show(a,b):
    print(a+b)

show(1,9)
print(show.__name__)

"""
---我是装饰器参数---
10
show

"""
1人推荐
随时随地看视频
慕课网APP