diff options
Diffstat (limited to 'src/model')
-rw-r--r-- | src/model/difficulty.py | 2 | ||||
-rw-r--r-- | src/model/status.py | 30 |
2 files changed, 31 insertions, 1 deletions
diff --git a/src/model/difficulty.py b/src/model/difficulty.py index e38f19b..526cdb9 100644 --- a/src/model/difficulty.py +++ b/src/model/difficulty.py @@ -23,7 +23,7 @@ def parse(string: str) -> Optional[Difficulty]: if string == "Easy": return Difficulty.EASY elif string == "Normal": - return Difficulty.MIDDLE + return Difficulty.NORMAL elif string == "Hard": return Difficulty.HARD else: diff --git a/src/model/status.py b/src/model/status.py new file mode 100644 index 0000000..6881e0a --- /dev/null +++ b/src/model/status.py @@ -0,0 +1,30 @@ +from enum import IntEnum +from typing import Optional + +class Status(IntEnum): + READY = 0 + WAITING = 1 + MAYBE = 2 + +values = [ + Status.READY, + Status.WAITING, + Status.MAYBE] + +def format(status: Status) -> str: + if status == Status.READY: + return "Ready" + elif status == Status.WAITING: + return "Waiting" + elif status == Status.MAYBE: + return "Maybe" + +def parse(string: str) -> Optional[Status]: + if string == "Ready": + return Status.READY + elif string == "Waiting": + return Status.WAITING + elif string == "Maybe": + return Status.MAYBE + else: + return None |