diff options
Diffstat (limited to 'src/ogProcess.py')
| -rw-r--r-- | src/ogProcess.py | 31 |
1 files changed, 17 insertions, 14 deletions
diff --git a/src/ogProcess.py b/src/ogProcess.py index def76f1..27954a7 100644 --- a/src/ogProcess.py +++ b/src/ogProcess.py @@ -7,33 +7,36 @@ if platform.system() == 'Linux': from src.linux import ogOperations class ogProcess(): - def processOperation(self, op, URI, sock): + def processOperation(self, op, URI, client): if ("poweroff" in URI): - sock.send(bytes(ogRest.getResponse(ogRest.ogResponses.IN_PROGRESS), 'utf-8')) - sock.close() - self.process_poweroff() - return 0 + self.process_poweroff(client) elif ("reboot" in URI): - sock.send(bytes(ogRest.getResponse(ogRest.ogResponses.IN_PROGRESS), 'utf-8')) - sock.close() - self.process_reboot() - return 0 + self.process_reboot(client) elif ("probe" in URI): - sock.send(bytes(ogRest.getResponse(ogRest.ogResponses.OK), 'utf-8')) + self.process_probe(client) else: - sock.send(bytes(ogRest.getResponse(ogRest.ogResponses.BAD_REQUEST), 'utf-8')) + client.send(ogRest.getResponse(ogRest.ogResponses.BAD_REQUEST)) - return 1 + return 0 - def process_reboot(self): + def process_reboot(self, client): # Rebooting thread def rebt(): ogOperations.reboot() + + client.send(ogRest.getResponse(ogRest.ogResponses.IN_PROGRESS)) + client.disconnect() threading.Thread(target=rebt).start() - def process_poweroff(self): + def process_poweroff(self, client): # Powering off thread def pwoff(): time.sleep(2) ogOperations.poweroff() + + client.send(ogRest.getResponse(ogRest.ogResponses.IN_PROGRESS)) + client.disconnect() threading.Thread(target=pwoff).start() + + def process_probe(self, client): + client.send(ogRest.getResponse(ogRest.ogResponses.OK)) |