mirror of
https://github.com/avatao-content/baseimage-tutorial-framework
synced 2025-06-28 21:55:13 +00:00
Move FSM instantiation out of FSMManagingSocketHandler
This commit is contained in:
@ -4,21 +4,17 @@ import zmq
|
||||
from tornado.web import Application
|
||||
from tornado.ioloop import IOLoop
|
||||
|
||||
from sql_injection_fsm import SQLInjectionFSM
|
||||
from tfw.config import WEB_PORT
|
||||
from tfw.config.logs import logging
|
||||
log = logging.getLogger(__name__)
|
||||
from tfw.networking.server.zmq_websocket_handler import FSMManagingSocketHandler
|
||||
from sql_injection_fsm import SQLInjectionFSM
|
||||
|
||||
|
||||
class SQLInjectionFSMManagingSocketHandler(FSMManagingSocketHandler):
|
||||
def __init__(self, application, request, **kwargs):
|
||||
super().__init__(SQLInjectionFSM, application, request, **kwargs)
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
fsm = SQLInjectionFSM()
|
||||
routes = [
|
||||
(r'/ws', SQLInjectionFSMManagingSocketHandler),
|
||||
(r'/ws', FSMManagingSocketHandler, {'fsm': fsm}),
|
||||
]
|
||||
application = Application(
|
||||
routes,
|
||||
|
Reference in New Issue
Block a user