From c365de32cf5ebb7efc631a4ed08d31ae71bab778 Mon Sep 17 00:00:00 2001 From: Arksine Date: Wed, 27 Jan 2021 06:22:15 -0500 Subject: [PATCH] machine: add "restart_service" method Move the service restart functionality out of the "service restart" request into its own method. This allows the "service restart" to be called internally by other modules. Signed-off-by: Eric Callahan --- moonraker/plugins/machine.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/moonraker/plugins/machine.py b/moonraker/plugins/machine.py index b8e8a3b..fa5c56c 100644 --- a/moonraker/plugins/machine.py +++ b/moonraker/plugins/machine.py @@ -39,13 +39,15 @@ class Machine: async def reboot_machine(self): await self._execute_cmd("sudo shutdown -r now") + async def restart_service(self, service_name): + await self._execute_cmd(f'sudo systemctl restart {service_name}') + async def _handle_service_restart(self, web_request): name = web_request.get('service') if name == "klipper": - await self._execute_cmd(f'sudo systemctl restart {name}') + await self.restart_service(name) elif name == "moonraker": - IOLoop.current().spawn_callback( - self._execute_cmd, f'sudo systemctl restart {name}') + IOLoop.current().spawn_callback(self.restart_service, name) else: raise self.sever.error( f"Invalid argument recevied for 'name': {name}")