feature(py): Adding initial items, fixing trigger to update backpack weight and fixing query to get backpack items

Author: Nanashii76

source/create_character.py

@@ -120,14 +120,14 @@ def create_inventory_if_not_exists(self, conn):
             )
             result = cur.fetchone()
 
-            if not result:  # Se não existe, cria o inventário
+            if not result:  # if not exist, create new inventory
                 cur.execute(
                     "INSERT INTO inventario (personagem_id) VALUES (%s) RETURNING id",
                     (self.id,)
                 )
                 inventory_id = cur.fetchone()[0]
                 conn.commit()
-                print(f"✅ Inventário criado para {self.nome} (ID {inventory_id})")
+                print(f"Inventário criado para {self.nome} (ID {inventory_id})")
             else:
                 inventory_id = result[0]
 
@@ -155,3 +155,34 @@ def define_initial_spells(self, conn):
             print(f"Erro ao definir feitiços iniciais: {e}")
             conn.rollback()
 
+    def add_initial_items(self, conn):
+        try:
+            inventory_id = self.create_inventory_if_not_exists(conn)
+
+            elixir_da_vida_id = 133  
+            mana_liquida_id = 134  
+
+            with conn.cursor() as cur:
+                cur.execute("""
+                    INSERT INTO item_instancia (item_id, inventario_id)
+                    VALUES (%s, %s)
+                    RETURNING item_id;
+                """, (elixir_da_vida_id, inventory_id))
+
+                cur.execute("""
+                    INSERT INTO item_instancia (item_id, inventario_id)
+                    VALUES (%s, %s)
+                    RETURNING item_id;
+                """, (elixir_da_vida_id, inventory_id))
+
+                cur.execute("""
+                    INSERT INTO item_instancia (item_id, inventario_id)
+                    VALUES (%s, %s)
+                    RETURNING item_id;
+                """, (mana_liquida_id, inventory_id))
+
+            conn.commit()
+            print(f"Elixir da Vida e Mana Líquida adicionados ao inventário de {self.nome}.")
+        except Exception as e:
+            print(f"Erro ao adicionar itens iniciais: {e}")
+            conn.rollback()

source/database/ddl/triggers.sql

@@ -1,18 +1,29 @@
 -- PostGreSQL:
 -- `incrementar_peso_acessorio`:
 -- when character gets a new `acessorio`, it will increment the `peso` of the `mochila`, and if it surpasses `peso_total`, do not allow.
+-- PostGreSQL:
 CREATE OR REPLACE FUNCTION incrementar_peso_acessorio()
 RETURNS TRIGGER AS $$
 BEGIN
-    UPDATE mochila
-    SET peso = peso + NEW.peso
-    WHERE id = NEW.inventario_id;
+    -- search weight in table 'item' with 'item_id'
+    DECLARE
+        item_peso NUMERIC;
+    BEGIN
+        -- get item weight
+        SELECT peso INTO item_peso
+        FROM item
+        WHERE id = NEW.item_id;
 
-    IF (SELECT peso FROM mochila WHERE id = NEW.inventario_id) > (SELECT peso_total FROM mochila WHERE id = NEW.inventario_id) THEN
-        RAISE EXCEPTION 'Peso total da mochila excedido';
-    END IF;
+        UPDATE mochila
+        SET peso = peso + item_peso
+        WHERE id = NEW.inventario_id;
 
-    RETURN NEW;
+        IF (SELECT peso FROM mochila WHERE id = NEW.inventario_id) > (SELECT peso_total FROM mochila WHERE id = NEW.inventario_id) THEN
+            RAISE EXCEPTION 'Peso total da mochila excedido';
+        END IF;
+
+        RETURN NEW;
+    END;
 END;
 $$ LANGUAGE plpgsql;
 

source/interface.py

@@ -64,7 +64,7 @@ def inventory(character, conn):
     else:
        for item in items:
            nome, descricao, qtd = item
-           print(f"- {nome} ({descricao}): {qtd}")
+           print(f"- {nome} ({descricao}) - {qtd}")
 
     print("-" * 40)
 
@@ -298,6 +298,7 @@ def game_loop(conn):
             character = Character(conn) 
             character.add_database() 
             character.define_initial_spells(conn)
+            character.add_initial_items(conn)
             debug(f"Personagem {character.nome} criado com sucesso!")
             ok = True
 

source/queries/query.py

@@ -117,9 +117,25 @@ def list_character_id(conn, character_id):
 
 def list_item_inventory(conn, character_id):
     with conn.cursor() as cur:
-        cur.execute(f"SELECT i.nome, i.descricao FROM inventario inv JOIN item_instancia ii ON inv.id = ii.inventario_id JOIN item i ON ii.item_id = i.id WHERE inv.personagem_id = {character_id};")
-        result = cur.fetchall()
-        return result
+        cur.execute("""
+            SELECT id FROM inventario WHERE personagem_id = %s AND tipo = 'Mochila'
+        """, (character_id,))
+        
+        inventarios = cur.fetchall()
+        
+        items = []
+        for inventario in inventarios:
+            inventario_id = inventario[0]
+            cur.execute("""
+                SELECT item.nome, item.descricao, COUNT(item_instancia.id) as quantidade
+                FROM item_instancia
+                JOIN item ON item_instancia.item_id = item.id
+                WHERE item_instancia.inventario_id = %s
+                GROUP BY item.nome, item.descricao
+            """, (inventario_id,))
+            
+            items.extend(cur.fetchall())
+    return items
 
 def get_civilian_info(conn, npc_name):
     with conn.cursor() as cur: