将对象类型作为参数传递

Question

我有一些代码希望以比我的直觉更优雅的方式实现。 我会尽我所能来描述我正在尝试的东西。

class Fruit():
    pass

class Apple(Fruit):
    pass

class Orange(Fruit):
    pass

def create_fruit(fruit_type):
    test = ???? # code here to create instance of fruit of desired type called test

好的，所以希望这段代码有意义。 我在一个模块中有一个函数，它需要一堆参数来创建一个类的实例。 理想情况下，我想传递一个参数，说明要创建的类的类型（但它们都是同一个超类的实例或子类）。 每个子类的参数都是相同的（截至目前）。

我可能可以很容易地用 if 语句做一些事情并一起破解（比如， if fruit_type==1 , test=Apple() , if f fruit_type == 2 , test=Orange()等...），但试图作为一名python程序员进行改进，我想知道是否有更好的方法来做到这一点。 我已经简要地阅读了装饰器和函数式编程（尽管它对我来说仍然很抽象，并且需要更多时间来理解），所以也许这与此相同？

Answer 1

如果您只是使用类名调用 create_fruit 然后实例化参数会怎样：

def create_fruit(fruit_type):
    test = fruit_type()

create_fruit(Apple)

（编辑以将赋值添加到“测试”变量）或者你也可以做这样的事情，这实际上允许你在 create_fruit 之外对你创建的水果做一些事情：

def create_fruit(fruit_type):
    return fruit_type()

test = create_fruit(Apple)
test.bite()

Answer 2

您可以使用检查找到可用的类并从那里创建实例

import inspect
import sys

class Fruit():
    pass

class Apple(Fruit):
    pass

class Orange(Fruit):
    pass

clsmembers = dict(inspect.getmembers(sys.modules[__name__], inspect.isclass))

def create_fruit(fruit_type):
    try:
        return clsmembers[fruit_type]()
    except:
        print('Could not match Fruit type')

fruit1 = create_fruit('Apple')
print(fruit1)
# <__main__.Apple object at 0x1105de940>

fruit2 = create_fruit('Orange')
print(fruit2)
# <__main__.Orange object at 0x1105de978>

fruit3 = create_fruit('Grape')
# Could not match Fruit type

Answer 3

对于如此简单的任务，我只需使用 dict

def create_fruit(fruit_type):
    fruits = {1: Apple, 2: Orange}
    if fruit_type not in fruits.keys():
        raise Exception('fruit type does\'t exist!')
    klass = fruits[fruit_type]()
    print(klass) # <__main__.Apple object ...>

create_fruit(1)

这里有一些与您的问题非常相似的重复项

python 有相当于 Java Class.forName() 的吗？

你能用一个字符串来实例化一个类吗？

如何在python中动态创建类的实例？