快捷導(dǎo)航

Python中通過@classmethod 實現(xiàn)多態(tài)的示例

更新時間：2022年11月26日 16:01:09 作者：Bruce小鬼

這篇文章主要介紹了Python中通過@classmethod 實現(xiàn)多態(tài),python中通常使用對象創(chuàng)建多態(tài)模式，python還支持類創(chuàng)建多態(tài)模式，下面通過一個例子展示它如何實現(xiàn)多態(tài)，需要的朋友可以參考下

通過@classmethod 實現(xiàn)多態(tài)

1.概述

python中通常使用對象創(chuàng)建多態(tài)模式，python還支持類創(chuàng)建多態(tài)模式。下面通過一個例子展示它如何實現(xiàn)多態(tài)。

通過對象創(chuàng)建多態(tài)和類創(chuàng)建多態(tài)開發(fā)模式區(qū)別

對象多態(tài)模式：接收的是一個父類類型對象，然后通過傳入父類的子類對象調(diào)用普通同名方法，實現(xiàn)不同的行為。
類多態(tài)模式：接收的一個父類，然后通過傳入父類的子類調(diào)用同名的類方法，實現(xiàn)不通的行為

2.類方法創(chuàng)建多態(tài)模式示例

2.1.普通模式

先通過一個的示例看下常規(guī)方式發(fā)開的代碼沒有使用多態(tài)時候存在的問題，然后在通過類多態(tài)來優(yōu)化代碼。

下面實現(xiàn)一個讀取數(shù)據(jù)，然后處理數(shù)據(jù)的示例，他有兩條繼承體系。
輸入信息體系：將輸入信息的方式創(chuàng)建為類繼承關(guān)系，可以根據(jù)讀取信息的方式創(chuàng)建不同的InputData子類。
處理數(shù)據(jù)體系：將處理信息的方式創(chuàng)建類繼承關(guān)系，根據(jù)對信息處理的方式創(chuàng)建不同的Worker子類。

然后通過mapreduce函數(shù)組合業(yè)務(wù)執(zhí)行邏輯，最后輸出運行結(jié)果。

# 接收輸入信息類體系
class InputData:
    def read(self):
        raise NotImplementedError

class PathInputData(InputData):
    def __init__(self,path):
        super().__init__()
        self.path = path

    def read(self):
        with open(self.path) as f:
            return f.read()

# 處理信息類體系
class Worker:
    def __init__(self, input_data):
        self.input_data = input_data
        self.result = None

    def map(self):
        raise NotImplementedError

    def reduce(self, other):
        raise NotImplementedError

class LineCountWorker(Worker):
    def map(self):
        data = self.input_data
        self.result = data.count('/n')

    def reduce(self, other):
        self.result += other.result

# 組合業(yè)務(wù)
import os
def generate_inputs(data_dit):
    for name in os.listdir(data_dit):
        # 讀文件內(nèi)容
        yield PathInputData(os.path.join(data_dit, name))

def create_workers(input_list):
    workers = []
    for input_data in input_list:
        # 處理數(shù)據(jù)
        workers.append(LineCountWorker(input_data))
    return workers

from threading import Thread
def execute(workers):
    threads = [Thread(target=w.map) for w in workers]
    for thread in threads: thread.start()
    for thread in threads: thread.join()

    first, *rest = workers
    for worker in rest:
        first.reduce(worker)
    return first.result

def mapreduce(data_dir):
    inputs = generate_inputs(data_dir)
    workers = create_workers(inputs)
    return execute(workers)

import os
import random
def write_test_files(tmpdir):
    os.makedirs(tmpdir)
    for i in range(100):
        with open(os.path.join(tmpdir, str(i)), 'w') as f:
            f.write('\n' * random.randint(0, 100))
tmpdir = 'test_inputs'
write_test_files(tmpdir)
result = mapreduce(tmpdir)
print(f'There are  {result} lines')

上面接收信息處理信息的示例存在兩個問題：

mapreduce函數(shù)通用性不好，不易擴展。例如現(xiàn)在創(chuàng)建了新的InputData、Worker子類，那么就要創(chuàng)建一個新的mapreduce函數(shù)來組合新的業(yè)務(wù)流程，這樣會導(dǎo)致重復(fù)的代碼越來越多，不利于維護。
python不能向java語言可以在一個類中以重載的形式創(chuàng)建多個構(gòu)造器，創(chuàng)建一個構(gòu)造器多態(tài)讓子類可以根據(jù)不同的構(gòu)造器接收不同的參數(shù)。而python不能這么做，因為python的類只能有一個構(gòu)造方法(init)，所以沒有辦法為不同的子類提供多種形式的形參構(gòu)造器，在繼承中也沒有辦法要求所有的子類都只接收只有一種方式參數(shù)的構(gòu)造方法。因為子類要根據(jù)自身的特點接收不同類型的參數(shù)。

這個問題最好能夠通過類方法多態(tài)來解決，這種多態(tài)與實例方法多態(tài)很像，只不過他針對的是類，而不是這些類的對象。

2.2.類方法多態(tài)重構(gòu)業(yè)務(wù)

類方法多態(tài)的實現(xiàn)非常簡單，下面將代碼中關(guān)鍵的點提煉出來。

在父類中創(chuàng)建一個通用的方法，不需要實現(xiàn)任何業(yè)務(wù)邏輯，然后讓子類中重寫該方法，實現(xiàn)不同的行為。在方法上使用@classmethod裝飾器聲明為類方法，方便調(diào)用時候可以以類調(diào)用，而不是實例對象調(diào)用，通過子類重寫父類的類方法就是類方法多態(tài)。
在GenericInputData類中定義一個類方法(generate_inputs)用來解決python只有一個構(gòu)造方法不能實現(xiàn)多種方式接收參數(shù)的弊端，子類通過重寫該方法，實現(xiàn)接收多種參數(shù)行為。

class GenericInputData:
    def read(self):
        raise NotImplementedError
	# 創(chuàng)建一個多態(tài)的類方法，讓子類實現(xiàn)不同的功能
    @classmethod
    def generate_inputs(cls, config):
        raise NotImplementedError

class PathInputData(GenericInputData):
    def __init__(self, path):
        super().__init__()
        self.path = path

    def read(self):
        with open(self.path) as f:
            return f.read()
	# 子類重寫父類的類方法
    @classmethod
    def generate_inputs(cls, config):
        data_dir = config['data_dir']
        for name in os.listdir(data_dir):
            yield cls(os.path.join(data_dir, name))


class GenericWorker:
    def __init__(self, input_data):
        self.input_data = input_data
        self.result = None

    def map(self):
        raise NotImplementedError

    def reduce(self, other):
        raise NotImplementedError

    @classmethod
    def create_workers(cls, input_class, config):
        workers = []
        for input_data in input_class.generate_inputs(config):
            workers.append(cls(input_data))
        return workers

class LineCountWorker(GenericWorker):
    def map(self):
        data = self.input_data.read()
        self.result = data.count('\n')

    def reduce(self, other):
        self.result += other.result

# 定義的形參類型為父類，實現(xiàn)了類方法多態(tài)
def mapreduce(worker_class, input_class, config):
    workers = worker_class.create_workers(input_class, config)
    return execute(workers)
config = {'data_dir': tmpdir}
result = mapreduce(LineCountWorker, PathInputData, config)
print(f'There are {result} lines')

通過類方法多態(tài)重構(gòu)代碼后，mapreduce函數(shù)支持了多態(tài)，它接收的是一個父類類型，根據(jù)傳入的實際子類實現(xiàn)把不同的功能。當(dāng)再擴展GenericInputData、GenericWorker子類后，mapreduce函數(shù)不需要修改代碼，實現(xiàn)了左開右閉原則。

到此這篇關(guān)于Python中通過@classmethod 實現(xiàn)多態(tài)的文章就介紹到這了,更多相關(guān)@classmethod 實現(xiàn)多態(tài)內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: