Python 类装饰器，使用小演示
编辑日期: 2024-11-28 文章阅读: 次
Python 类装饰器，使用小演示
实现同样一个功能，用Java语言可能得50行，而用Python可能只需10行，可能很多读者在没有学Python前，就从用过Python的前辈那里，听说过这个，然后自己也开始去学Python的。 Python的确简洁、优雅，很多读者包括我，都为之着迷。
今天通过一个小例子，再次认识Python的clean之道：
我们想要检查每个参数，确保不能出现整数类型。使用类装饰器，便能做到clean、通用：
import functools
class ValidateParameters:
def __init__(self, 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)
print(concat("a", "n", "b"))
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.
