type annotation

Author: atstaeff

eventsourcingdb/container.py

@@ -12,7 +12,7 @@
 class Container:
     def __init__(
         self,
-    ):
+    ) -> None:
         self._image_name: str = 'thenativeweb/eventsourcingdb'
         self._image_tag: str = 'latest'
         self._api_token: str = 'secret'
@@ -57,7 +57,7 @@ def _create_container(self) -> None:
             detach=True,
         )  # type: ignore
 
-    def _extract_port_from_container_info(self, container_info):
+    def _extract_port_from_container_info(self, container_info) -> int | None:
         port = None
         valid_mapping = True
         port_mappings = None
@@ -95,7 +95,7 @@ def _fetch_mapped_port(self) -> None:
         self._stop_and_remove_container()
         raise RuntimeError('Failed to determine mapped port')
 
-    def _get_container_info(self):
+    def _get_container_info(self) -> dict | None:
         if self._container is None:
             return None
         return self._docker_client.api.inspect_container(self._container.id)
@@ -142,7 +142,7 @@ def _pull_or_get_image(self) -> None:
         except errors.APIError as e:
             self._handle_image_pull_error(e)
 
-    def _handle_image_pull_error(self, error):
+    def _handle_image_pull_error(self, error) -> None:
         image_name = f"{self._image_name}:{self._image_tag}"
         try:
             self._docker_client.images.get(image_name)
@@ -211,7 +211,7 @@ def _check_endpoint_available(self, url: str) -> bool:
         return self._check_response_ok(response)
 
     # pylint: disable=R6301
-    def _check_response_ok(self, response) -> bool:
+    def _check_response_ok(self, response: requests.Response) -> bool:
         return response is not None and response.status_code == HTTPStatus.OK
 
     def with_api_token(self, token: str) -> 'Container':