Add missing types of parameter (#20)

Signed-off-by: Patrick ZAJDA <[email protected]>
Nardol · Nov 22, 2022 · 18e0727 · 18e0727
1 parent 67838c7
commit 18e0727
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/pysmsboxnet/api.py b/pysmsboxnet/api.py
@@ -12,7 +12,9 @@
 class Client:
     """API client class."""
 
-    def __init__(self, session: ClientSession, host: str, cleApi: str, timeout=30):
+    def __init__(
+        self, session: ClientSession, host: str, cleApi: str, timeout: int = 30
+    ):
         """Initialize the SMS."""
         self.host = host
         self.cleApi = cleApi

diff --git a/pysmsboxnet/exceptions.py b/pysmsboxnet/exceptions.py
@@ -4,7 +4,7 @@
 class SMSBoxException(BaseException):
     """Base exception for SMSBox API."""
 
-    def __init__(self, message="Unknown API error"):
+    def __init__(self, message: str = "Unknown API error"):
         """Initialize SMSBox base exception.
 
         An optional message can be specified.
@@ -58,6 +58,6 @@ def __init__(self):
 class HTTPException(SMSBoxException):
     """Exception when API returns ERROR 03."""
 
-    def __init__(self, errorCode):
+    def __init__(self, errorCode: int):
         """Initialize HTTP error Exception."""
         super().__init__(f"HTTP error ({errorCode})")