diff --git a/src/arcaea_offline/utils/__init__.py b/src/arcaea_offline/utils/__init__.py index e69de29..091fbdb 100644 --- a/src/arcaea_offline/utils/__init__.py +++ b/src/arcaea_offline/utils/__init__.py @@ -0,0 +1,23 @@ +from typing import Optional + +RATING_CLASS_TEXT_MAP = { + 0: "Past", + 1: "Present", + 2: "Future", + 3: "Beyond", +} + +RATING_CLASS_SHORT_TEXT_MAP = { + 0: "PST", + 1: "PRS", + 2: "FTR", + 3: "BYD", +} + + +def rating_class_to_text(rating_class: int) -> Optional[str]: + return RATING_CLASS_TEXT_MAP.get(rating_class) + + +def rating_class_to_short_text(rating_class: int) -> Optional[str]: + return RATING_CLASS_SHORT_TEXT_MAP.get(rating_class)