AlgRelay

[fukurou]

class AlgRelay:
def __init__(self):
super().__init__()
self.dic: dict[str, int] = {}
# code 1: next mode
        # code 2: goal achieved
        # code 3: error/failure
        # code 4: empty str
        # code 5: any other str

    def add_next_mode(self, *keys: str) -> None:
for key in keys:
self.dic[key] = 1

    def add_goal_achieved(self, *keys: str) -> None:
for key in keys:
self.dic[key] = 2

    def add_error(self, *keys: str) -> None:
for key in keys:
self.dic[key] = 3

    def relay(self, str1: str) -> int:
if str1 in self.dic:
return self.dic[str1]
if len(str1) == 0:
return 4
        return 5