引言

众所周知，python 中，装饰器是非常好玩的，你能够在很多场景中看到它。

有很多人可能经常会使用他人设计的装饰器，自己却很少设计过几个装饰器。当然也不乏有的大神非常善于设计装饰器。

但不管如何，装饰器确实能够激发出非常多的使用场景。

本文将设计一个非常好玩的装饰器，当然不光好玩，也在某些场景下还挺实用的。

一、本文装饰器的用途

本文设计的装饰器到底有什么用呢？我们来看一个场景。

很多时候，我们在程序中要告知用户，当前进度如何。比如一个下载进度：

下载进度条

怎么样，这个场景是非常常见的吧。

本文将通过设计一个 装饰器，它能让一个普通函数拥有进度条的功能。

二、安装库

pip install tqdm

本文中的示例是通过控制台来显示进度条的，如果你的程序是有GUI的，请自行替换为GUI版本的进度条即可。

三、完整代码

import time
from tqdm import trange


def bar(desc='', unit='it'):
    def decorator(func):
        def inner(*args, **kwargs):
            pbar = None

            gen = func(*args, **kwargs)

            try:
                while True:
                    i = next(gen)
                    if pbar is None:
                        pbar = trange(i, desc=desc, unit=unit)

                    pbar.update(1)
            except StopIteration as e:
                pbar.close()

                return e.value

        return inner
    return decorator


@bar(desc='测试的进度条')
def main():
    total = 100
    yield total

    for i in range(total):
        yield
        time.sleep(0.05)

    return '我是函数的结果'


if __name__ == '__main__':
    result = main()
    print(result)

可以看到，改装成 装饰器 后，想让一个函数变成 进度条 版本的，将变得非常简单。主要有以下几个步骤：

1. 为你的函数添加 bar 装饰器
2. 添加 第一个yield 返回总进度
3. 在工作中使用 yield，返回相应的进度
4. 给函数一个返回值

通过一个简单改造，你的函数就将拥有进度条。

第一个yield，为什么需要返回总进度

可能有同学问了，为什么第一个 yield 要返回总进度？

那是因为如果你 不返回总进度，就无法预估什么时候到百分之几，必须能够计算出总进度的函数才适合使用这个装饰器。

四、结尾

可能会有大神质疑，这个装饰器不就是脱裤子放屁多此一举了吗？

没关系，确实装饰器可能并没有提升多大的优势，但可以让不少没怎么用过装饰器的朋友见识到装饰器的强大。

通过理解这个装饰器，学会发散思考，如何玩转代码，给代码带来活力！