Added CommonHTTPStatus

Author: michalpokusa

adafruit_httpserver/response.py

@@ -10,7 +10,7 @@
 from socketpool import SocketPool
 
 from .mime_type import MIMEType
-from .status import HTTPStatus, OK_200, NOT_FOUND_404
+from .status import HTTPStatus, CommonHTTPStatus
 
 class HTTPResponse:
     """Details of an HTTP response. Use in `HTTPServer.route` decorator functions."""
@@ -27,7 +27,7 @@ class HTTPResponse:
 
     def __init__(
         self,
-        status: HTTPStatus = OK_200,
+        status: HTTPStatus = CommonHTTPStatus.OK_200,
         body: str = "",
         headers: Dict[str, str] = None,
         content_type: str = MIMEType.TXT,
@@ -52,7 +52,7 @@ def __init__(
     @staticmethod
     def _construct_response_bytes(
         http_version: str = "HTTP/1.1",
-        status: HTTPStatus = OK_200,
+        status: HTTPStatus = CommonHTTPStatus.OK_200,
         content_type: str = MIMEType.TXT,
         content_length: Union[int, None] = None,
         headers: Dict[str, str] = None,
@@ -93,9 +93,9 @@ def send(self, conn: Union[SocketPool.Socket, socket.socket]) -> None:
             except OSError:
                 self._send_response(
                     conn,
-                    status = NOT_FOUND_404,
+                    status = CommonHTTPStatus.NOT_FOUND_404,
                     content_type = MIMEType.TXT,
-                    body = f"{NOT_FOUND_404} {self.filename}",
+                    body = f"{CommonHTTPStatus.NOT_FOUND_404} {self.filename}",
                 )
         else:
             self._send_response(

adafruit_httpserver/server.py

@@ -9,7 +9,7 @@
 from .request import HTTPRequest
 from .response import HTTPResponse
 from .route import HTTPRoute
-from .status import BAD_REQUEST_400
+from .status import CommonHTTPStatus
 
 class HTTPServer:
     """A basic socket-based HTTP server."""
@@ -106,7 +106,7 @@ def poll(self):
 
                 # If no handler exists and request method is not GET, return 400 Bad Request.
                 else:
-                    response = HTTPResponse(status=BAD_REQUEST_400)
+                    response = HTTPResponse(status=CommonHTTPStatus.BAD_REQUEST_400)
 
                 response.send(conn)
         except OSError as ex:

adafruit_httpserver/status.py

@@ -17,7 +17,16 @@ def __str__(self):
         return f"{self.code} {self.text}"
 
 
-OK_200 = HTTPStatus(200, "OK")
-BAD_REQUEST_400 = HTTPStatus(400, "Bad Request")
-NOT_FOUND_404 = HTTPStatus(404, "Not Found")
-INTERNAL_SERVER_ERROR_500 = HTTPStatus(500, "Internal Server Error")
+class CommonHTTPStatus(HTTPStatus):
+
+    OK_200 = HTTPStatus(200, "OK")
+    """200 OK"""
+
+    BAD_REQUEST_400 = HTTPStatus(400, "Bad Request")
+    """400 Bad Request"""
+
+    NOT_FOUND_404 = HTTPStatus(404, "Not Found")
+    """404 Not Found"""
+
+    INTERNAL_SERVER_ERROR_500 = HTTPStatus(500, "Internal Server Error")
+    """500 Internal Server Error"""