Python中attrs如何提高面向?qū)ο缶幊绦?/h1>

發(fā)布時間：2021-09-27 10:45:32 來源：億速云 閱讀：167 作者：小新 欄目：開發(fā)技術(shù)

小編給大家分享一下Python中attrs如何提高面向?qū)ο缶幊绦?，相信大部分人都還不怎么了解，因此分享這篇文章給大家參考一下，希望大家閱讀完這篇文章后大有收獲，下面讓我們一起去了解一下吧！

前言：

Python是面向?qū)ο蟮恼Z言，一般情況下使用面向?qū)ο缶幊虝沟瞄_發(fā)效率更高，軟件質(zhì)量更好，并且代碼更易于擴(kuò)展，可讀性和可維護(hù)性也更高。但是如果在一個較大的項目中，如果實體類非常多并且有非常復(fù)雜的屬性，你就會逐漸覺得Python的類寫起來是真·“累”。為什么這樣說，看下下面這個Box類，屬性有長（length）、寬（width）、高（hight）：

class Box:
     def __init__(self, length, width, hight):
         
         self.length = length
         self.width = width
         self.hight = hight

這樣倒沒有什么問題，而且是符合Python規(guī)范的寫法，初始化函數(shù)內(nèi)每一個參數(shù)都需要用self.xxx = xxx進(jìn)行賦值，參數(shù)少了還好，但是如果參數(shù)過多了，只是參數(shù)賦值操作都夠?qū)懸粫牧?，如果一個項目中類也有很多，那就真的要麻木了。

而且我們知道，在Python中，想要自定義對象本身的打印輸出結(jié)果的時候，需要在它的類中實現(xiàn)__repr__() 方法，例如：

def __repr__(self):
        return '{}(length={}, width={}, hight={})'.format(
            self.__class__.__name__, self.length, self.width, self.hight)

實現(xiàn)了__repr__() 方法，當(dāng)我們打印對象本身的時候，才會輸出我們自定義的字符。

box = Box(20, 15, 15)
print(box)
# 結(jié)果輸出為 Box(length=20, width=15, hight=15)

但是有時會因為麻煩，我們不愿意去實現(xiàn)__repr__()方法，但是不實現(xiàn)打印結(jié)果又不友好，就陷入了糾結(jié)之中。

如果我么想要實現(xiàn)對象比較，有時候需要判斷2個對象是否相等或者比較大小，就要實現(xiàn)__eq__() 、 __lt__() 、__gt__()等各種方法來實現(xiàn)對象之間的比較，例如：

def __eq__(self, other):
    if not isinstance(other, self.__class__):
        return NotImplemented
    return (self.length, self.width, self.hight) == (
        other.length, other.width, other.hight)

這樣我們又需要實現(xiàn)這幾個用于比較的方法。 比如說我們已經(jīng)實現(xiàn)了上面說過的所有方法，然后又突然添加一個屬性結(jié)實度hardness，那么整個類的方法都需要修改，這是非常折磨人的。 那么有沒有一種方法，可以在創(chuàng)建類的時候自動添加上類似于上面提到的這些東西，答案是有的，那就是我們接下來要介紹的attrs模塊，它可以幫助我們很方便的定義類。

1、attrs的使用

我們可以使用pip install attrs進(jìn)行安裝。

然后將上面的代碼改造一下：

from attr import attrs, attrib


@attrs
class Box:
    length = attrib(type=int, default=0)
    width = attrib(type=int, default=0)
    hight = attrib(type=int, default=0)


box1 = Box(20, 15, 15)
print(box1)
box2 = Box(30, 20, 20)
print(box2 == box1)
print(box2 > box1)

用模塊內(nèi)的attrs修飾了Box類，再使用attrib定義所有的屬性，同時指定了屬性的類型和默認(rèn)值。而且我們沒有實現(xiàn)任何一個上面所提到的方法，但是確實實現(xiàn)了所有的功能。 現(xiàn)在如果我們添加一個屬性顏色color，這個屬性不參與對象的比較，但是打印的時候要輸出，添加一個屬性結(jié)實度hardness，這個屬性參與對象的比較，但是打印對象的時候不輸出，就非常簡單了：

from attr import attrs, attrib


@attrs
class Box:
    length = attrib(type=int, default=0)
    width = attrib(type=int, default=0)
    hight = attrib(type=int, default=0)
    color = attrib(repr=True, cmp=False, default=(0, 0, 0))
    hardness = attrib(repr=False, cmp=True, default=0)


box1 = Box(20, 15, 15, (255, 255, 255), 80)
print("box1:", box1)
box2 = Box(20, 15, 15, (255, 255, 0), 100)
print("box2:", box2)
print(box2 == box1)
print(box2 > box1)

執(zhí)行結(jié)果為：

這里我們只重點說一下驗證器和轉(zhuǎn)換器，其他的參數(shù)都很好理解。

2、驗證器

有時候在設(shè)置一個屬性的時候必須要滿足某個條件，比如上面的顏色color屬性，我們使用的是RGB三原色方式，例如黑色是（255，255，255），對于這種情況，我們就需要驗證屬性是否合法。

例如：

def color_is_valid(instance, attr, value):
    if not isinstance(value, set):
        raise ValueError(f"參數(shù){attr.name}:{value}不合法！")
    for i in value:
        if not 0 <= i<= 255:
            raise ValueError(f"參數(shù){attr.name}:{value}不合法！")




@attrs
class Box:
    length = attrib(type=int, default=0)
    width = attrib(type=int, default=0)
    hight = attrib(type=int, default=0)
    color = attrib(repr=True, cmp=False, validator=color_is_valid, default=(0, 0, 0))
    hardness = attrib(repr=False, cmp=True, default=0)


box1 = Box(20, 15, 15, (255, 255, 260), 80)

執(zhí)行結(jié)果為：

而且attrs模塊也提供了許多內(nèi)置驗證器，這里就不做贅述了。

3、轉(zhuǎn)換器

轉(zhuǎn)換器主要做一些值的處理和轉(zhuǎn)換，增加類的容錯性，比如一個屬性接收的是int類型，我們想要的是傳入了數(shù)字字符串也不會報錯，那我們就可以增加轉(zhuǎn)換器，將字符串自動轉(zhuǎn)為數(shù)字，例如：

def to_int(value):
    try:
        return int(value)
    except:
        return None


@attrs
class Box:
    length = attrib(type=int, default=0, converter=to_int)
    width = attrib(type=int, default=0, converter=to_int)
    hight = attrib(type=int, default=0)
    color = attrib(repr=True, cmp=False, default=(0, 0, 0))
    hardness = attrib(repr=False, cmp=True, default=0)


box1 = Box("20", "15", 15, (255, 255, 255), 80)
print("box1:", box1)
box2 = Box("2a", 15, 15, (255, 255, 0), 100)
print("box2:", box2)

以上是“Python中attrs如何提高面向?qū)ο缶幊绦省边@篇文章的所有內(nèi)容，感謝各位的閱讀！相信大家都有了一定的了解，希望分享的內(nèi)容對大家有所幫助，如果還想學(xué)習(xí)更多知識，歡迎關(guān)注億速云行業(yè)資訊頻道！