python 基础 —装饰器练习

由于对装饰器不是特别理解，因此做了以下练习，并debug自行理解，在此做一个记录

以下练习均出自菜鸟教程装饰器一章 http://www.runoob.com/w3cnote/python-func-decorators.html

第一个练习

from functools import wraps

def deco_name(f):
    @wraps(f)
    def decorated(*args,**kwargs):
        if not can_run:
            return "Function will not run"
        return f(*args,**kwargs)
    return decorated

@deco_name  #func=deco_name(func)
def func():  #执行了deco_name装饰器后返回来了decorated，因此func=decorated
    return("Function is running")

can_run =True
print(func())
#执行func()相当于执行decorated()
# can_run =True，因此decorated()返回func()返回的值："Function is running"

can_run =False
print(func())
#can_run =False，decorated()进入if选择，返回了"Function will not run"

 

当使用装饰器装饰一个函数时，函数本身就已经是一个新的函数；即函数名称或属性产生了变化。

在python的functools模块中提供了wraps装饰函数来确保原函数在使用装饰器时不改变自身的函数名及应有属性。
因此在装饰器的编写中建议加入wraps确保被装饰的函数不会因装饰器带来异常情况。

第二个练习

from functools import wraps

def logit(func):
    @wraps(func)
    def with_logging(*args, **kwargs):
        print(func.__name__ + " was called") #addition_func=func ==》输出addition_func_was called
        return func(*args, **kwargs) #addition_func=func，因此返回addition_func执行的结果

    return with_logging


@logit  #  addition_func=logit(addition_func=func)返回with_logging ==》addition_func=with_logging
def addition_func(x):
    """Do some math."""
    return x + x

#addition_func(4)= with_logging(4)=func(4)
result = addition_func(4)
print(result)

 

第三个练习

带参数的装饰器

from functools import wraps
def logit(logfile='out.log'):
    def logging_decorator(func):
        @wraps(func)
        def wrapped_function(*args, **kwargs):
            log_string = func.__name__ + " was called"
            print(log_string)
            # 打开logfile，并写入内容
            with open(logfile, 'a') as opened_file:
                # 现在将日志打到指定的logfile
                opened_file.write(log_string + '\n')
            return func(*args, **kwargs)
        return wrapped_function
    return logging_decorator

@logit() #myfunc1=logit(myfunc1)
def myfunc1():
    pass

myfunc1()
# Output: myfunc1 was called
# 现在一个叫做 out.log 的文件出现了，里面的内容就是上面的字符串

@logit(logfile='func2.log')
def myfunc2():
    pass

myfunc2()
# Output: myfunc2 was called
# 现在一个叫做 func2.log 的文件出现了，里面的内容就是上面的字符串