Python 类装饰器,使用小演示

Python与算法社区

共 1521字,需浏览 4分钟

 ·

2020-11-19 23:23

Python与算法社区
433篇原创,干货满满
值得星标


01

02

03


三步加星标





实现同样一个功能,用Java语言可能得50行,而用Python可能只需10行,可能很多读者在没有学Python前,就从用过Python的前辈那里,听说过这个,然后自己也开始去学Python的。


Python的确简洁、优雅,很多读者包括我,都为之着迷。


今天通过一个小例子,再次认识Python的clean之道:


我们想要检查每个参数,确保不能出现整数类型。使用类装饰器,便能做到clean,通用:


import functools


class ValidateParameters:
    def __init__(self, func):
        # 确保func被装饰后函数信息不被改变
        functools.update_wrapper(self, func)
        self.func = func

    # 重写此方法,让类对象变得可调用
    def __call__(self, *parameters):
        if any([isinstance(item, int) for item in parameters]):
            raise TypeError("Parameter shouldn't be int!!")
        else:
            return self.func(*parameters)

使用上面的类,开始装饰我们的各种函数,比如连接字符串函数 concat,第一次调用传参都为字符串,类型满足要求;第二次调用传参有整数类型,抛出异常:Parameter shouldn't be int!!

@ValidateParameters
def concat(*list_string):
    return "".join(list_string)


#  returns anb
print(concat("a""n""b"))

# raises Error.
print(concat("a"1"c"))

任意一个想要有类型检查的函数,都可以装饰上ValidateParemeters,如下capital函数,实现串联字符串后,且首字母大写:

@ValidateParameters
def capital(*list_string):
    return concat(*list_string).capitalize()


print(capital("a""n""b"))

print(capital(2"n""b"))

如果输入参数有整型,必然抛出上面的异常,再次方便的实现类型检查。

以上使用Python类装饰器,实现代码clean的一个小演示。

常用的如Pandas、TensorFlow、Keras等框架,里面的源码,都能看到类似用法,这类语法让Python真正做到clean.

要想掌握Python,需要由表及里,不断深入理解。今年,我大概花费4个多月时间,整理了20个Python核心专题,非常适合入门的读者,下面列举几个章节的目录:





掌握理解这20个专题后,再去看看几本有深度的Python教程,相信那时你就是专家。

以上这20个专题PDF,还有需要的微信我,备注 专题:

不必打赏
给我点个赞
就心满意足了
浏览 50
点赞
评论
收藏
分享

手机扫一扫分享

分享
举报
评论
图片
表情
推荐
点赞
评论
收藏
分享

手机扫一扫分享

分享
举报