mirror of
https://github.com/avatao-content/baseimage-tutorial-framework
synced 2024-11-14 16:47:17 +00:00
85 lines
2.7 KiB
Python
85 lines
2.7 KiB
Python
import logging
|
|
from collections import defaultdict
|
|
from datetime import datetime
|
|
|
|
from transitions import Machine, MachineError
|
|
|
|
from tfw.internals.callback_mixin import CallbackMixin
|
|
|
|
LOG = logging.getLogger(__name__)
|
|
|
|
|
|
class FSMBase(Machine, CallbackMixin):
|
|
"""
|
|
A general FSM base class you can inherit from to track user progress.
|
|
See linear_fsm.py for an example use-case.
|
|
TFW uses the transitions library for state machines, please refer to their
|
|
documentation for more information on creating your own machines:
|
|
https://github.com/pytransitions/transitions
|
|
"""
|
|
states, transitions = [], []
|
|
|
|
def __init__(self, initial=None, accepted_states=None):
|
|
"""
|
|
:param initial: which state to begin with, defaults to the last one
|
|
:param accepted_states: list of states in which the challenge should be
|
|
considered successfully completed
|
|
"""
|
|
self.accepted_states = accepted_states or [self.states[-1].name]
|
|
self.trigger_predicates = defaultdict(list)
|
|
self.event_log = []
|
|
|
|
Machine.__init__(
|
|
self,
|
|
states=self.states,
|
|
transitions=self.transitions,
|
|
initial=initial or self.states[0],
|
|
send_event=True,
|
|
ignore_invalid_triggers=True,
|
|
after_state_change='execute_callbacks'
|
|
)
|
|
|
|
def execute_callbacks(self, event_data):
|
|
self._execute_callbacks(event_data.kwargs)
|
|
|
|
def is_solved(self):
|
|
return self.state in self.accepted_states # pylint: disable=no-member
|
|
|
|
def subscribe_predicate(self, trigger, *predicates):
|
|
self.trigger_predicates[trigger].extend(predicates)
|
|
|
|
def unsubscribe_predicate(self, trigger, *predicates):
|
|
self.trigger_predicates[trigger] = [
|
|
predicate
|
|
for predicate in self.trigger_predicates[trigger]
|
|
not in predicates
|
|
]
|
|
|
|
def step(self, trigger):
|
|
predicate_results = (
|
|
predicate()
|
|
for predicate in self.trigger_predicates[trigger]
|
|
)
|
|
|
|
if all(predicate_results):
|
|
try:
|
|
from_state = self.state
|
|
self.trigger(trigger)
|
|
self.update_event_log(from_state, trigger)
|
|
return True
|
|
except (AttributeError, MachineError):
|
|
LOG.debug('FSM failed to execute nonexistent trigger: "%s"', trigger)
|
|
return False
|
|
|
|
def update_event_log(self, from_state, trigger):
|
|
self.event_log.append({
|
|
'from_state': from_state,
|
|
'to_state': self.state,
|
|
'trigger': trigger,
|
|
'timestamp': datetime.utcnow()
|
|
})
|
|
|
|
@property
|
|
def in_accepted_state(self):
|
|
return self.state in self.accepted_states
|