attr可看作是一个类装饰器但又不仅仅是简单的装饰器,他使得我们编写类变得更加简单轻松。下面先通过一个简单的例子来见识下attr的强大吧。
现在我们需要编写一个类,有a,b,c两个属性,正常的写法:
class A(object): def __init__(self, a, b, c): self.a = a self.b = b self.c = c
看起来也没那么复杂嘛。好了,现在领导说类的打印输出不好看,又不能进行比较,行吧,有需求就加吧。
from functools import total_ordering@total_orderingclass A(object): def __init__(self, a, b, c): self.a = a self.b = b def __repr__(self): return "ArtisanalClass(a={}, b={})".format(self.a, self.b) def __eq__(self, other): if not isinstance(other, self.__class__): return NotImplemented return (self.a, self.b) == (other.a, other.b) def __lt__(self, other): if not isinstance(other, self.__class__): return NotImplemented return (self.a, self.b) < (other.a, other.b)
虽然看起来有点复杂,但是借助total_ordering,也算是以尽量少的代码实现了所需功能了吧。
但是,有没有更简洁的方式呢?让我们来看看借助于attr的实现:
import attr @attr.sclass A(object): a = attr.ib() b = attr.ib()
更多学习内容,请点击。