谈谈python装饰器

今天的任务比较繁重,因为我们要一起来学习python中比较重要比较牛逼比较难的装饰器。我将会和大家一起通过代码的形式来模拟银行存款取款的功能,然后通过引入装饰器来一步一步优化代码。废话不多说,开始撸代码。我们去ATM存款取款的过程用代码来表示大概如下(不大准确,仅仅是为了演示需要)

上面是按下按钮1就存款,否则则取款。不对,存取款要输入密码啊!!!所以,我们要加密码验证代码。

可以看到,虽然实现了密码验证功能,但是代码冗余度比较高,而且现在只模拟了取款和存款功能,然而还有查询功能,转账功能等等,那么冗余度就更高了,而且相对于取款和存款函数来说,复用性没有那么高,所以我们要进一步优化代码,把验证函数写到取款和存款函数内部。

代码已经优化了一点点,但还是存在很多问题,比如函数内部又实现了另外一个函数,我们说过,一般情况下,一个函数最好仅仅实现一个功能,而且,这样直接修改函数的做法违背了''开放封闭''原则,代码的可维护性较差。为了在不改变原函数的情况下实现密码验证功能,我们把存取款函数写到密码验证函数里面。

虽然原函数没有发生改变,这个问题已经解决了,然而业务逻辑代码已经改变,也就是说业务逻辑代码需要重新更改,两三行还好,如果有成百上千行,那要改到天荒地老啊!!!可见,代码的维护性很差。有没有什么方法,可以在不改变原函数以及原函数的调用的情况下扩展原函数的功能呢?当然是有的,这就是python中著名的装饰器。还是使用上面的代码来演示

从现在开始,上面红框的代码不再进行更改,打死也不改,我们来实现有密码验证的存取款功能。

可以看到,即使原函数和原函数的调用都没有发生改变,但是函数的功能却扩展了,可维护性很高,这就是装饰器的强大之处。上面红框部分是不是觉得很熟悉,它就是我们昨天刚刚学过的闭包,而check_password()则是deposit()和withdrawal()的装饰器函数。语法糖上面的代码还可以继续简化。

其中@check_password就是语法糖,这个名字是不是很形象?甜甜的感觉,写起代码来也觉得很爽。装饰器装饰有参函数看下面这段代码

大家看看这段代码有没有什么问题?运行结果

看报错原因就知道,inner()函数多了一个参数。大家一定要注意一点,因为装饰器函数的返回值是inner,也就是说现在myprint是等同于inner的。我们来改一下代码

为了使装饰器能够装饰更多函数,我们进一步优化代码。

装饰器装饰有返回值函数

如图可以看到,无论被装饰的函数有无返回值,其结果都无返回值,原因其实很简单,因为inner()函数根本就没有返回值。为了实现有返回值的函数被装饰之后仍然有返回值,需要inner函数与被装饰函数的返回值保持一致。

可以看到,有返回值的函数被装饰之后依然有返回值,没有返回值的函数被装饰之后则没有返回值,符合我们想要的结果。双重语法糖为了更加直观地演示双重语法糖,我就不再使用上面的例子了。我先来定义一个函数

现在我有一个需求,我想在'python小白联盟'的上面打印一行'='和一行'*',也就是要实现这种效果。

那么应该怎么写呢?大家可能都会想到语法糖的叠加,那叠加的方式是以下哪一种呢?

我们来运行一下就知道了

可以看到,叠加的方式应该是第一种,也就是说最外层的语法糖先执行。带参数的语法糖为了演示带参数的语法糖,我再把上面的一张图搬下来

显然,红框部分的代码冗余度比较高,我们可以使用带参的语法糖来进行简化

装饰器就讲到这里,祝大家有个愉快的一天!

  • 发表于:
  • 原文链接:https://kuaibao.qq.com/s/20180722G0B6U900?refer=cp_1026
  • 腾讯「云+社区」是腾讯内容开放平台帐号(企鹅号)传播渠道之一,根据《腾讯内容开放平台服务协议》转载发布内容。

扫码关注云+社区

领取腾讯云代金券