feat(wireless configuration): Fixed empty request causing crash

firmware/lib/request.py

@@ -1,3 +1,4 @@
+
 class Request:
     def __init__(self, request):
         self.raw = request
@@ -10,7 +11,6 @@ def __init__(self, request):
         self.parse_request()
 
     def parse_request(self):
-        print(self.raw)
         try:
             request, self.body = self.raw.split("\r\n\r\n")
         except ValueError:

firmware/lib/webpage.py

@@ -15,12 +15,15 @@ def default_route(*args, **kwargs):
         {"ssid": "SSID:"},
         {"password": "Password:"},
         {"babybuddy": "BabyBuddy URL:"},
-        {"babyauth": "BabyBuddy API Key"}]
+        {"babyauth": "BabyBuddy API Key"},
+    ]
     render = ""
     for option in options:
         for key, value in option.items():
-            render += f''' <input type="text" id="{key}" name="{key}" placeholder="{value}"><br><br>'''
-    return template.render_template(load_webpage("webpages/default.html"), {"render": render})
+            render += f""" <input type="text" id="{key}" name="{key}" placeholder="{value}"><br><br>"""
+    return template.render_template(
+        load_webpage("webpages/default.html"), {"render": render}
+    )
 
 
 def config_route(*args, **kwargs):
@@ -37,7 +40,8 @@ def config_route(*args, **kwargs):
             if request.query_strings.get("babybuddy"):
                 secret_json["BASE_URL"] = request.query_strings.get("babybuddy")
             if request.query_strings.get("babyauth"):
-                secret_json["AUTHORIZATION"]["Authorization"] = request.query_strings.get("babyauth")
+                secret_json["AUTHORIZATION"][
+                    "Authorization"
+                ] = request.query_strings.get("babyauth")
             secret.write(json.dumps(secret_json))
     machine.reset()
-

firmware/lib/webrouter.py

@@ -37,27 +37,27 @@ def open_socket(ip, port):
     @staticmethod
     def determine_mimetype(path):
         mimetype = "text/html"
-        if path.endswith('.js'):
+        if path.endswith(".js"):
             mimetype = "text/javascript"
-        if path.endswith('.css'):
+        if path.endswith(".css"):
             mimetype = "text/css"
 
-
     def serve(self):
         while True:
             client = self.socket.accept()[0]
             request = client.recv(1024)
-            request = Request(request.decode("utf-8"))
-            path = request.path
-            header = 'HTTP/1.1 200 OK\n'
-            mimetype = self.determine_mimetype(path)
-            if path in self.routes.keys():
-                webpage = self.routes[path](request=request)
-            elif path in self.static_files:
-                with open(f"{self.static_location}/{path}", "rb") as static:
-                    webpage = static.read()
-            else:
-                webpage = self.routes["default"](request=request)
-            header += f"Content-Type: {mimetype}\n\n"
-            client.send(webpage)
-            client.close()
+            if request:
+                request = Request(request.decode("utf-8"))
+                path = request.path
+                header = "HTTP/1.1 200 OK\n"
+                mimetype = self.determine_mimetype(path)
+                if path in self.routes.keys():
+                    webpage = self.routes[path](request=request)
+                elif path in self.static_files:
+                    with open(f"{self.static_location}/{path}", "rb") as static:
+                        webpage = static.read()
+                else:
+                    webpage = self.routes["default"](request=request)
+                header += f"Content-Type: {mimetype}\n\n"
+                client.sendall(webpage)
+                client.close()