From 0b6991b77fea008163d08ccc1777ac405e2b0a41 Mon Sep 17 00:00:00 2001 From: Olivier Gayot Date: Thu, 23 Dec 2021 17:42:57 +0100 Subject: Add type annotations to player.py and status.py Signed-off-by: Olivier Gayot --- swiftstory/player.py | 12 ++++++------ swiftstory/status.py | 5 +++-- 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}}) -- cgit v1.2.3