Move FSM instantiation out of FSMManagingSocketHandler

This commit is contained in:
Bálint Bokros
2018-02-08 14:46:00 +01:00
parent 849fd7b61c
commit 8b9f7e50cb
2 changed files with 5 additions and 13 deletions

View File

@ -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,