elias/gardentron
1
0
Fork 0
mirror of https://github.com/eliasrenman/gardentron.git synced 2026-03-16 20:46:07 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix: body parser

Browse Source
This commit is contained in:
Elias Renman
2023-06-01 22:39:13 +02:00
parent 2853d8a4c0
commit 2c93801867
1 changed files with 18 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
pico-w/src/decorators.py
View File

@@ -40,17 +40,21 @@ class Endpoint:
return return
try: try:
body = get_body(request) body = get_body(request)
if body is str and body != "":
if isinstance(body, str) and body != "":
kwargs = dict(**kwargs, body=body) kwargs = dict(**kwargs, body=body)
val = function(instance, *args, **kwargs) val = function(instance, *args, **kwargs)
respond(cl, 200, val) respond(cl, 200, val)
return val return True
except HttpError as e: except HttpError as e:
respond(cl, e.status, respond(cl, e.status,
e.message) e.message)
return True
except Exception as e: except Exception as e:
print("Some other unkown error occured:", e) print("Some other unkown error occured:", e)
respond(cl, 500, {"status": "error", "message": e})
return True
return wrapper return wrapper
@@ -62,17 +66,25 @@ Please note that this only supports content-type 'application/json'
def get_body(request: str): def get_body(request: str):
req = request.split("\r\n") req = request.split("\r\n")
content_type_index = -10 content_length_index = 100
out = ""
for index in range(len(req)): for index in range(len(req)):
item = req[index] item = req[index]
if item.find("Content-Type") != -1: if item.find("Content-Type") != -1:
if not item.find("application/json"): if not item.find("application/json"):
return return
content_type_index = index
if (content_type_index + 4) == index:
return item if item.find("Content-Length") != -1:
content_length_index = index
if index > content_length_index:
out += item
out = out.replace("\n", "")
out = out.replace("\t", "")
if out != "":
return out
class ServerHandler(object): class ServerHandler(object):