Fix small bug again

cuappdev · Mar 13, 2024 · e6552cf · e6552cf
1 parent 30bedc2
commit e6552cf
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/category/controllers/populate_category.py b/src/category/controllers/populate_category.py
@@ -24,8 +24,10 @@ def generate_dining_hall_categories(self, json_event, event):
          "Halal", "Kosher Station", "Flat Top Grill"]
 
         def sort_menu(menu):
-            order = category_order.find(menu["category"].strip())
-            return order if order != -1 else len(category_order)
+            try:
+                return category_order.index(menu["category"].strip())
+            except ValueError:
+                return len(category_order)
 
         for json_menu in sorted(json_event["menu"], key=sort_menu):
             data = {"event": event, "category": json_menu["category"]}