Small refactoring

Author: laike9m

views/utils.py

@@ -7,7 +7,6 @@
 import tokenize
 from dataclasses import dataclass, field
 from enum import StrEnum
-from functools import lru_cache
 from pathlib import Path
 from typing import ClassVar
 
@@ -41,12 +40,6 @@ def parse_code(self):
         self.user_code, _, self.test_code = self.code.partition(self.CODE_SPLITTER)
 
 
-@dataclass(frozen=True, slots=True)
-class ChallengeInfo:
-    name: ChallengeName
-    level: Level
-
-
 @dataclass(frozen=True, slots=True)
 class TypeCheckResult:
     message: str
@@ -56,20 +49,7 @@ class TypeCheckResult:
 class ChallengeManager:
     def __init__(self):
         self.challenges: dict[ChallengeName, Challenge] = self._load_challenges()
-        self.challenges_groupby_level = self.get_challenges_groupby_level()
-
-    def get_challenges_groupby_level(self) -> dict[Level, list[ChallengeName]]:
-        groups: dict[str, list[ChallengeName]] = {}
-
-        for challenge in self.challenges.values():
-            groups.setdefault(challenge.level, []).append(challenge.name)
-
-        # Sort challenge by name alphabetically.
-        for challenge_names in groups.values():
-            challenge_names.sort()
-
-        # Make sure groups are ordered by level (from easy to hard)
-        return {level: groups[level] for level in Level}
+        self.challenges_groupby_level = self._get_challenges_groupby_level()
 
     def has_challenge(self, name: str) -> bool:
         return name in self.challenges
@@ -97,6 +77,19 @@ def _load_challenges() -> dict[ChallengeName, Challenge]:
 
         return challenges
 
+    def _get_challenges_groupby_level(self) -> dict[Level, list[ChallengeName]]:
+        groups: dict[str, list[ChallengeName]] = {}
+
+        for challenge in self.challenges.values():
+            groups.setdefault(challenge.level, []).append(challenge.name)
+
+        # Sort challenge by name alphabetically.
+        for challenge_names in groups.values():
+            challenge_names.sort()
+
+        # Make sure groups are ordered by level (from easy to hard)
+        return {level: groups[level] for level in Level}
+
     EXPECT_ERROR_COMMENT = "expect-type-error"
 
     # Pyright error messages look like: