bot/db_utils.py

@@ -26,14 +26,14 @@ def get_random_character() -> Character | None:
     if not characters:
         return None
 
-    weights = [c['weight'] for c in characters]
+    weights = [config.RARITY_TO_WEIGHT[c['rarity']] for c in characters]
     chosen = choices(characters, weights=weights, k=1)[0]
 
     return {
         'id': chosen['id'],
         'name': chosen['name'],
         'rarity': chosen['rarity'],
-        'weight': chosen['weight'],
+        'weight': config.RARITY_TO_WEIGHT[chosen['rarity']],
         'image_url': chosen['file_id']
     }