diff options
-rw-r--r-- | pyfunkwhale/client.py | 4 | ||||
-rw-r--r-- | pyfunkwhale/utils.py | 4 |
2 files changed, 5 insertions, 3 deletions
diff --git a/pyfunkwhale/client.py b/pyfunkwhale/client.py index 2221a35..49df688 100644 --- a/pyfunkwhale/client.py +++ b/pyfunkwhale/client.py @@ -4,6 +4,7 @@ import json from time import time from requests_oauthlib import OAuth2Session +from requests.models import Response from pyfunkwhale.utils import read_file, write_file @@ -70,7 +71,8 @@ class Client(object): client_secret=self.client_secret) write_file(self.token_filename, self.token) - def call(self, endpoint, method, params=None, data=None): + def call(self, endpoint: str, method: str, params: dict = None, + data: dict = None) -> Response: """ Call the API diff --git a/pyfunkwhale/utils.py b/pyfunkwhale/utils.py index eb6da03..b92d2b9 100644 --- a/pyfunkwhale/utils.py +++ b/pyfunkwhale/utils.py @@ -4,7 +4,7 @@ import json -def read_file(filename): +def read_file(filename: str) -> str: """ Simple wrapper for read a file content @@ -19,7 +19,7 @@ def read_file(filename): return data -def write_file(filename, datas): +def write_file(filename: str, datas: str) -> str: """ Simple wrapper for write data in file |