Python中的装饰器的简单介绍01

一. 装饰器是什么？

　　简单来说，装饰器其实也就是一个函数，一个用来包装函数的函数，返回一个修改之后的函数对象，将其重新赋值原来的标识符，并永久丧失对原始函数对象的访问。

二.装饰器语法

　　(1)无参数装饰器

def deco(func):  
    print (func)  
    return func  
@deco  
def foo():
    pass  
foo() 

　　第一个函数deco是装饰函数，它的参数就是被装饰的函数对象。我们可以在deco函数内对传入的函数对象做一番“装饰”，然后返回这个对象（记住一定要返回 ，不然外面调用foo的地方将会无函数可用。实际上此时foo=deco(foo)

　　下面是一个简单的例子，检查函数有没有说明文档：

def deco_functionNeedDoc(func):  
    if func.__doc__ == None :  
        print (func.__name__, "has no __doc__, it's a bad habit.")  
    else:  
        print (func.__name__, ':', func.__doc__, '.')  
    return func  
@deco_functionNeedDoc  
def f():  
    print ('f() Do something')  
@deco_functionNeedDoc  
def g():  
    'I have a __doc__'  
    print ('g() Do something')  
f()  
g() 

#输出如下
#f has no __doc__, it's a bad habit.
#g : I have a __doc__ .
#f() Do something
#g() Do something
#[Finished in 0.2s]

 　　(2)有参数装饰器 

　　不使用装饰器的实现代码：

import time
import datetime


def outter(func):
    def inner(args):
        start=datetime.datetime.now()
        func(args)
        end=datetime.datetime.now()
        cost=end-start

        print("execute %s spend %s"%(func.__name__,cost.total_seconds()))
    return (inner)

def func2(args):
    time.sleep(args)

f=outter(func2)
f(1)

#输出如下：
#execute func2 spend 1.014
#[Finished in 1.2s]

　　使用装饰器的代码实现如下：

import time
import datetime
import functools


def outter(func):
    @functools.wraps(func)
    def inner(*args,**kwargs):
        '''This is inner'''
        start=datetime.datetime.now()
        func(*args,**kwargs)
        end=datetime.datetime.now()
        cost=end-start

        print("execute %s spend %s"%(func.__name__,cost.total_seconds()))
    return (inner)

@outter
def func2(args):
    '''This is func2'''
    time.sleep(args)

@outter
def func3(m,n):
    '''This is func3'''
    print("m+n={0}".format(m+n))

func2(1)
print(func2.__name__)
print(func2.__doc__)

func3(m=2,n=3)
print(func3.__name__)
print(func3.__doc__)

#加了@functools.wraps(func)装饰器的输出结果
# execute func2 spend 1.000413
# func2
# This is func2
# m+n=5
# execute func3 spend 0.0
# func3
# This is func3
# [Finished in 1.2s]

#不加@functools.wraps(func)装饰器的输出结果

# execute func2 spend 1.000413
# inner
# This is inner
# m+n=5
# execute func3 spend 0.0
# inner
# This is inner
# [Finished in 1.2s]