30 lines
1.2 KiB
Python
30 lines
1.2 KiB
Python
from extensions import db
|
|
from models.user_service_role import UserServiceRole
|
|
|
|
def seed_user_service_roles():
|
|
# Definindo um usuário com até 10 diferentes service_role_ids para a mesma service_instance_id
|
|
user_id = 1 # Exemplo: um único usuário (id 1)
|
|
service_instance_id = 1 # Exemplo: instância de serviço para associar os papéis
|
|
|
|
# Criar até 10 service_role_ids para o usuário 1 na instância 1
|
|
for service_role_id in range(1, 11): # Gerando service_role_id de 1 até 10
|
|
user_role = UserServiceRole(
|
|
user_id=user_id,
|
|
service_instance_id=service_instance_id,
|
|
service_role_id=service_role_id
|
|
)
|
|
|
|
# Verificar se a combinação de user_id, service_instance_id e service_role_id já existe
|
|
existing_role = db.session.query(UserServiceRole).filter_by(
|
|
user_id=user_role.user_id,
|
|
service_instance_id=user_role.service_instance_id,
|
|
service_role_id=user_role.service_role_id
|
|
).first()
|
|
|
|
# Se não existir, adicionar à sessão
|
|
if not existing_role:
|
|
db.session.add(user_role)
|
|
|
|
# Commit a transação para salvar os dados no banco
|
|
db.session.commit()
|