Add type annotations to player.py and status.py

Signed-off-by: Olivier Gayot <olivier.gayot@sigexec.com>

2 files changed, 9 insertions, 8 deletions

diff --git a/swiftstory/player.py b/swiftstory/player.py
index 011ae51..525a4e6 100644
--- a/swiftstory/player.py
+++ b/swiftstory/player.py
@@ -1,11 +1,11 @@
 import asyncio
 import json
-from typing import Any
+from typing import Any, Dict, Tuple
 
 
 class Player:
-    def __init__(self):
-        self.cards = {}
+    def __init__(self) -> None:
+        self.cards: Dict[int, Tuple[int, str]] = {}
         self.next_idx = 0
 
         self.score = 0
@@ -15,17 +15,17 @@ class Player:
         self.name = 'default'
         self.notifications: asyncio.Queue = asyncio.Queue()
 
-    def pop_card(self, card_id):
+    def pop_card(self, card_id: int) -> Tuple[int, str]:
         return self.cards.pop(card_id)
 
-    def inc_score(self):
+    def inc_score(self) -> None:
         self.score += 1
         self.register_notification({
             'op': 'updated_score',
             'content': self.score,
             })
 
-    def receive_card(self, card):
+    def receive_card(self, card: Tuple[int, str]) -> int:
         self.cards[self.next_idx] = card
         self.next_idx += 1
         return self.next_idx - 1
diff --git a/swiftstory/status.py b/swiftstory/status.py
index fb8347b..d794580 100644
--- a/swiftstory/status.py
+++ b/swiftstory/status.py
@@ -1,8 +1,9 @@
+from typing import Any
 import json
 
 
-def error(msg, code=255):
+def error(msg: str, code: int = 255) -> str:
     return json.dumps({'type': 'response', 'content': {'status': code, 'info': msg}})
 
-def success(obj):
+def success(obj: Any) -> str:
     return json.dumps({'type': 'response', 'content': {'status': 0, 'result': obj}})