move belt assets helper into utils

dojo_plugin/api/v1/scoreboard.py

@@ -20,7 +20,7 @@
 from ...models import Dojos, DojoChallenges, DojoUsers, DojoMembers, DojoAdmins, DojoStudents, DojoModules, DojoChallengeVisibilities
 from ...utils import dojo_standings, user_dojos, first_bloods, daily_solve_counts
 from ...utils.dojo import dojo_route, dojo_accessible
-from ...utils.belts import get_belts
+from ...utils.belts import get_belts, belt_asset
 
 SCOREBOARD_CACHE_TIMEOUT_SECONDS = 60 * 60 * 2 # two hours make to cache all scoreboards
 scoreboard_namespace = Namespace("scoreboard")
@@ -34,18 +34,6 @@ def email_symbol_asset(email):
         group = "hacker.png"
     return url_for("views.themes", path=f"img/dojo/{group}")
 
-
-def belt_asset(color):
-    if color == "black":
-        belt = "black.svg"
-    elif color == "blue":
-        belt = "blue.svg"
-    elif color == "yellow":
-        belt = "yellow.svg"
-    else:
-        belt = "white.svg"
-    return url_for("views.themes", path=f"img/dojo/{belt}")
-
 @cache.memoize(timeout=SCOREBOARD_CACHE_TIMEOUT_SECONDS)
 def get_scoreboard_for(model, duration):
     duration_filter = (

dojo_plugin/utils/belts.py

@@ -1,4 +1,5 @@
 from CTFd.cache import cache
+from flask import url_for
 from ..models import Dojos
 
 
@@ -9,6 +10,10 @@
     "blue": "software-exploitation",
 }
 
+def belt_asset(color):
+    belt = color + ".svg" if color in BELT_REQUIREMENTS else "white.svg"
+    return url_for("views.themes", path=f"img/dojo/{belt}")
+
 def get_user_belts(user):
     result = [ ]
     for belt, dojo_id in BELT_REQUIREMENTS.items():