danielraynaud commited on
Commit
547a09d
·
verified ·
1 Parent(s): 326b98d

Update database/db_operations.py

Browse files
Files changed (1) hide show
  1. database/db_operations.py +3 -6
database/db_operations.py CHANGED
@@ -8,8 +8,7 @@ from .db_manager import DatabaseManager
8
  logging.basicConfig(level=logging.INFO)
9
  logger = logging.getLogger(__name__)
10
 
11
- def adicionar_lote_questoes(db: DatabaseManager, questoes: list) -> Tuple[bool, str]:
12
- """Adiciona várias questões de uma vez"""
13
  try:
14
  conn = db.get_connection()
15
  cursor = conn.cursor()
@@ -18,7 +17,7 @@ def adicionar_lote_questoes(db: DatabaseManager, questoes: list) -> Tuple[bool,
18
  cursor.execute('''
19
  INSERT INTO previous_questions
20
  (year, area, question_text, options, correct_answer,
21
- explanation, difficulty, references, tags)
22
  VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)
23
  ''', (
24
  questao.get('year'),
@@ -28,13 +27,11 @@ def adicionar_lote_questoes(db: DatabaseManager, questoes: list) -> Tuple[bool,
28
  questao.get('correct_answer'),
29
  questao.get('explanation'),
30
  questao.get('difficulty', 'medium'),
31
- json.dumps(questao.get('references', [])),
32
  json.dumps(questao.get('tags', []))
33
  ))
34
 
35
  conn.commit()
36
- logger.info(f"Adicionadas {len(questoes)} questões com sucesso")
37
  return True, "Questões adicionadas com sucesso"
38
  except Exception as e:
39
- logger.error(f"Erro ao adicionar questões: {e}")
40
  return False, f"Erro ao adicionar questões: {str(e)}"
 
8
  logging.basicConfig(level=logging.INFO)
9
  logger = logging.getLogger(__name__)
10
 
11
+ def adicionar_lote_questoes(db: DatabaseManager, questoes: list):
 
12
  try:
13
  conn = db.get_connection()
14
  cursor = conn.cursor()
 
17
  cursor.execute('''
18
  INSERT INTO previous_questions
19
  (year, area, question_text, options, correct_answer,
20
+ explanation, difficulty, ref_sources, tags) # Alterado aqui
21
  VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)
22
  ''', (
23
  questao.get('year'),
 
27
  questao.get('correct_answer'),
28
  questao.get('explanation'),
29
  questao.get('difficulty', 'medium'),
30
+ json.dumps(questao.get('references', [])), # O nome no dicionário pode continuar como references
31
  json.dumps(questao.get('tags', []))
32
  ))
33
 
34
  conn.commit()
 
35
  return True, "Questões adicionadas com sucesso"
36
  except Exception as e:
 
37
  return False, f"Erro ao adicionar questões: {str(e)}"