mrtoth/baseimage-tutorial-framework
1
0
Fork 0
mirror of https://github.com/avatao-content/baseimage-tutorial-framework synced 2025-07-04 08:46:23 +00:00
Code Issues Releases Wiki Activity

Move common parts of FSM to a baseclass

Browse Source
This commit is contained in:
Bálint Bokros
2017-12-06 01:29:09 +01:00
parent 0a3c06109e
commit d3db21f26e
3 changed files with 27 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
src/app/sql_injection_fsm.py
View File

@ -1,9 +1,7 @@
from transitions import Machine
import event_handler_connector
from fsm_base import FSMBase
class SQLInjectionFSM:
class SQLInjectionFSM(FSMBase):
states = ['start', 'stripped_code', 'sql', 'commented_code', 'sql_with_substitutions', 'sql_output', 'end']
transitions = [
{'trigger': 'anchor_webide', 'source': 'start', 'dest': 'stripped_code'},
@ -16,17 +14,7 @@ class SQLInjectionFSM:
]
def __init__(self):
self.machine = Machine(model=self,
states=SQLInjectionFSM.states,
transitions=SQLInjectionFSM.transitions,
initial='start',
send_event=True,
ignore_invalid_triggers=True,
after_state_change='forward_message')
def forward_message(self, event_data):
message = event_data.kwargs.get('message')
event_handler_connector.send_message(message)
super().__init__('start')
fsm = SQLInjectionFSM()