From f493abacf63fa72c6d648d3550c5a28c1b8aa00e Mon Sep 17 00:00:00 2001 From: 283375 Date: Sat, 17 Jun 2023 02:36:43 +0800 Subject: [PATCH] feat(utils): rating_class (int) to str --- src/arcaea_offline/utils/__init__.py | 23 +++++++++++++++++++++++ 1 file changed, 23 insertions(+) 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)