如何使用函数装饰器?

需求:
某些时候我们想为多个函数,统一添加某种功能,比如计时统计、日志记录、缓存运算结果等等
我们不想在每个函数内一一添加完全相同的代码,有什么好的解决方案?

思路:
定义装饰器函数,用它来生成一个在原函数基础上添加了新功能的函数,替代原函数

代码:

def memo(func):
    cache = {}
    def wrap(*args):
        res = cache.get(args) # 此处args传入的是一个元组,元组可以作为字典的键
        if not res:
            res = cache[args] = func(*args) # 将传入的元组参数打散来
        return res
    return wrap

# [题目1] 斐波那契数列(Fibonacci sequence):
# F(0)=1,F(1)=1,F(n)=F(n-1)+F(n-2)(n>=2)
# 1,1,2,3,5,8,13,21,34,...
# 求数列第n项的值
# 使用缓存,可以去除重复的大量运算,提高运算的速度
@memo # 等价于fibonacci = memo(fibonacci)
def fibonacci(n):
    if n <= 1:
        return 1
    res = fibonacci(n-1) + fibonacci(n-2)
    return res


# [题目2] 走楼梯问题
# 有100阶楼梯,一个人可以迈1~3阶,一共有多少走法?
@memo
def climb(n,steps):
    count = 0
    if n == 0:
        count = 1
    elif n > 0:
        for step in steps:
            count += climb(n-step,steps)
    return count

# fibonacci = memo(fibonacci)  # 偷梁换柱,用户直接调用的还是fibonacci,不感知函数内部
print(fibonacci(50))

# climb = memo(climb)  # 偷梁换柱,用户直接调用的还是climb,不感知函数内部
print(climb(100,(1,2,3)))
posted @ 2020-11-11 10:23  Richardo-M-Lu  阅读(79)  评论(0)    收藏  举报