feat: implementa sistema de comprovantes com centralizações e PIX
This commit is contained in:
65
create_test_users.py
Normal file
65
create_test_users.py
Normal file
@@ -0,0 +1,65 @@
|
||||
from functions.database import Usuario, Role, get_db_connection
|
||||
|
||||
def create_test_users():
|
||||
"""Cria usuários de teste para o sistema"""
|
||||
session = get_db_connection()
|
||||
try:
|
||||
# Buscar roles
|
||||
secretario_celula = session.query(Role).filter_by(nivel=Role.SECRETARIO_CELULA).first()
|
||||
secretario_setor = session.query(Role).filter_by(nivel=Role.SECRETARIO_SETOR).first()
|
||||
secretario_cr = session.query(Role).filter_by(nivel=Role.SECRETARIO_CR).first()
|
||||
secretario_geral = session.query(Role).filter_by(nivel=Role.SECRETARIO_GERAL).first()
|
||||
|
||||
# Criar usuários de teste
|
||||
usuarios = [
|
||||
{
|
||||
'username': 'celula',
|
||||
'email': 'celula@example.com',
|
||||
'password': 'celula123',
|
||||
'role': secretario_celula,
|
||||
'tipo': 'SECRETARIO_CELULA'
|
||||
},
|
||||
{
|
||||
'username': 'setor',
|
||||
'email': 'setor@example.com',
|
||||
'password': 'setor123',
|
||||
'role': secretario_setor,
|
||||
'tipo': 'SECRETARIO_SETOR'
|
||||
},
|
||||
{
|
||||
'username': 'cr',
|
||||
'email': 'cr@example.com',
|
||||
'password': 'cr123',
|
||||
'role': secretario_cr,
|
||||
'tipo': 'SECRETARIO_CR'
|
||||
},
|
||||
{
|
||||
'username': 'geral',
|
||||
'email': 'geral@example.com',
|
||||
'password': 'geral123',
|
||||
'role': secretario_geral,
|
||||
'tipo': 'SECRETARIO_GERAL'
|
||||
}
|
||||
]
|
||||
|
||||
for user_data in usuarios:
|
||||
# Verificar se o usuário já existe
|
||||
if not session.query(Usuario).filter_by(username=user_data['username']).first():
|
||||
user = Usuario(
|
||||
username=user_data['username'],
|
||||
email=user_data['email']
|
||||
)
|
||||
user.set_password(user_data['password'])
|
||||
user.tipo = user_data['tipo']
|
||||
user.roles.append(user_data['role'])
|
||||
session.add(user)
|
||||
|
||||
session.commit()
|
||||
print("Usuários de teste criados com sucesso!")
|
||||
|
||||
except Exception as e:
|
||||
print(f"Erro ao criar usuários de teste: {e}")
|
||||
session.rollback()
|
||||
raise
|
||||
finally:
|
||||
session.close()
|
||||
Reference in New Issue
Block a user