在类中使用关键字来调用特定的方法

Question

假设一个Python类有不同的方法，根据用户指定的不同，在主函数calculate()中执行不同的方法。

在下面的示例中，用户需要指定关键字参数'methodOne' 或'methodTwo'。如果没有指定或指定了不正确的关键字，则默认为'methodOne'。

class someClass(object):
    def __init__(self,method=None):
        methodList = ['methodOne','methodTwo']
        if method in methodList:
            self.chosenMethod = method
        else:
            self.chosenMethod = self.methodOne

    def methodOne(self):
        return 1

    def methodTwo(self):
        return 2

    def calculate(self):
        return self.chosenMethod()

上面显然不起作用，因为method 是一个字符串而不是一个函数。如何根据我的关键字参数method 选择self.methedOne() 或self.methedOne()？原则上，以下工作：

def __init__(self,method=None):
    if method == 'methodOne':
        self.chosenMethod = self.methodOne
    elif method == 'methodTwo':
        self.chosenMethod = self.methodTwo
    else:
        self.chosenMethod = self.methodOne

但如果我有两个以上的方法，这会变得相当难看。有没有办法做到这一点类似于我的原始代码？

Answer 1

您可以为此使用getattr()：

class someClass(object):
    def __init__(self,method=None):
        methodList = ['methodOne','methodTwo']
        if method in methodList:
            self.chosenMethod = method
        else:
            self.chosenMethod = self.methodOne

    def methodOne(self):
        return 1

    def methodTwo(self):
        return 2

    def calculate(self):
        return getattr(self, self.chosenMethod)()

x = someClass(method='methodOne')
print x.calculate()
>>> 1

Answer 2

您可以使用getattr 来获取类对象上的实际方法。

class someClass(object):
    def __init__(self,method=None):
        # store it with the object so we can access it later in calculate method
        self.method = method

    def methodOne(self):
        return 1

    def methodTwo(self):
        return 2

    def calculate(self):
        # get the actual method from the string here
        # if no such method exists then use methodOne instead
        return getattr(self, self.method, self.methodOne)()


> someClass('methodOne').calculate()
# 1

> someClass('methodTwo').calculate()
# 2