带有抽象类的面向对象的编程

Question

我想实现以下内容：

def do_something(request):
    company_name = request.get("company_name", DEFAULT_COMPANY)
    data = request.get("data")
    response = transform_data_according_to(data, company_name)
    return response

我为此做了以下操作：

class Transform(ABC):
    def __init__(self, data):
        self.data = data

    @abstractmethod
    def transform(self):
        pass

class CompanyA(Transform):
    def transform(self):
        # do_transformation
        return transformed_data

def do_something(request):
    company_name = request.get("company_name", DEFAULT_COMPANY)
    data = request.get("data")
    if company_name == CompanyA:
        response = CompanyA.transform(data)
    return response

相反，我想使用正确的对象的原则来做类似的事情：

def do_something(request):
    company_name = request.get("company_name", DEFAULT_COMPANY)
    data = request.get("data")
    response = Transform(data, company_name)
    return response

我想知道我可能会在所需的方法和实现方法方面想在哪里思考。实施方法是否正确，在这种情况下，如果其他检查可以增长很大。

Answer 1

感谢 teraflop

简单，惯用的方法可以在Python中进行此操作。词典中的变换子类：

transform_classes = {
    "CompanyA": CompanyA,
    # ...
}

def do_something(request):
    company_name = request.get("company_name", DEFAULT_COMPANY)
    data = request.get("data")
    transformer = transform_classes[company_name](data)
    return transformer.transform()

如果您希望更严格地面向对象，则可以将字典包裹在对象（例如transform lookupbyname）中，而不是直接访问它。

您还可以自动使用各种元编程魔术来自动构建字典，而无需明确命名每个子类。例如，这将收集当前源文件中的所有transform子类：

transform_classes = {
    k:v for k,v in globals().items() 
        if isinstance(v, type) and issubclass(v, Transform) and v != Transform
}