Python類中怎么定義多個(gè)構(gòu)造器方法重載與泛方法

這篇文章主要介紹“Python類中怎么定義多個(gè)構(gòu)造器方法重載與泛方法”，在日常操作中，相信很多人在Python類中怎么定義多個(gè)構(gòu)造器方法重載與泛方法問(wèn)題上存在疑惑，小編查閱了各式資料，整理出簡(jiǎn)單好用的操作方法，希望對(duì)大家解答”P(pán)ython類中怎么定義多個(gè)構(gòu)造器方法重載與泛方法”的疑惑有所幫助！接下來(lái)，請(qǐng)跟著小編一起來(lái)學(xué)習(xí)吧！

什么是“泛方法”？

由多個(gè)方法組成的方法，這些方法為不同的類型實(shí)現(xiàn)相同的操作。

舉個(gè)栗子 

現(xiàn)在有個(gè)需求，需要你通過(guò)以下幾種方式創(chuàng)建一個(gè)自定義的日期類（CustomDate）：

• 時(shí)間戳

• 年、月、日（包含三個(gè)整數(shù)的元組）

• ISO 格式的字符串

• Datetime 類

一般實(shí)現(xiàn)

from datetime import date, datetime
class CustomDate:
    def __init__(self, arg):
        if isinstance(arg, (int, float)):
            self.__date = date.fromtimestamp(arg)
        elif isinstance(arg, tuple) and len(arg) == 3 and all(map(lambda x: isinstance(x, int), arg):
            self.__date = date(*arg)
        elif isinstance(arg, str):
            self.__date = date.fromisoformat(arg)
        elif isinstance(arg, datetime):
            self.__date = datetime.date()
        else:
            raise TypeError("could not create instance from " + type(arg).__name__)
    @property
    def date():
        return self.__date

注：這里暫不討論傳入的日期/時(shí)間戳合不合法，僅僅只對(duì)類型做大致判斷。

有沒(méi)有更好的方式？

我們可以將不同的構(gòu)建方式拆分為多個(gè)方法，并利用 functools 中的 singledispatchmethod 裝飾器來(lái)根據(jù)傳入的參數(shù)類型決定調(diào)用哪個(gè)方法。

from datetime import date, datetime
from functools import singledispatchmethod
class CustomDate:
    @singledispatchmethod
    def __init__(self, arg):
        raise TypeError("could not create instance from " + type(arg).__name__)
    @__init__.register(int)
    @__init__.register(float)
    def __from_timestamp(self, arg):
        self.__date = date.fromtimestamp(arg)
    @__init__.register(tuple)
    def __from_tuple(self, arg):
        if len(arg) == 3 and all(map(lambda x: isinstance(x, int), arg)):
            self.__date = date(*arg)
        else:
            raise ValueError("could not create instance from a malformed tuple")
    @__init__.register(str)
    def __from_isoformat(self, arg):
        self.__date = date.fromisoformat(arg)
    @__init__.register(datetime)
    def __from_datetime(self, arg):
        self.__date = arg.date()
    @property
    def date(self):
        return self.__date

這樣一來(lái)，我們便能將每種參數(shù)類型的初始化獨(dú)立成一個(gè)個(gè)的方法了。

缺點(diǎn)

#1 單分派

在調(diào)用期間應(yīng)該使用哪個(gè)方法實(shí)現(xiàn)由分派算法決定。如果該算法只基于單個(gè)參數(shù)的類型來(lái)決定使用哪個(gè)方法實(shí)現(xiàn)，則稱其為單分派。

singledispatchmethod 就是就是單分派的。也就是說(shuō)，只有第一個(gè)參數(shù)會(huì)作為考量。這在實(shí)際業(yè)務(wù)中是遠(yuǎn)遠(yuǎn)不足的。

#2 不支持 typing

然而，如上，對(duì)元組中元素類型判斷還是需要我們用 if/else 實(shí)現(xiàn)。也就是說(shuō)，我們不能使用 typing.Tuple[int, int, int]。

作為一種折中的方案，或許我們可以定義一個(gè) ThreeIntTuple 類來(lái)對(duì)其進(jìn)行限定，將這些判斷從 CustomDate 類中隔離開(kāi)來(lái)。

這里僅提供一個(gè)思路讓大家參考，我就不實(shí)現(xiàn)了（因?yàn)槲覀冇懈玫姆绞?xD）。

替代方案：multimethod 庫(kù)

這個(gè)庫(kù)不是標(biāo)準(zhǔn)庫(kù)之一，需要通過(guò) pip 安裝：

pip install multimethod

優(yōu)勢(shì)

multimethod 采用的是多分派算法，能更好地滿足更復(fù)雜的場(chǎng)景。此外，該庫(kù)對(duì) typing 中的類型也有不錯(cuò)的支持。

更更好的實(shí)踐方式

回到上面的問(wèn)題，我們可以這么改進(jìn)：

• 使用 multimethod 方法來(lái)替代 singledispatchmethod；

• 使用 Tuple[int, int, int] 來(lái)替代 tuple，不再需要手動(dòng)校驗(yàn)元組的長(zhǎng)度和元素類型了；

from datetime import date, datetime
from typing import Tuple, Union
from multimethod import multimethod
class CustomDate:
    @multimethod
    def __init__(self, arg):
        raise TypeError("could not create instance from " + type(arg).__name__)
    @__init__.register
    def __from_timestamp(self, arg: Union[int, float]):
        self.__date = date.fromtimestamp(arg)
    @__init__.register
    def __from_tuple(self, arg: Tuple[int, int, int]):
        self.__date = date(*arg)
    @__init__.register
    def __from_isoformat(self, arg: str):
        self.__date = date.fromisoformat(arg)
    @__init__.register
    def __from_datetime(self, arg: datetime):
        self.__date = arg.date()
    @property
    def date(self):
        return self.__date

究極好的實(shí)踐方式（真正的方法重載）

在此之前，先問(wèn)大家一個(gè)簡(jiǎn)單的問(wèn)題（這跟我們之后的內(nèi)容有很大的聯(lián)系）：

class A:
    def a(self):
        print(1)
    def a(self):
        print(2)
A().a()

以上這段代碼會(huì)輸出什么？還是會(huì)拋出錯(cuò)誤？

輸出 2。

在 Python 中，如果定義了重名的方法，最后一個(gè)方法是會(huì)覆蓋掉之前的方法的。

但你或許不知，我們可以通過(guò)元類（metaclass）來(lái)改變這一行為：

class MetaA(type):
    class __prepare__(dict):
        def __init__(*args):
            pass
        def __setitem__(self, key, value):
            if self.get('a'):  # Line 7
                super().__setitem__('b', value)  # Line 8
            else:	
                super().__setitem__(key, value)
class A(metaclass=MetaA):
    def a(self):
        print(1)
    def a(self):
        print(2)
A().a()  # => 1
A().b()  # => 2  # Line 22

在第 7 和第 8 行，我們將重名的 a 方法改名為 b，并在第 22 行成功地調(diào)用它了。

multimethod 的維護(hù)者們很好地運(yùn)用了這一點(diǎn)，對(duì)重名的方法進(jìn)行了處理，以達(dá)到一種“特殊的效果”。

回到正題，我們可以做出如下改進(jìn)：

• 將 multimethod.multidata 設(shè)置為 CustomDate 類的元類；

• 將所有方法命名為 __init__。

from datetime import date, datetime
from typing import Tuple, Union
from multimethod import multimeta
class CustomDate(metaclass=multimeta):
    def __init__(self, arg: Union[int, float]):
        self.__date = date.fromtimestamp(arg)
    def __init__(self, arg: Tuple[int, int, int]):
        self.__date = date(*arg)
    def __init__(self, arg: str):
        self.__date = date.fromisoformat(arg)
    def __init__(self, arg: datetime):
        self.__date = arg.date()
    def __init__(self, arg):
        raise TypeError("could not create instance from " + type(arg).__name__)
    @property
    def date(self):
        return self.__date

從效果上來(lái)看，這完全和靜態(tài)語(yǔ)言的方法重載一模一樣！

到此，關(guān)于“Python類中怎么定義多個(gè)構(gòu)造器方法重載與泛方法”的學(xué)習(xí)就結(jié)束了，希望能夠解決大家的疑惑。理論與實(shí)踐的搭配能更好的幫助大家學(xué)習(xí)，快去試試吧！若想繼續(xù)學(xué)習(xí)更多相關(guān)知識(shí)，請(qǐng)繼續(xù)關(guān)注億速云網(wǎng)站，小編會(huì)繼續(xù)努力為大家?guī)?lái)更多實(shí)用的文章！