Added option to pass status to HTTPResponse as tuple, overwrited eq method on HTTPStatus

Author: michalpokusa

adafruit_httpserver/response.py

@@ -8,7 +8,7 @@
 """
 
 try:
-    from typing import Optional, Dict, Union
+    from typing import Optional, Dict, Union, Tuple
     from socket import socket
     from socketpool import SocketPool
 except ImportError:
@@ -37,7 +37,7 @@ class HTTPResponse:
 
     def __init__(  # pylint: disable=too-many-arguments
         self,
-        status: HTTPStatus = CommonHTTPStatus.OK_200,
+        status: Union[HTTPStatus, Tuple[int, str]] = CommonHTTPStatus.OK_200,
         body: str = "",
         headers: Dict[str, str] = None,
         content_type: str = MIMEType.TYPE_TXT,
@@ -50,8 +50,7 @@ def __init__(  # pylint: disable=too-many-arguments
 
         Returns ``body`` if ``filename`` is ``None``, otherwise returns contents of ``filename``.
         """
-
-        self.status = status
+        self.status = status if isinstance(status, HTTPStatus) else HTTPStatus(*status)
         self.body = body
         self.headers = headers or {}
         self.content_type = content_type

adafruit_httpserver/status.py

@@ -26,6 +26,9 @@ def __repr__(self):
     def __str__(self):
         return f"{self.code} {self.text}"
 
+    def __eq__(self, other: "HTTPStatus"):
+        return self.code == other.code and self.text == other.text
+
 
 class CommonHTTPStatus(HTTPStatus):  # pylint: disable=too-few-public-methods
     """Common HTTP status codes."""