AWAY, EVIL EXCEPTION

pwncollege · Jan 13, 2024 · de0821e · de0821e
1 parent 14ab4f9
commit de0821e
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/dojo_plugin/api/v1/belts.py b/dojo_plugin/api/v1/belts.py
@@ -20,13 +20,13 @@ def get_belts():
     }
 
     for n,(color,dojo_id) in enumerate(BELT_REQUIREMENTS.items()):
-        result["dates"][color] = {}
-        try:
-            dojo = Dojos.query.filter_by(id=dojo_id).first()
-        except sqlalchemy.exc.NoResultsFound:
+        dojo = Dojos.query.filter_by(id=dojo_id).first()
+        if not dojo:
             # We are likely missing the correct dojos in the DB (e.g., custom deployment)
             break
 
+        result["dates"][color] = {}
+
         for user,date in dojo.completions():
             if result["users"].get(user.id, {"rank_id":-1})["rank_id"] != n-1:
                 continue