Autogenerated type safety parameterisation of all methods

providers/spot/hema.py

@@ -1,6 +1,9 @@
+from __future__ import annotations
+
 import logging
 import re
 from datetime import datetime
+from typing import Any
 
 import pytz
 import requests
@@ -27,17 +30,17 @@ class HEMA(HTTPSpotProvider):
     FREQ_MODE_PATTERN = re.compile("^([\\d.]*) \\((.*)\\)$")
     SPOTTER_COMMENT_PATTERN = re.compile("^\\((.*)\\) (.*)$")
 
-    def __init__(self, provider_config):
+    def __init__(self, provider_config: dict[str, Any]) -> None:
         super().__init__("HEMA", provider_config, self.SPOT_SEED_URL, self.POLL_INTERVAL_SEC)
-        self._spot_seed = ""
+        self._spot_seed: str = ""
 
-    def _http_response_to_spots(self, http_response):
+    def _http_response_to_spots(self, http_response: requests.Response) -> list[Spot]:
         # OK, source data is actually just the spot seed at this point. We'll then go on to fetch real data if we know
         # this has changed.
         spot_seed_changed = http_response.text != self._spot_seed
         self._spot_seed = http_response.text
 
-        new_spots = []
+        new_spots: list[Spot] = []
         # OK, if the spot seed actually changed, now we make the real request for data.
         if spot_seed_changed:
             try:
@@ -89,10 +92,10 @@ class HEMA(HTTPSpotProvider):
                 logger.warning("Connection error when accessing HEMA spots API.")
         return new_spots
 
-    def can_submit_spot(self, activity):
+    def can_submit_spot(self, activity: str) -> bool:
         return activity == ActivityName.HEMA
 
-    def submit_spot(self, spot, credentials):
+    def submit_spot(self, spot: Spot, credentials: dict[str, str]) -> None:
         # TODO: Implement. Currently blocked awaiting their API team to make a change to allow us to spot with a
         # reference and not a reference *number*.
         raise NotImplementedError("HEMA upstream spot submission is not yet implemented")