feature(py): Populating storage enemies

Author: Nanashii76

source/database/defaults/storage/enemie_storage.py

@@ -0,0 +1,57 @@
+from database import Database
+from utils import debug, error
+from colorama import Style
+import random
+
+def enemie_storage(db: Database):
+
+    table_name = Style.BRIGHT + "ARMAZENAMENTO INIMIGO" + Style.NORMAL
+
+    try:
+        db.cur.execute("SELECT id, vida_maxima FROM inimigo")  
+        inimigos = db.cur.fetchall()
+
+        db.cur.execute("SELECT id, tipo, nome, drop_inimigos_media FROM item")  
+        itens = db.cur.fetchall()
+
+        valores_insercao = []
+
+        for inimigo in inimigos:
+            inimigo_id, vida_maxima = inimigo
+
+            if vida_maxima < 50:
+                categoria = 1 
+            elif vida_maxima < 100:
+                categoria = 2 
+            else:
+                categoria = 3  
+
+            for item in itens:
+                item_id, tipo, nome, drop_inimigos_media = item
+
+                if categoria == 1:
+                    chance = random.randint(1, 3) 
+                elif categoria == 2:
+                    chance = random.randint(1, 6)  
+                else:
+                    chance = random.randint(1, 10) 
+
+                if chance == 1:
+                    valores_insercao.append((inimigo_id, item_id))
+
+        db.cur.executemany(
+            """
+            INSERT INTO armazenamento_inimigo (inimigo_id, armazenamento_id)
+            VALUES (%s, %s)
+            """, valores_insercao
+        )
+
+        db.conn.commit()
+        debug(f"default: {table_name} updated successfully!")
+
+        return len(valores_insercao)
+
+    except Exception as e:
+        db.conn.rollback()
+        error(f"default: error occurred while populating {table_name}: {e}")
+        return 0

source/database/defaults/storage/storage.py

@@ -31,7 +31,7 @@ def storage(item_name: str, quantity: int, db: Database) -> int:
 
 def populate_storage(db: Database):
 
-    table_name = Style.BRIGHT + "STORAGE" + Style.NORMAL
+    table_name = Style.BRIGHT + "ARMAZENAMENTO" + Style.NORMAL
 
     try:
         db.cur.execute("SELECT nome FROM item;")

source/default.py

@@ -7,7 +7,7 @@
 
 from database.defaults.npc.citizen import citizens, merchants, questers
 
-from database.defaults.storage import quest, storage
+from database.defaults.storage import quest, storage, enemie_storage
 
 from database.defaults.map import regions as r, sub_regions as sr, sub_regions_connections as src
 
@@ -63,8 +63,9 @@ def populate_database(db: Database):
 
         writted_scroll.writted_scrolls(db, scrolls_id_start)
 
-        quest.quests(db)
         storage.populate_storage(db)
+        enemie_storage.enemie_storage(db)
+        quest.quests(db)
 
         #potions
         potions.potions(db)