Question

我有一堂课：

class BaseModel:

    def __init__(self):
        pass

    def read_data(self):
        df = ...
        return df

    def transform_input(self, df):
        df = ...
        return df

    def execute(self, df):
        df = ...
        return df

    def run(self):
        data = self.read_data()
        data = self.transform_input(data)
        data = self.execute(data)

如何避免这些方法一个接一个地调用？是不是可以这样做：

data = self.read_data().transform_input().execute()

？

是否有可能以某种方式链接这些方法并解决使用此方法链传递参数（数据）的问题？

Answer 1

这些不是类方法，它们是实例方法。对于方法链接，每个方法必须返回一个实现链中下一个方法的对象。由于您的所有方法都是BaseModel的实例方法，因此您需要返回BaseModel（或其后代）的实例。这显然意味着您无法返回df（因为大概df不是BaseClass的对象。将其存储在实例中并在链的末尾检索它。

class BaseModel:

    def __init__(self):
        pass

    def read_data(self):
        self.df = ...
        return self

    def transform_input(self):
        self.df = ...
        return self

    def execute(self):
        self.df = ...
        return self

    def run(self):
        data = self.read_data().transform_input().execute().df

对于实例方法和类方法之间的区别，this answer提供了一个很好的概述。

Answer 2

这些是实例方法，而不是类方法。

您似乎在这里混合了两种对象类型。您需要回答的第一个问题是这个类是否有状态。如果它是有状态的，那么你的数据通常是对象本身的一部分，在这种情况下，方法链接非常简单。你只需从你的方法中归还自己并将其链条带走。

class BaseModel:

    def __init__(self, data):
        self.data = data

    @classmethod
    def read_data(cls):
        data = #Some data read from somewhere
        return cls(data)

    def transform_input(self):
        #Transform your data
        return self

    def execute(self):
        #Do Something with data
        return self

my_object = BaseModel().transform_input().execute()

就我个人而言，我不喜欢方法链。

如果您需要无状态对象，那么您可以按照您在示例中所做的方式进行链接，但是您可以执行其他操作。

class BaseModel:

    def __init__(self):
        pass

    def read_data(self):
        data = #Some data read from somewhere
        return data

    def transform_input(self, data):
        #Transform your data
        return data

    def execute(self, data):
        #Do Something with data
        return data

my_object = BaseModel()
data = my_object.execute(my_object.transform_data(my_object.read_data()))

我也觉得这很难读，但这就是你无国界的方式

我个人觉得在自己的行上调用每个方法更具可读性

Python类方法链接

2 个答案: