app: Simplify request handlers
There is no need to pass the auth, server, or websocket manager objects to request handlers. They can be retreived directly from the application reference. Signed-off-by: Eric Callahan <arksine.code@gmail.com>
This commit is contained in:
parent
7abc86847e
commit
92d1715d88
|
@ -114,10 +114,8 @@ class MoonrakerApp:
|
|||
self.mutable_router = MutableRouter(self)
|
||||
app_handlers = [
|
||||
(AnyMatches(), self.mutable_router),
|
||||
(r"/websocket", WebSocket,
|
||||
{'wsm': self.wsm, 'auth': self.auth}),
|
||||
(r"/api/version", EmulateOctoprintHandler,
|
||||
{'server': self.server, 'auth': self.auth})]
|
||||
(r"/websocket", WebSocket, {'main_app': self}),
|
||||
(r"/api/version", EmulateOctoprintHandler, {'main_app': self})]
|
||||
|
||||
self.app = tornado.web.Application(
|
||||
app_handlers,
|
||||
|
@ -137,6 +135,15 @@ class MoonrakerApp:
|
|||
port, address=host, max_body_size=self.max_upload_size,
|
||||
xheaders=True)
|
||||
|
||||
def get_server(self):
|
||||
return self.server
|
||||
|
||||
def get_auth(self):
|
||||
return self.auth
|
||||
|
||||
def get_websocket_manager(self):
|
||||
return self.wsm
|
||||
|
||||
async def close(self):
|
||||
if self.tornado_server is not None:
|
||||
self.tornado_server.stop()
|
||||
|
@ -156,8 +163,7 @@ class MoonrakerApp:
|
|||
f"Websocket: {', '.join(api_def.ws_methods)}")
|
||||
self.wsm.register_remote_handler(api_def)
|
||||
params = {}
|
||||
params['server'] = self.server
|
||||
params['auth'] = self.auth
|
||||
params['main_app'] = self
|
||||
params['arg_parser'] = api_def.parser
|
||||
params['remote_callback'] = api_def.endpoint
|
||||
self.mutable_router.add_handler(
|
||||
|
@ -174,8 +180,7 @@ class MoonrakerApp:
|
|||
if "http" in protocol:
|
||||
msg += f" - HTTP: ({' '.join(request_methods)}) {uri}"
|
||||
params = {}
|
||||
params['server'] = self.server
|
||||
params['auth'] = self.auth
|
||||
params['main_app'] = self
|
||||
params['methods'] = request_methods
|
||||
params['arg_parser'] = api_def.parser
|
||||
params['callback'] = callback
|
||||
|
@ -199,11 +204,11 @@ class MoonrakerApp:
|
|||
logging.info(f"Invalid file path: {file_path}")
|
||||
return
|
||||
logging.debug(f"Registering static file: ({pattern}) {file_path}")
|
||||
params = {'server': self.server, 'auth': self.auth, 'path': file_path}
|
||||
params = {'main_app': self, 'path': file_path}
|
||||
self.mutable_router.add_handler(pattern, FileRequestHandler, params)
|
||||
|
||||
def register_upload_handler(self, pattern):
|
||||
params = {'server': self.server, 'auth': self.auth}
|
||||
params = {'main_app': self}
|
||||
self.mutable_router.add_handler(pattern, FileUploadHandler, params)
|
||||
|
||||
def remove_handler(self, endpoint):
|
||||
|
@ -253,8 +258,8 @@ class MoonrakerApp:
|
|||
|
||||
# ***** Dynamic Handlers*****
|
||||
class RemoteRequestHandler(AuthorizedRequestHandler):
|
||||
def initialize(self, remote_callback, server, auth, arg_parser):
|
||||
super(RemoteRequestHandler, self).initialize(server, auth)
|
||||
def initialize(self, main_app, remote_callback, arg_parser):
|
||||
super(RemoteRequestHandler, self).initialize(main_app)
|
||||
self.remote_callback = remote_callback
|
||||
self.query_parser = arg_parser
|
||||
|
||||
|
@ -277,9 +282,8 @@ class RemoteRequestHandler(AuthorizedRequestHandler):
|
|||
self.finish({'result': result})
|
||||
|
||||
class LocalRequestHandler(AuthorizedRequestHandler):
|
||||
def initialize(self, callback, server, auth,
|
||||
methods, arg_parser):
|
||||
super(LocalRequestHandler, self).initialize(server, auth)
|
||||
def initialize(self, main_app, callback, methods, arg_parser):
|
||||
super(LocalRequestHandler, self).initialize(main_app)
|
||||
self.callback = callback
|
||||
self.methods = methods
|
||||
self.query_parser = arg_parser
|
||||
|
@ -339,9 +343,6 @@ class FileRequestHandler(AuthorizedFileHandler):
|
|||
self.finish({'result': filename})
|
||||
|
||||
class FileUploadHandler(AuthorizedRequestHandler):
|
||||
def initialize(self, server, auth):
|
||||
super(FileUploadHandler, self).initialize(server, auth)
|
||||
|
||||
async def post(self):
|
||||
file_manager = self.server.lookup_plugin('file_manager')
|
||||
try:
|
||||
|
|
|
@ -180,9 +180,10 @@ class Authorization:
|
|||
self.prune_handler.stop()
|
||||
|
||||
class AuthorizedRequestHandler(tornado.web.RequestHandler):
|
||||
def initialize(self, server, auth):
|
||||
self.server = server
|
||||
self.auth = auth
|
||||
def initialize(self, main_app):
|
||||
self.server = main_app.get_server()
|
||||
self.auth = main_app.get_auth()
|
||||
self.wsm = main_app.get_websocket_manager()
|
||||
|
||||
def prepare(self):
|
||||
if not self.auth.check_authorized(self.request):
|
||||
|
@ -210,10 +211,10 @@ class AuthorizedRequestHandler(tornado.web.RequestHandler):
|
|||
# Due to the way Python treats multiple inheritance its best
|
||||
# to create a separate authorized handler for serving files
|
||||
class AuthorizedFileHandler(tornado.web.StaticFileHandler):
|
||||
def initialize(self, server, auth, path, default_filename=None):
|
||||
def initialize(self, main_app, path, default_filename=None):
|
||||
super(AuthorizedFileHandler, self).initialize(path, default_filename)
|
||||
self.server = server
|
||||
self.auth = auth
|
||||
self.server = main_app.get_server()
|
||||
self.auth = main_app.get_auth()
|
||||
|
||||
def prepare(self):
|
||||
if not self.auth.check_authorized(self.request):
|
||||
|
|
|
@ -268,9 +268,9 @@ class WebsocketManager:
|
|||
self.websockets = {}
|
||||
|
||||
class WebSocket(WebSocketHandler):
|
||||
def initialize(self, wsm, auth):
|
||||
self.wsm = wsm
|
||||
self.auth = auth
|
||||
def initialize(self, main_app):
|
||||
self.auth = main_app.get_auth()
|
||||
self.wsm = main_app.get_websocket_manager()
|
||||
self.rpc = self.wsm.rpc
|
||||
self.uid = id(self)
|
||||
|
||||
|
|
Loading…
Reference in New Issue