feat(attacks): Limit to Pokémon type or Colorless

modules/details.py

@@ -74,26 +74,32 @@ def rand_weight():
     return f'{random_weight} lbs.'
 
 
-def rand_attack(attacks, name):
+def rand_attack(attacks, name, energy_type=None, colorless_only_allowed=False):
     random_attack = random.choices(attacks)[0]
+
+    if type and not colorless_only_allowed:
+        while energy_type not in random_attack["cost"]:
+            random_attack = random.choices(attacks)[0]
+    elif type and  colorless_only_allowed:
+        while energy_type not in random_attack["cost"] and 'Colorless' not in random_attack["cost"]:
+            random_attack = random.choices(attacks)[0]
+
     random_attack['text'] = random_attack['text'].replace('<name>', name)
 
     return random_attack
 
 
-def rand_attacks(attacks, name, n=2):
-    attack1 = rand_attack(attacks, name)
+def rand_attacks(attacks, name, energy_type=None, n=2):
+    attack1 = rand_attack(attacks, name, energy_type)
 
     if n > 1:
-        attack2 = rand_attack(attacks, name)
+        attack2 = rand_attack(attacks, name, energy_type, True)
         while attack1['text'] == attack2['text']:
-            attack2 = rand_attack(attacks, name)
+            attack2 = rand_attack(attacks, name, energy_type, True)
         return [attack1, attack2]
     else:
         return [attack1]
 
-    return attacks
-
 
 def rand_retreat():
     return random.randrange(0, 4, 1)
@@ -118,7 +124,7 @@ def rand_details(lists=load_lists(['attacks', 'descriptions', 'species'])):
         "species": rand_species(lists["species"]),
         "length": rand_length(),
         "weight": rand_weight(),
-        "attacks": rand_attacks(lists["attacks"], name),
+        "attacks": rand_attacks(lists["attacks"], name, energy_type=energy_type),
         "weakness": rand_type(can_be_none=True),
         "resistance": rand_type(can_be_none=True),
         "retreat": rand_retreat(),