AIdeaText commited on
Commit
755d1f1
·
verified ·
1 Parent(s): 4ae25ba

Update modules/database/sql_db.py

Browse files
Files changed (1) hide show
  1. modules/database/sql_db.py +11 -3
modules/database/sql_db.py CHANGED
@@ -17,7 +17,8 @@ def get_user(username, role=None):
17
  if role:
18
  query += f" AND c.role = '{role}'"
19
  items = list(container.query_items(
20
- query=query # Removido enable_cross_partition_query
 
21
  ))
22
  return items[0] if items else None
23
  except Exception as e:
@@ -36,6 +37,10 @@ def get_student_user(username):
36
  def get_teacher_user(username):
37
  return get_user(username, role='Profesor')
38
 
 
 
 
 
39
  def create_user(username, password, role, additional_info=None):
40
  """Crea un nuevo usuario"""
41
  container = get_container("users")
@@ -49,12 +54,13 @@ def create_user(username, password, role, additional_info=None):
49
  'password': password,
50
  'role': role,
51
  'timestamp': datetime.now(timezone.utc).isoformat(),
52
- 'additional_info': additional_info or {}
 
53
  }
54
 
55
  container.create_item(
56
  body=user_data,
57
- partition_key=username
58
  )
59
  logger.info(f"Usuario {role} creado: {username}")
60
  return True
@@ -63,6 +69,8 @@ def create_user(username, password, role, additional_info=None):
63
  logger.error(f"Error al crear usuario {role}: {str(e)}")
64
  return False
65
 
 
 
66
  def create_student_user(username, password, additional_info=None):
67
  return create_user(username, password, 'Estudiante', additional_info)
68
 
 
17
  if role:
18
  query += f" AND c.role = '{role}'"
19
  items = list(container.query_items(
20
+ query=query,
21
+ enable_cross_partition_query=True # Añadimos esta línea
22
  ))
23
  return items[0] if items else None
24
  except Exception as e:
 
37
  def get_teacher_user(username):
38
  return get_user(username, role='Profesor')
39
 
40
+
41
+
42
+ ###################################################
43
+
44
  def create_user(username, password, role, additional_info=None):
45
  """Crea un nuevo usuario"""
46
  container = get_container("users")
 
54
  'password': password,
55
  'role': role,
56
  'timestamp': datetime.now(timezone.utc).isoformat(),
57
+ 'additional_info': additional_info or {},
58
+ 'partitionKey': username # Añadimos esta línea
59
  }
60
 
61
  container.create_item(
62
  body=user_data,
63
+ enable_cross_partition_query=True # Añadimos esta línea
64
  )
65
  logger.info(f"Usuario {role} creado: {username}")
66
  return True
 
69
  logger.error(f"Error al crear usuario {role}: {str(e)}")
70
  return False
71
 
72
+
73
+ #######################################################
74
  def create_student_user(username, password, additional_info=None):
75
  return create_user(username, password, 'Estudiante', additional_info)
76