1 files changed, 82 insertions, 0 deletions
diff --git a/cogs/adv_inc/Enemy.py b/cogs/adv_inc/Enemy.py
new file mode 100644
index 0000000..f5f76e6
--- /dev/null
+++ b/cogs/adv_inc/Enemy.py
@@ -0,0 +1,82 @@
+import pymysql
+import math
+import random
+import importlib
+import utils
+
+importlib.reload(utils)
+
+
+async def create_enemy(enemy_stats, player_stats, params):
+    # Update player state for combat
+    await utils.sql_postgres("CALL adventurersinc.set_playerstate_combat(%s)", (params['nick'],), False)
+
+    # If challenge_rating is an INT we'll swap for a boss
+    if player_stats['challenge_rating'] == int(math.ceil(player_stats['challenge_rating'] / 4) * 4):
+        str_class = "boss"
+    else:
+        str_class = "normal"
+
+    # Get enemy weighting list
+    all_enemies = await utils.sql_postgres(f"SELECT adventurersinc.get_all_enemies(%s)", (str_class,), True)
+    int_total_weight = random.randint(1, all_enemies[0][0]['total_weight'])
+
+    # Apply challenge rating modifier to enemy
+    int_challenge_modifier = player_stats['challenge_rating'] / 2
+
+    for enemy in all_enemies:
+        int_total_weight = int_total_weight - enemy[0]['enemy_weight']
+        if int_total_weight <= 0:
+            # Set stats
+            enemy[0]['enemy_colour'] = enemy[0]['enemy_colour']
+            enemy[0]['enemy_name'] = str(enemy[0]['enemy_name'])
+            enemy[0]['enemy_class'] = str(enemy[0]['enemy_class'])
+            enemy[0]['unlock'] = str(enemy[0]['unlock'])
+            enemy[0]['hp'] = math.floor(enemy[0]['hp'] * int_challenge_modifier)
+            enemy[0]['enemy_weight'] = math.floor(enemy[0]['enemy_weight'] * int_challenge_modifier)
+            enemy[0]['challenge_rating'] = math.floor(enemy[0]['challenge_rating'] * int_challenge_modifier)
+            enemy[0]['gold_carried'] = math.floor(enemy[0]['gold_carried'] * int_challenge_modifier)
+            enemy[0]['strength'] = math.floor(enemy[0]['strength'] * int_challenge_modifier)
+            enemy[0]['dexterity'] = math.floor(enemy[0]['dexterity'] * int_challenge_modifier)
+            enemy[0]['constitution'] = math.floor(enemy[0]['constitution'] * int_challenge_modifier)
+            enemy[0]['intelligence'] = math.floor(enemy[0]['intelligence'] * int_challenge_modifier)
+            enemy[0]['dodge'] = math.floor(enemy[0]['dodge'] * int_challenge_modifier)
+            enemy[0]['crit'] = math.floor(enemy[0]['crit'] * int_challenge_modifier)
+            enemy[0]['accuracy'] = math.floor(enemy[0]['accuracy'] * int_challenge_modifier)
+            enemy[0]['resistance'] = math.floor(enemy[0]['resistance'] * int_challenge_modifier)
+            enemy[0]['luck'] = math.floor(enemy[0]['luck'] * int_challenge_modifier)
+            enemy[0]['block'] = math.floor(enemy[0]['block'] * int_challenge_modifier)
+            break
+
+    await utils.sql_postgres("CALL adventurersinc.add_player_enemy(%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s)",
+                             (params['nick'], enemy[0]['enemy_colour'], enemy[0]['enemy_name'], enemy[0]['enemy_class'], enemy[0]['unlock'], enemy[0]['hp'], enemy[0]['enemy_weight'],enemy[0]['challenge_rating'], enemy[0]['gold_carried'],
+                              enemy[0]['strength'], enemy[0]['dexterity'], enemy[0]['constitution'], enemy[0]['intelligence'], enemy[0]['dodge'], enemy[0]['crit'], enemy[0]['accuracy'], enemy[0]['resistance'], enemy[0]['luck'], enemy[0]['block'],), False)
+
+    return enemy[0]
+
+
+async def get_enemy_stats(params):
+    enemy = await utils.sql_postgres("SELECT adventurersinc.get_player_enemy(%s)", (params['nick'],), True)
+    if enemy[0][0] is None:
+        return None
+
+    enemy[0][0]['enemy_colour'] = enemy[0][0]['enemy_colour']
+    enemy[0][0]['enemy_name'] = enemy[0][0]['enemy_name']
+    enemy[0][0]['enemy_class'] = enemy[0][0]['enemy_class']
+    enemy[0][0]['unlock'] = enemy[0][0]['unlock']
+    enemy[0][0]['hp'] = enemy[0][0]['hp']
+    enemy[0][0]['gold_carried'] = enemy[0][0]['gold_carried']
+    enemy[0][0]['strength'] = enemy[0][0]['strength']
+    enemy[0][0]['dexterity'] = enemy[0][0]['dexterity']
+    enemy[0][0]['constitution'] = enemy[0][0]['constitution']
+    enemy[0][0]['intelligence'] = enemy[0][0]['intelligence']
+    enemy[0][0]['dodge'] = enemy[0][0]['dodge']
+    enemy[0][0]['crit'] = enemy[0][0]['crit']
+    enemy[0][0]['accuracy'] = enemy[0][0]['accuracy']
+    enemy[0][0]['resistance'] = enemy[0][0]['resistance']
+    enemy[0][0]['luck'] = enemy[0][0]['luck']
+    enemy[0][0]['block'] = enemy[0][0]['block']
+    enemy[0][0]['armour'] = 0
+    enemy[0][0]['unit_type'] = 'enemy'
+
+    return enemy[0][0]