Autogenerated type safety parameterisation of all methods

providers/spot/wota.py

@@ -1,10 +1,13 @@
+from __future__ import annotations
+
 import logging
 import re
 from datetime import datetime
-from typing import cast
+from typing import Any, cast
 from xml.parsers.expat import ExpatError
 
 import pytz
+import requests
 from rss_parser import Parser
 from rss_parser.models.rss import RSS
 
@@ -23,11 +26,11 @@ class WOTA(HTTPSpotProvider):
     SPOTS_URL = "https://www.wota.org.uk/spots_rss.php"
     RSS_DATE_TIME_FORMAT = "%a, %d %b %Y %H:%M:%S %z"
 
-    def __init__(self, provider_config):
+    def __init__(self, provider_config: dict[str, Any]) -> None:
         super().__init__("WOTA", provider_config, self.SPOTS_URL, self.POLL_INTERVAL_SEC)
 
-    def _http_response_to_spots(self, http_response):
-        new_spots = []
+    def _http_response_to_spots(self, http_response: requests.Response) -> list[Spot]:
+        new_spots: list[Spot] = []
         try:
             rss = cast(RSS, Parser.parse(http_response.content.decode("utf-8-sig")))
             # Iterate through source data
@@ -110,9 +113,9 @@ class WOTA(HTTPSpotProvider):
 
         return new_spots
 
-    def can_submit_spot(self, activity):
+    def can_submit_spot(self, activity: str) -> bool:
         return activity == ActivityName.WOTA
 
-    def submit_spot(self, spot, credentials):
+    def submit_spot(self, spot: Spot, credentials: dict[str, str]) -> None:
         # TODO Ask M5TEA if he's happy to share how this is done from his app
         raise NotImplementedError("WOTA upstream spot submission is not yet implemented")