本文將帶你走進python3.7的新特性dataclass,通過本文你將學會dataclass的使用並避免踏入某些陷阱。
dataclass簡介
dataclass的定義位於PEP-557,根據定義一個dataclass是指“一個帶有預設值的可變的namedtuple”,廣義的定義就是有一個類,它的屬性均可公開訪問,可以帶有預設值並能被修改,而且類中含有與這些屬性相關的類方法,那麼這個類就可以稱為dataclass,再通俗點講,dataclass就是一個含有資料及運算元據方法的容器。
乍一看可能會覺得這個概念不就是普通的class麼,然而還是有幾處不同:
- 相比普通class,dataclass通常不包含私有屬性,資料可以直接訪問
- dataclass的repr方法通常有固定格式,會列印出型別名以及屬性名和它的值
- dataclass擁有
__eq__
和__hash__
魔法方法 - dataclass有著模式單一固定的構造方式,或是需要過載運算子,而普通class通常無需這些工作
基於上述原因,通常自己實現一個dataclass是繁瑣而無聊的,而dataclass單一固定的行為正適合程式為我們自動生成,於是dataclasses
模組誕生了。
配合型別註解語法,我們可以輕鬆生成一個實現了__init__
,__repr__
,__cmp__
等方法的dataclass:
from dataclasses import dataclass
@dataclass
class InventoryItem:
```Class for keeping track of an item in inventory.```
name: str
unit_price: float
quantity_on_hand: int = 0
def total_cost(self) -> float:
return self.unit_price * self.quantity_on_hand
同時使用dataclass也有一些好處,它比namedtuple更靈活。同時因為它是一個常規的類,所以你可以享受繼承帶來的便利。
dataclass的使用
我們分x步介紹dataclass的使用,首先是如何定義一個dataclass。
定義一個dataclass
dataclasses
模組提供了一個裝飾器幫助我們定義自己的資料類:
@dataclass
class Lang:
"""a dataclass that describes a programming language"""
name: str = `python`
strong_type: bool = True
static_type: bool = False
age: int = 28
我們定義了一個描述某種程式語言特性的資料類——Lang
,在接下來的例子中我們都會用到這個類。
在資料類被定義後,會根據給出的型別註解生成一個如下的初始函式:
def __init__(self, name: str=`python`,
strong_type: bool=True,
static_type: bool=False,
age: int=28):
self.name = name
self.strong_type = strong_type
self.static_type = static_type
self.age = age
可以看到初始化操作都已經自動生成了,讓我們試用一下:
>>> Lang()
Lang(name=`python`, strong_type=True, static_type=False, age=28)
>>> Lang(`js`, False, False, 23)
Lang(name=`js`, strong_type=False, static_type=False, age=23)
>>> Lang(`js`, False, False, 23) == Lang()
False
>>> Lang(`python`, True, False, 28) == Lang()
True
例子中可以看出__repr__
和__eq__
方法也已經為我們生成了,如果沒有其他特殊要求的話這個dataclass已經具備了投入生產環境的能力,是不是很神奇?
深入dataclass裝飾器
dataclass的魔力源泉都在dataclass
這個裝飾器中,如果想要完全掌控dataclass的話那麼它是你必須瞭解的內容。
裝飾器的原型如下:
dataclasses.dataclass(*, init=True, repr=True, eq=True, order=False, unsafe_hash=False, frozen=False)
dataclass
裝飾器將根據類屬性生成資料類和資料類需要的方法。
我們的關注點集中在它的kwargs
上:
key | 含義 |
---|---|
init | 指定是否自動生成__init__ ,如果已經有定義同名方法則忽略這個值,也就是指定為True也不會自動生成 |
repr | 同init,指定是否自動生成__repr__ ;自動生成的列印格式為class_name(arrt1:value1, attr2:value2, ...) |
eq | 同init,指定是否生成__eq__ ;自動生成的方法將按屬性在類內定義時的順序逐個比較,全部的值相同才會返回True |
order | 自動生成__lt__ ,__le__ ,__gt__ ,__ge__ ,比較方式與eq相同;如果order指定為True而eq指定為False,將引發ValueError ;如果已經定義同名函式,將引發TypeError |
unsafehash | 如果是False,將根據eq和frozen引數來生成__hash__ :1. eq和frozen都為True, __hash__ 將會生成2. eq為True而frozen為False, __hash__ 被設為None 3. eq為False,frozen為True, __hash__ 將使用超類(object)的同名屬性(通常就是基於物件id的hash)當設定為True時將會根據類屬性自動生成 __hash__ ,然而這是不安全的,因為這些屬性是預設可變的,這會導致hash的不一致,所以除非能保證物件屬性不可隨意改變,否則應該謹慎地設定該引數為True |
frozen | 設為True時對field賦值將會引發錯誤,物件將是不可變的,如果已經定義了__setattr__ 和__delattr__ 將會引發TypeError |
有預設值的屬性必須定義在沒有預設值的屬性之後,和對kw引數的要求一樣。
上面我們偶爾提到了field的概念,我們所說的資料類屬性,資料屬性實際上都是被field的物件,它代表著一個資料的實體和它的元資訊,下面我們瞭解一下dataclasses.field
。
資料類的基石——dataclasses.field
先看下field的原型:
dataclasses.field(*, default=MISSING, default_factory=MISSING, repr=True, hash=None, init=True, compare=True, metadata=None)
通常我們無需直接使用,裝飾器會根據我們給出的型別註解自動生成field,但有時候我們也需要定製這一過程,這時dataclasses.field
就顯得格外有用了。
default和default_factory引數將會影響預設值的產生,它們的預設值都是None,意思是呼叫時如果為指定則產生一個為None的值。其中default是field的預設值,而default_factory控制如何產生值,它接收一個無引數或者全是預設引數的callable
物件,然後用呼叫這個物件獲得field的初始值,之後再將default(如果值不是MISSING)複製給callable
返回的這個物件。
舉個例子,對於list,當複製它時只是複製了一份引用,所以像dataclass裡那樣直接複製給例項的做法的危險而錯誤的,為了保證使用list時的安全性,應該這樣做:
@dataclass
class C:
mylist: List[int] = field(default_factory=list)
當初始化C
的例項時就會呼叫list()
而不是直接複製一份list的引用:
>>> c1 = C()
>>> c1.mylist += [1,2,3]
>>> c1.mylist
[1, 2, 3]
>>> c2 = C()
>>> c2.mylist
[]
資料汙染得到了避免。
init引數如果設定為False,表示不為這個field生成初始化操作,dataclass提供了hook——__post_init__
供我們利用這一特性:
@dataclass
class C:
a: int
b: int
c: int = field(init=False)
def __post_init__(self):
self.c = self.a + self.b
__post_init__
在__init__
後被呼叫,我們可以在這裡初始化那些需要前置條件的field。
repr參數列示該field是否被包含進repr的輸出,compare和hash參數列示field是否參與比較和計算hash值。metadata不被dataclass自身使用,通常讓第三方元件從中獲取某些元資訊時才使用,所以我們不需要使用這一引數。
如果指定一個field的型別註解為dataclasses.InitVar
,那麼這個field將只會在初始化過程中(__init__
和__post_init__
)可以被使用,當初始化完成後訪問該field會返回一個dataclasses.Field
物件而不是field原本的值,也就是該field不再是一個可訪問的資料物件。舉個例子,比如一個由資料庫物件,它只需要在初始化的過程中被訪問:
@dataclass
class C:
i: int
j: int = None
database: InitVar[DatabaseType] = None
def __post_init__(self, database):
if self.j is None and database is not None:
self.j = database.lookup(`j`)
c = C(10, database=my_database)
這個例子中會返回c.i
和c.j
的資料,但是不會返回c.database
的。
一些常用函式
dataclasses
模組中提供了一些常用函式供我們處理資料類。
使用dataclasses.asdict
和dataclasses.astuple
我們可以把資料類例項中的資料轉換成字典或者元組:
>>> from dataclasses import asdict, astuple
>>> asdict(Lang())
{`name`: `python`, `strong_type`: True, `static_type`: False, `age`: 28}
>>> astuple(Lang())
(`python`, True, False, 28)
使用dataclasses.is_dataclass
可以判斷一個類或例項物件是否是資料類:
>>> from dataclasses import is_dataclass
>>> is_dataclass(Lang)
True
>>> is_dataclass(Lang())
True
dataclass繼承
python3.7引入dataclass的一大原因就在於相比namedtuple,dataclass可以享受繼承帶來的便利。
dataclass
裝飾器會檢查當前class的所有基類,如果發現一個dataclass,就會把它的欄位按順序新增進當前的class,隨後再處理當前class的field。所有生成的方法也將按照這一過程處理,因此如果子類中的field與基類同名,那麼子類將會無條件覆蓋基類。子類將會根據所有的field重新生成一個建構函式,並在其中初始化基類。
看個例子:
@dataclass
class Python(Lang):
tab_size: int = 4
is_script: bool = True
>>> Python()
Python(name=`python`, strong_type=True, static_type=False, age=28, tab_size=4, is_script=True)
@dataclass
class Base:
x: float = 25.0
y: int = 0
@dataclass
class C(Base):
z: int = 10
x: int = 15
>>> C()
C(x=15, y=0, z=10)
Lang
的field被Python
繼承了,而C
中的x
則覆蓋了Base
中的定義。
沒錯,資料類的繼承就是這麼簡單。
總結
合理使用dataclass將會大大減輕開發中的負擔,將我們從大量的重複勞動中解放出來,這既是dataclass的魅力,不過魅力的背後也總是有陷阱相伴,最後我想提幾點注意事項:
- dataclass通常情況下是unhashable的,因為預設生成的
__hash__
是None
,所以不能用來做字典的key,如果有這種需求,那麼應該指定你的資料類為frozen dataclass - 小心當你定義了和
dataclass
生成的同名方法時會引發的問題 - 當使用可變型別(如list)時,應該考慮使用
field
的default_factory
- 資料類的屬性都是公開的,如果你有屬性只需要初始化時使用而不需要在其他時候被訪問,請使用
dataclasses.InitVar
只要避開這些陷阱,dataclass一定能成為提高生產力的利器。