feature(py): Adding initial spells to every new character

Author: Nanashii76

source/create_character.py

@@ -37,7 +37,7 @@
 
 class Character:
 
-    def __init__(self, conn, id = None):
+    def __init__(self, conn, id=None):
         self.conn = conn
         self.id = None
         self.nome = None
@@ -56,8 +56,8 @@ def __init__(self, conn, id = None):
         if not id:
             self.get_information()
         else:
-            character_info = self.get_character_info(conn, id) 
-            self.id = character_info[0] 
+            character_info = self.get_character_info(conn, id)
+            self.id = character_info[0]
             self.sub_regiao_id = character_info[1]
             self.nome = character_info[2]
             self.elemento = character_info[3]
@@ -71,63 +71,87 @@ def __init__(self, conn, id = None):
             self.inteligencia = character_info[11]
             self.moedas = character_info[12]
             self.nivel = character_info[13]
-    
-    def definy_initial_spells(self, conn):
-        try:
-            spells = initial_spells[self.elemento]
-
-            with conn.cursor() as cur:
-                for spell in spells:
-                    spell_id = spell[0]
-                    cur.execute(
-                        """
-                        SELECT aprender_feitico (%s, %s);
-                        """, (self.id, spell_id)
-                    )
-
-            debug(f"Initial Spells learned for '{self.nome}' with success")
-        except Exception as e:
-            debug(f"Error: {e}")
-            raise
 
     def get_information(self):
         print("\n === Criação de Personagem === ")
         self.nome = input("Digite o nome do personagem: ")
-        
+
         def ask():
             return input(f"Escolha o elemento ({', '.join(elements)}): ").lower()
-        
+
         elemento = ask()
         lower_case_elements = [elemento.lower() for elemento in elements]
         while elemento not in lower_case_elements:
             print("Elemento inválido.")
             elemento = ask()
-        
+
         self.elemento = elemento.capitalize()
-        # self.definy_initial_spells(self.conn)
 
     def add_database(self):
         try:
             with self.conn.cursor() as cur:
                 cur.execute("SELECT criar_personagem(%s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s)", (self.sub_regiao_id, self.nome, self.elemento, self.conhecimento_arcano, self.vida, self.vida_maxima, self.xp, self.xp_total, self.energia_arcana, self.energia_arcana_maxima, self.inteligencia, self.moedas, self.nivel))
-                self.id = cur.fetchone()[0]
-                self.conn.commit()
-
-                debug(f"Character: Personagem '{self.nome}' adicionado com sucesso!")
+                result = cur.fetchone()
+                if result:
+                    self.id = result[0]
+                    self.conn.commit()
+                    debug(f"Character: Personagem '{self.id}' '{self.nome}' adicionado com sucesso!")
+                else:
+                    debug("Character: Não foi retornado um ID após a criação do personagem.")
                 return self
 
         except Exception as e:
             debug(f"Character: Erro ao adicionar personagem: {e}")
 
-    def get_character_info(self, conn, id): 
+    def get_character_info(self, conn, id):
         with conn.cursor() as cur:
             cur.execute("""
                 SELECT *
                 FROM personagem
                 WHERE id = %s
             """, (id,))
-            result = cur.fetchone() 
+            result = cur.fetchone()
             return result
-    
 
+    def create_inventory_if_not_exists(self, conn):
+        with conn.cursor() as cur:
+            cur.execute(
+                "SELECT id FROM inventario WHERE personagem_id = %s", (self.id,)
+            )
+            result = cur.fetchone()
+
+            if not result:  # Se não existe, cria o inventário
+                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})")
+            else:
+                inventory_id = result[0]
+            
+            return inventory_id
+
+
+    def define_initial_spells(self, conn):
+        print(f"Definindo feitiços iniciais para {self.nome}...")  
+        try:
+            inventory_id = self.create_inventory_if_not_exists(conn)  
+            spells = initial_spells[self.elemento]
+
+            with conn.cursor() as cur:
+                for spell in spells:
+                    spell_id = spell[0]
+                    # print(f"Inserindo feitiço {spell_id} no inventário {inventory_id}")  # Debug
+                    cur.execute(
+                        """
+                        INSERT INTO feitico_aprendido (inventario_id, feitico_id)
+                        VALUES (%s, %s);
+                        """, (inventory_id, spell_id)
+                    )
+            conn.commit()
+        except Exception as e:
+            print(f"Erro ao definir feitiços iniciais: {e}")
+            conn.rollback()
 

source/database/ddl/procedures.sql

@@ -451,47 +451,57 @@ DECLARE
     v_possui_feitico_atual BOOLEAN;
     v_conhecimento_arcano_suficiente BOOLEAN;
 BEGIN
-    SELECT id
-    INTO v_inventario_id
+    -- get character inventory
+    SELECT id INTO v_inventario_id
     FROM inventario
     WHERE personagem_id = p_personagem_id;
 
-    SELECT de_id
-    INTO v_requisito_id
+    -- if doesn't have a inventory, return error
+    IF v_inventario_id IS NULL THEN
+        RAISE EXCEPTION 'O personagem % não possui inventário.', p_personagem_id;
+    END IF;
+
+    SELECT de_id INTO v_requisito_id
     FROM feitico_requerimento
     WHERE para_id = p_feitico_id;
 
-    -- Verifica se já possui o feitiço pré-requisito
-    SELECT EXISTS (
-        SELECT 1
-        FROM feitico_aprendido
-        WHERE inventario_id = v_inventario_id
-          AND feitico_id = v_requisito_id
+    SELECT COALESCE(
+        (SELECT EXISTS (
+            SELECT 1 FROM feitico_aprendido 
+            WHERE inventario_id = v_inventario_id AND feitico_id = v_requisito_id
+        )), TRUE
     ) INTO v_pre_requisito_aprendido;
 
-    -- Verifica se já possui o feitiço atual
     SELECT EXISTS (
-        SELECT 1
-        FROM feitico_aprendido
-        WHERE inventario_id = v_inventario_id
-          AND feitico_id = p_feitico_id
+        SELECT 1 FROM feitico_aprendido 
+        WHERE inventario_id = v_inventario_id AND feitico_id = p_feitico_id
     ) INTO v_possui_feitico_atual;
 
-    -- Verifica se possui conhecimento arcano suficiente para o feitiço
-    SELECT (p.conhecimento_arcano >= f.conhecimento_arcano_necessario)
-    INTO v_conhecimento_arcano_suficiente
+    SELECT COALESCE(
+        (p.conhecimento_arcano >= f.conhecimento_arcano_necessario), FALSE
+    ) INTO v_conhecimento_arcano_suficiente
     FROM personagem p
     JOIN feitico f ON f.id = p_feitico_id
     WHERE p.id = p_personagem_id;
 
-    -- Só aprende o feitiço se ainda não tiver, já possuir o pré-requisito e tiver conhecimento arcano
-    IF NOT v_possui_feitico_atual
-       AND v_pre_requisito_aprendido
-       AND v_conhecimento_arcano_suficiente THEN
-        INSERT INTO feitico_aprendido(inventario_id, feitico_id)
-        VALUES (v_inventario_id, p_feitico_id);
+    IF v_possui_feitico_atual THEN
+        RAISE NOTICE 'O personagem % já aprendeu este feitiço.', p_personagem_id;
+        RETURN p_personagem_id;
     END IF;
 
+    IF NOT v_pre_requisito_aprendido THEN
+        RAISE NOTICE 'O personagem % não tem o pré-requisito necessário para este feitiço.', p_personagem_id;
+        RETURN p_personagem_id;
+    END IF;
+
+    IF NOT v_conhecimento_arcano_suficiente THEN
+        RAISE NOTICE 'O personagem % não tem conhecimento arcano suficiente para aprender este feitiço.', p_personagem_id;
+        RETURN p_personagem_id;
+    END IF;
+
+    INSERT INTO feitico_aprendido(inventario_id, feitico_id)
+    VALUES (v_inventario_id, p_feitico_id);
+
     RETURN p_personagem_id;
 END;
 $$ LANGUAGE plpgsql;

source/interface.py

@@ -68,6 +68,32 @@ def inventory(character, conn):
 
     print("-" * 40)
 
+def list_spells(conn, character_id):
+    with conn.cursor() as cur:
+        cur.execute("""
+            SELECT feitico.descricao, feitico.energia_arcana
+            FROM inventario
+            JOIN feitico_aprendido ON inventario.id = feitico_aprendido.inventario_id
+            JOIN feitico ON feitico_aprendido.feitico_id = feitico.id
+            WHERE inventario.personagem_id = %s;
+        """, (character_id,))
+        
+        spells = cur.fetchall()
+
+    clear_screen()
+    print(Style.BRIGHT + Fore.CYAN + f"\n--- Feitiços Disponíveis ---\n" + Style.RESET_ALL)
+    
+    if not spells:
+        print(Fore.RED + "Nenhum feitiço aprendido." + Style.RESET_ALL)
+    else:
+        for descricao, custo in spells:
+            print(Fore.CYAN + f"  Custo de energia: {custo}" + Style.RESET_ALL)
+            print(f"  {descricao}")
+            print("-" * 40)
+    
+    input("\nPressione Enter para continuar...")
+
+
 def show_enemies(conn, character):
     enemys = list_enemys_subregion(conn, character.sub_regiao_id)
     if enemys:
@@ -271,7 +297,7 @@ def game_loop(conn):
         elif option == "criar":
             character = Character(conn) 
             character.add_database() 
-            character.definy_initial_spells(conn)
+            character.define_initial_spells(conn)
             debug(f"Personagem {character.nome} criado com sucesso!")
             ok = True
 
@@ -282,7 +308,8 @@ def game_loop(conn):
         print("1. Navegar")
         print("2. Ver status do personagem")
         print("3. Ver Inventário")
-        print("4. Sair")
+        print("4. Ver feitiços")
+        print("5. Sair")
 
         option = input("Escolha uma opção: ").lower()
         if option == "1":
@@ -294,6 +321,8 @@ def game_loop(conn):
             inventory(character, conn)
             input("\nPressione Enter para continuar...")
         elif option == "4":
+            list_spells(conn, character.id)
+        elif option == "5":
             print("Saindo do jogo...")
             break
         else: