gundam-state/base_gundam.py at main · amur0ray/gundam-state · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
import logging
import sys
import time

from transitions import Machine


class BaseGundam:
    """A state machine representing a basic Gundam model.

    Can switch between different modes (i.e. states) in a rock, paper, scissors
    style game.
    """

    states = ["docked", "airframe", "fighter", "transforming"]
    transitions = [
        ["takeoff", "docked", "airframe"],
        ["dock", "airframe", "docked"],
        {
            "trigger": "transform",
            "source": "fighter",
            "dest": "transforming",
            "after": "transform_airframe",
        },
        {
            "trigger": "transform",
            "source": "airframe",
            "dest": "transforming",
            "after": "transform_fighter",
        },
        ["go_airframe", "transforming", "airframe"],
        ["go_fighter", "transforming", "fighter"],
    ]

    def __init__(self, name: str = None, initial: str = "docked") -> None:
        self.machine = Machine(
            model=self,
            states=BaseGundam.states,
            transitions=BaseGundam.transitions,
            initial=initial,
        )
        self._name = name
        self._max_dist = 2

    @property
    def name(self):
        return self._name

    def transform_airframe(self):
        time.sleep(5)

        self.go_airframe()

        logging.info(f"{self.name} is now in 'airframe' mode.")

    def transform_fighter(self):
        time.sleep(5)

        self.go_fighter()

        logging.info(f"{self.name} is now in 'fighter' mode.")

    def move(self, dist: int) -> int:
        return min(dist, self._max_dist)

    def attack(self):
        raise NotImplementedError()