aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--pyfunkwhale/client.py4
-rw-r--r--pyfunkwhale/utils.py4
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