From 6764fc44e1391c21de4d838b409fe26a3f6e1945 Mon Sep 17 00:00:00 2001 From: Alvaro Neira Ayuso Date: Fri, 27 Dec 2019 14:37:12 +0100 Subject: Split the commands taking into account GET/POST operations --- src/ogRest.py | 26 ++++++++++++++++---------- 1 file changed, 16 insertions(+), 10 deletions(-) diff --git a/src/ogRest.py b/src/ogRest.py index ac02a4f..cec6aa8 100644 --- a/src/ogRest.py +++ b/src/ogRest.py @@ -42,16 +42,22 @@ class ogRest(): return msg def processOperation(self, op, URI, cmd, client): - if ("poweroff" in URI): - self.process_poweroff(client) - elif ("reboot" in URI): - self.process_reboot(client) - elif ("probe" in URI): - self.process_probe(client) - elif ("shell/run" in URI): - self.process_shellrun(client, cmd) - elif ("shell/output" in URI): - self.process_shellout(client) + if ("GET" in op): + if ("probe" in URI): + self.process_probe(client) + elif ("shell/output" in URI): + self.process_shellout(client) + else: + client.send(self.getResponse(ogResponses.BAD_REQUEST)) + elif ("POST" in op): + if ("poweroff" in URI): + self.process_poweroff(client) + elif ("reboot" in URI): + self.process_reboot(client) + elif ("shell/run" in URI): + self.process_shellrun(client, cmd) + else: + client.send(self.getResponse(ogResponses.BAD_REQUEST)) else: client.send(self.getResponse(ogResponses.BAD_REQUEST)) -- cgit v1.2.3-18-g5258