Ft Rest client - fix typing

This commit is contained in:
Matthias
2024-03-31 09:37:52 +02:00
parent b1c83ce189
commit dcb9d5d611
+7 -4
View File
@@ -7,7 +7,7 @@ so it can be used as a standalone script, and can be installed independently.
import json import json
import logging import logging
from typing import Optional from typing import Any, Dict, List, Optional, Union
from urllib.parse import urlencode, urlparse, urlunparse from urllib.parse import urlencode, urlparse, urlunparse
import requests import requests
@@ -16,6 +16,9 @@ from requests.exceptions import ConnectionError
logger = logging.getLogger("ft_rest_client") logger = logging.getLogger("ft_rest_client")
ParamsT = Optional[Dict[str, Any]]
PostDataT = Optional[Union[Dict[str, Any], List[Dict[str, Any]]]]
class FtRestClient: class FtRestClient:
@@ -58,13 +61,13 @@ class FtRestClient:
except ConnectionError: except ConnectionError:
logger.warning("Connection error") logger.warning("Connection error")
def _get(self, apipath, params: Optional[dict] = None): def _get(self, apipath, params: ParamsT = None):
return self._call("GET", apipath, params=params) return self._call("GET", apipath, params=params)
def _delete(self, apipath, params: Optional[dict] = None): def _delete(self, apipath, params: ParamsT = None):
return self._call("DELETE", apipath, params=params) return self._call("DELETE", apipath, params=params)
def _post(self, apipath, params: Optional[dict] = None, data: Optional[dict] = None): def _post(self, apipath, params: ParamsT = None, data: PostDataT = None):
return self._call("POST", apipath, params=params, data=data) return self._call("POST", apipath, params=params, data=data)
def start(self): def start(self):