mirror of
https://github.com/avatao-content/baseimage-tutorial-framework
synced 2024-11-05 14:31:21 +00:00
Begin refactoring app.py to TFWServer class
This commit is contained in:
parent
1e9642912c
commit
bcd31de852
@ -14,30 +14,37 @@ log = logging.getLogger(__name__)
|
||||
from tfw.networking.server.zmq_websocket_handler import FSMManagingSocketHandler
|
||||
|
||||
|
||||
def zmq_callback(stream, msg_parts):
|
||||
key, data = deserialize_all(*msg_parts)
|
||||
if key == 'test':
|
||||
stream.send_multipart(serialize_all(key, 'OK'))
|
||||
if key == 'solution_check':
|
||||
stream.send_multipart(serialize_all(key, {
|
||||
'solved': fsm.is_solved(),
|
||||
'message': 'solved' if fsm.is_solved() else 'not solved'
|
||||
}))
|
||||
class TFWServer:
|
||||
def __init__(self, fsm_type):
|
||||
self._fsm = fsm_type()
|
||||
self.application = Application(
|
||||
[(r'/ws', FSMManagingSocketHandler, {'fsm': self.fsm})],
|
||||
autoreload=True
|
||||
)
|
||||
self.controller_connector = ControllerConnector()
|
||||
self.controller_connector.register_callback(self.zmq_callback)
|
||||
|
||||
@property
|
||||
def fsm(self):
|
||||
return self._fsm
|
||||
|
||||
def zmq_callback(self, stream, msg_parts):
|
||||
key, data = deserialize_all(*msg_parts)
|
||||
if key == 'test':
|
||||
stream.send_multipart(serialize_all(key, 'OK'))
|
||||
if key == 'solution_check':
|
||||
stream.send_multipart(serialize_all(key, {
|
||||
'solved': self.fsm.is_solved(),
|
||||
'message': 'solved' if self.fsm.is_solved() else 'not solved'
|
||||
}))
|
||||
|
||||
def listen(self, port):
|
||||
self.application.listen(port)
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
fsm = SQLInjectionFSM()
|
||||
controller_connector = ControllerConnector()
|
||||
controller_connector.register_callback(zmq_callback)
|
||||
routes = [
|
||||
(r'/ws', FSMManagingSocketHandler, {'fsm': fsm}),
|
||||
]
|
||||
application = Application(
|
||||
routes,
|
||||
autoreload=True
|
||||
)
|
||||
|
||||
application.listen(tfwenv.WEB_PORT)
|
||||
server = TFWServer(SQLInjectionFSM)
|
||||
server.listen(tfwenv.WEB_PORT)
|
||||
log.debug('Python version: {}'.format(sys.version[:5]))
|
||||
log.debug('Tornado version: {}'.format(tornado.version))
|
||||
log.debug('ZeroMQ version: {}'.format(zmq.zmq_version()))
|
||||
|
Loading…
Reference in New Issue
Block a user