你需要创建一个类变量，它是一个该类的子类。如何管理这个引用？

Question

请将以下代码视为问题示例进行考虑。

class State:
    START: State = StartState()    # Problem Here - StartState
    END: State = EndState()        # Problem Here - EndState

    @classmethod
    def get_current(cls, context: Context) ->"State":
        if context.just_beginning :
            return cls.START
        return cls.END

class StartState(State):
    ...

class EndState(State):
    ...

代码无法运行，因为未定义StartState和EndState。
但是，由于尚未定义State，我也不能将StartState和EndState放在State的上方。
请指导我如何在Python中解决这个问题。

RiggsFolly · Answer

也许对您的问题进行一些解释可能会有所帮助。但这里有一个代码片段，可能会对您有帮助：

class State:
    START: "State"
    END: "State"

    @classmethod
    def get_current(cls, context: "Context") -> "State":
        if context.just_beginning:
            return cls.START
        return cls.END

class StartState(State):
    ...

class EndState(State):
    ...

State.START = StartState()
State.END = EndState()

你可以在类的内部先声明 State.START 和 State.END 的类型，然后在类定义结束后，通过调用 StartState() 和 EndState() 函数来初始化这些值。这样，你就可以在类中使用开始和结束状态的实例，同时保持代码的结构清晰并允许扩展其他状态类。

RichardTheKiwi · Answer

当StartState和EndState仅打算表示各自单一的状态时，将它们定义为类并没有充分的理由。根据惯例，应将它们作为State的“常量”实例（通常全大写拼写）来定义。

get_current方法也应该更适当地定义为Context类的实例方法，因为你真正想要的是根据当前上下文找到状态：

class State:
    pass

class Context:
    def get_state(self) -> State:
        if self.just_beginning:
            return START_STATE
        return END_STATE

START_STATE = State()
END_STATE = State()