Merge pull request #48 from Neradoc/respond-to-file-head

Allow HEAD requests to files paths

Author: anecdata

adafruit_httpserver/response.py

@@ -172,6 +172,7 @@ def send_file(
         filename: str = "index.html",
         root_path: str = "./",
         buffer_size: int = 1024,
+        head_only: bool = False,
     ) -> None:
         """
         Send response with content of ``filename`` located in ``root_path``.
@@ -197,9 +198,10 @@ def send_file(
             content_length=file_length,
         )
 
-        with open(root_path + filename, "rb") as file:
-            while bytes_read := file.read(buffer_size):
-                self._send_bytes(self.request.connection, bytes_read)
+        if not head_only:
+            with open(root_path + filename, "rb") as file:
+                while bytes_read := file.read(buffer_size):
+                    self._send_bytes(self.request.connection, bytes_read)
         self._response_already_sent = True
 
     def send_chunk(self, chunk: str = "") -> None:

adafruit_httpserver/server.py

@@ -167,12 +167,16 @@ def poll(self):
                     handler(request)
 
                 # If no handler exists and request method is GET, try to serve a file.
-                elif handler is None and request.method == HTTPMethod.GET:
+                elif handler is None and request.method in (
+                    HTTPMethod.GET,
+                    HTTPMethod.HEAD,
+                ):
                     filename = "index.html" if request.path == "/" else request.path
                     HTTPResponse(request).send_file(
                         filename=filename,
                         root_path=self.root_path,
                         buffer_size=self.request_buffer_size,
+                        head_only=(request.method == HTTPMethod.HEAD),
                     )
                 else:
                     HTTPResponse(