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, 9):  # 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()