Question

在下面的代码片段中，我正在尝试定义一个工厂函数，该函数将根据参数返回从5派生的不同类的对象。

Hero

打算稍后使用它：

class Hero:
    Stats = namedtuple('Stats', ['health', 'defence', 'attack',
                                 'mana', 'experience'])
    RaceMaxStats = OrderedDict([
        ('Knight', Stats(100, 170, 150, 0, inf)),
        ('Barbarian', Stats(120, 150, 180, 0, inf)),
        ('Sorceress', Stats(50, 42, 90, 200, inf)),
        ('Warlock', Stats(70, 50, 100, 180, inf))
    ])

    @staticmethod
    def def_race(race: str):
        return type(race, (Hero,), {'max': Hero.RaceMaxStats[race]})

    Races = OrderedDict([
        (race, Hero.def_race(race)) for race in RaceMaxStats.keys()
    ])

    def __init__(self, lord, health, defence, attack, mana, experience):
        self.race = self.__class__.__name__
        self.lord = lord
        self.stats = Hero.Stats(min(health, self.max.health),
                                min(defence, self.max.defence),
                                min(attack, self.max.attack),
                                min(mana, self.max.mana),
                                min(experience, self.max.experience))

    @staticmethod
    def summon(race, *args, **kwargs):
        return Hero.Races[race](*args, **kwargs)

问题是我无法初始化子类，因为尚未定义knight = Hero.summon('Knight', 'Ronald', 90, 150, 150, 0, 20) warlock = Hero.summon('Warlock', 'Archibald', 50, 50, 100, 150, 50)：

Hero

显然，如果我用直接(race, Hero.def_race(race)) for race in RaceMaxStats.keys() NameError: name 'Hero' is not defined调用替换静态方法调用，我仍然需要定义type()。我的问题是如何最好地实施这种工厂。优先级是Hero方法保留相同的签名，并返回从summon()派生的类的实例。

P.S。以上代码都没有成功运行，因此可能包含其他错误。

Answer 1

您可以使用classmethods并将您的Races变量定义为一种方法，该方法在第一次调用类变量后对其结果进行缓存。它看起来像那样：

@classmethod
def def_race(cls, race: str):
    return type(race, (cls,), {'max': cls.RaceMaxStats[race]})

_Races = None

@classmethod
def Races(cls, race):
    if cls._Races is None:
        cls._Races = OrderedDict([
           (race, cls.def_race(race)) for race in cls.RaceMaxStats.keys()
        ])
    return cls._Races[race]

@classmethod
def summon(cls, race, *args, **kwargs):
    return cls.Races(race)(*args, **kwargs)

Answer 2

在课程定义之后，执行Hero.knight = Hero.summon(...)等等。

Answer 3

你可以尝试：

Hero('Knight', 'Ronald', 90, 150, 150, 0, 20).summon()

或者：

hero = Hero('Knight', 'Ronald', 90, 150, 150, 0, 20)
hero.summon()

Answer 4

现在我将使用summon()方法创建子类，并将其添加到Races（如果它尚未存在）。感谢eqperes提出此建议。

Races = OrderedDict()

@staticmethod
def def_race(race: str):
    return type(race, (Hero,), {'max': Hero.RaceMaxStats[race]})

@staticmethod
def summon(race: str, *args, **kwargs):
    try:
        hero = Hero.Races[race](*args, **kwargs)
    except KeyError:
        hero = Hero.def_race(race)(*args, **kwargs)
        Hero.Races[race] = hero.__class__
    return hero

Python：在类体中动态创建子类

4 个答案: