Replaced decorator that prevents sending Response multiple times with method

IDE was getting confused by decorated method and was not displaying the type hint properly

Author: michalpokusa

adafruit_httpserver/response.py

@@ -8,7 +8,7 @@
 """
 
 try:
-    from typing import Optional, Dict, Union, Tuple, Callable
+    from typing import Optional, Dict, Union, Tuple
     from socket import socket
     from socketpool import SocketPool
 except ImportError:
@@ -29,21 +29,6 @@
 from .headers import Headers
 
 
-def _prevent_multiple_send_calls(function: Callable):
-    """
-    Decorator that prevents calling ``send`` or ``send_file`` more than once.
-    """
-
-    def wrapper(self: "Response", *args, **kwargs):
-        if self._response_already_sent:  # pylint: disable=protected-access
-            raise ResponseAlreadySentError
-
-        result = function(self, *args, **kwargs)
-        return result
-
-    return wrapper
-
-
 class Response:
     """
     Response to a given `Request`. Use in `Server.route` handler functions.
@@ -162,7 +147,11 @@ def _send_headers(
             self.request.connection, response_message_header.encode("utf-8")
         )
 
-    @_prevent_multiple_send_calls
+    def _check_if_not_already_sent(self) -> None:
+        """Prevents calling ``send`` or ``send_file`` more than once."""
+        if self._response_already_sent:
+            raise ResponseAlreadySentError
+
     def send(
         self,
         body: str = "",
@@ -174,6 +163,7 @@ def send(
 
         Should be called **only once** per response.
         """
+        self._check_if_not_already_sent()
 
         if getattr(body, "encode", None):
             encoded_response_message_body = body.encode("utf-8")
@@ -214,7 +204,6 @@ def _get_file_length(file_path: str) -> int:
         except OSError:
             raise FileNotExistsError(file_path)  # pylint: disable=raise-missing-from
 
-    @_prevent_multiple_send_calls
     def send_file(  # pylint: disable=too-many-arguments
         self,
         filename: str = "index.html",
@@ -230,6 +219,7 @@ def send_file(  # pylint: disable=too-many-arguments
 
         Should be called **only once** per response.
         """
+        self._check_if_not_already_sent()
 
         if safe:
             self._check_file_path_is_valid(filename)