【随笔】【python】装饰器

python装饰器是啥?
'''
  装饰器本质上是一个python函数,它可以让其他函数在不需要做任何代码变动的前提下增加额外功能,
装饰器的返回值也是一个函数对象。它经常用于有切面需求的场景,比如插入日志、性能测试、事物处理、
缓存、权限校验等场景。装饰器是解决这类问题的绝佳设计,有了装饰器,我们就可以抽离出大量与函数
功能本身无关的雷同代码并继续重用。
  简单来说,装饰器的作用就是为已经存在的函数或对象添加额外的功能。
'''

# encoding: utf-8
import os
import sys
# 可变参数*args和关键字参数**kwargs
def debug(func):
    def wrapper(*args,**kwargs):
        print("[DEBUG]:函数名称 {}()".format(func.__name__))
        return func(*args,**kwargs)
    return wrapper

# 如果不使用 @语法糖 (使用语法糖看起来更加优雅),写法如下注释掉的代码:
#def hello(var):
#    print("hello 装饰器!{}".format(var))
#hello = debug(hello) # func 指向函数 hello , hello 重新指向函数 wrapper
                      # 这时hello() 就等价于 wrapper() , func 指向之前的函数 hello

@debug
def hello(var):
    print("hello 装饰器!{}".format(var))

if __name__=="__main__":
	hello("10086")

************************************
 输出结果:
 [DEBUG]:函数名称 hello()
 hello 装饰器!10086
************************************

'''
高级一点的装饰器

 - 带参数的装饰器
 - 基于类实现的装饰器
 - 带参数的类装饰器

'''
'*** 带参数的装饰器 ***'
# encoding: utf-8
import os
import sys

def logging(level):
    def wrapper(func):
        def inner_wrapper(*args, **kwargs):
            print "[{level}]: enter function {func}()".format(level=level,func=func.__name__)
            return func(*args, **kwargs)
        return inner_wrapper
    return wrapper

@logging(level='INFO')
def say(something):
    print "say {}!".format(something)

# 如果没有使用@语法,等同于
# say = logging(level='INFO')(say)

@logging(level='DEBUG')
def do(something):
    print "do {}...".format(something)

if __name__ == '__main__':
    say('hello 10086')
    do("my work")

************************************
 输出结果:
 [INFO]: enter function say()
 say hello 10086!
 [DEBUG]: enter function do()
 do my work...
************************************


'*** 基于类实现的装饰器 ***'
# encoding: utf-8
import os
import sys

class logging(object):
    def __init__(self, func):
        self.func = func

    def __call__(self, *args, **kwargs):
        print "[DEBUG]: enter function {func}()".format(
            func=self.func.__name__)
        return self.func(*args, **kwargs)
@logging
def say(something):
    print "say {}!".format(something)

if __name__ == '__main__':
    say('hello 10086')

************************************
 输出结果:
 [DEBUG]: enter function say()
 say hello 10086!
************************************


'*** 带参数的类装饰器 ***'
# encoding: utf-8
import os
import sys

class logging(object):
    def __init__(self, level='INFO'):
        self.level = level
        
    def __call__(self, func): # 接受函数
        def wrapper(*args, **kwargs):
            print "[{level}]: enter function {func}()".format(
                level=self.level,
                func=func.__name__)
            func(*args, **kwargs)
        return wrapper  #返回函数

@logging(level='DEBUG')
def say(something):
    print "say {}!".format(something)

if __name__ == '__main__':
    say('hello 10086')

************************************
 输出结果:
 [DEBUG]: enter function say()
 say hello 10086!
************************************

上一篇:手写封装javascript的call,apply和bind方法


下一篇:PAT 1148 Werewolf – Simple Version – 甲级