Question

我有一个基类和多个继承自它的子类。我需要根据提供的配置文件实例化正确的子类。现在，执行此操作的一种方法是使用if，else语句并检查配置文件以实例化子类，但这似乎是不好的编程代码。此外，稍后如果我添加更多子类，则if-else链会变得很长。有人可以建议一种更好的方法吗？

我有一个模板代码，而不是配置文件，我使用命令行参数来执行相同的操作。

class Shape(object):
    pass

class Rectangle(Shape):
    pass

class Circle(Shape):
    pass

class Polygon(Shape):
    pass

import argparse
if __name__ == '__main__':
    parser = argparse.ArgumentParser()
    parser.add_argument('-s', '--shape', help='Provide the shape')

    args = parser.parse_args()

    if args.shape == 'circle':
        shape = Circle()
        print(shape.__class__.__name__)
    elif args.shape == 'rectangle':
        shape = Rectangle()
        print(shape.__class__.__name__)
    elif args.shape == 'polygon':
        shape = Polygon()
        print(shape.__class__.__name__)
    else:
        raise Exception("Shape not defined")

Answer 1

您可以将所有类放在这样的字典对象中

my_shapes = { "rectangle" : Rectangle, "circle": Circle, "polygon": Polygon }
args = parser.parse_args()
if args.shape in my_shapes:
    shape = my_shapes[args.shape]() #Here you will do the same thing that the if else 
else:
    raise Exception("Shape not defined")

如何以更好的方式在配置文件的基础上实例化不同的子类？

1 个答案: