python - 将枚举类型作为构造函数参数传递

Question

我是 python 的新手，我想在函数内将枚举作为参数传递给构造函数。

编辑:我正在开发一个包含必须组织不同类型数据的类的程序，但大多数这些数据类型都可以以相同的方式处理。这些数据不会同时或按可预见的顺序全部添加。因此，我想保留相同的功能，只是改变构造函数存储数据的方式。让我们考虑这个更简单的例子:

假设我有一个枚举

from enum import Enum, auto

class HouseThing(Enum):
    people = auto()
    pets = auto()
    furniture = auto()

我有一个类 House，可以包含其中的部分或全部内容

class House():

    def __init__(self, address, people = None, pets = None, 
        furniture = None):
        self.address = address, 
        if self.people is not None:
            self.people = people

        etc....

现在我想要一个可以 build 新装修房屋的功能，但我想使用一个可以用于任何房屋的功能:

house_things = HouseThing.furniture

def make_house_with_some_house_things(neighborhood, house_things):
     neighborhood.append(House(house_things.name = house_things.name))

有没有一种方法可以在不首先测试哪种 HouseThing house_things 的情况下执行此操作？ house_things.name 传递一个字符串，但我希望它能够将它用作关键字。

Answer 1

我不确定你到底想在这里实现什么，但为了解决这个难题:

首先，改变 House 以确定它传递的是什么:

class House():

    def __init__(self, address, *house_things):
        self.address = address
        for ht in house_things:
            if ht is HouseThings.people:
                self.people = ht
            elif ht is HouseThings.pets:
                self.pets = ht
            elif ht is HouseThings.furniture:
                self.furniture = ht
            else:
                raise ValueError('unknown house thing: %r' % (ht, ))

然后，将 make_house_with_some_house_things 更改为只传递给它的房子:

def make_house_with_some_house_things(neighborhood, house_things):
     neighborhood.append(House(house_things))