Mirror
/
klipper-dgus
mirror of https://github.com/Desuuuu/klipper.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

webhooks: Add register_mux_endpoint() helper function 

Add ability to register an endpoint that invokes a handler based on a
parameter in the request.

Signed-off-by: Kevin O'Connor <kevin@koconnor.net>
This commit is contained in:
Kevin O'Connor 2021-07-22 20:50:16 -04:00
parent 809f1ccea6
commit cf2e941aec
1 changed files with 28 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
klippy/webhooks.py
View File

@ -289,6 +289,7 @@ class WebHooks:
self.printer = printer self.printer = printer
self._endpoints = {"list_endpoints": self._handle_list_endpoints} self._endpoints = {"list_endpoints": self._handle_list_endpoints}
self._remote_methods = {} self._remote_methods = {}
self._mux_endpoints = {}
self.register_endpoint("info", self._handle_info_request) self.register_endpoint("info", self._handle_info_request)
self.register_endpoint("emergency_stop", self._handle_estop_request) self.register_endpoint("emergency_stop", self._handle_estop_request)
self.register_endpoint("register_remote_method", self.register_endpoint("register_remote_method",
@ -300,6 +301,33 @@ class WebHooks:
raise WebRequestError("Path already registered to an endpoint") raise WebRequestError("Path already registered to an endpoint")
self._endpoints[path] = callback self._endpoints[path] = callback
def register_mux_endpoint(self, path, key, value, callback):
prev = self._mux_endpoints.get(path)
if prev is None:
self.register_endpoint(path, self._handle_mux)
self._mux_endpoints[path] = prev = (key, {})
prev_key, prev_values = prev
if prev_key != key:
raise self.printer.config_error(
"mux endpoint %s %s %s may have only one key (%s)"
% (path, key, value, prev_key))
if value in prev_values:
raise self.printer.config_error(
"mux endpoint %s %s %s already registered (%s)"
% (path, key, value, prev_values))
prev_values[value] = callback
def _handle_mux(self, web_request):
key, values = self._mux_endpoints[web_request.get_method()]
if None in values:
key_param = web_request.get(key, None)
else:
key_param = web_request.get(key)
if key_param not in values:
raise web_request.error("The value '%s' is not valid for %s"
% (key_param, key))
values[key_param](web_request)
def _handle_list_endpoints(self, web_request): def _handle_list_endpoints(self, web_request):
web_request.send({'endpoints': list(self._endpoints.keys())}) web_request.send({'endpoints': list(self._endpoints.keys())})