python装饰器

关于__name__

一个python文件，自己独立执行的时候名字叫__main__，被别的文件调用的时候，该python文件的名字叫他的文件名。所以经常会看到：

if __name__ == '__main__':

就是说，当本文件被独立执行时，执行if下的语句，被调用时，不执行if下的语句。

装饰器

装饰器是一个函数，并且传入它的参数是一个函数，它返回的也是一个函数。当函数后边加()时，表示这个函数被执行；不加()是，只是表示获取函数的属性（被传递）。

装饰器的目的一般在于给函数添加额外的属性。

# 定义一个装饰器
def decorator(func):
    def back():
        print('hello')
        func()
    return back
# 调用装饰器，注意，装饰器只是增加函数属性，但并不运行函数本身。
@decorator
# 定义被装饰的函数
def name():
    print ('Adam')
#运行被装饰过的函数
name()
# 打印可知，被装饰过的函数已经被back替代了
print(name.__name__)

输出：

hello
Adam
back

我们使用wraps来控制在给name函数赋予新的属性时，并不使其本身被back函数所替代。

from functools import wraps
def decorator(func):
    @wraps(func)
    def back():
        print('hello')
        func()
    return back
@decorator
def name():
    print ('Adam')
name()
print(name.__name__)

输出：

hello
Adam
name

参考：Python 函数装饰器