cem_vezes_comanda.py

import os
import glob
import shlex
import subprocess 
import concurrent.futures
from itertools import product


arquivos_teste = ['cem_vezes_simulacao.py']

def executar_arquivo(arquivo):
    try:
        num_round = [20]
        total_clients = [20]
        modelos = ['DNN']
        niid_iid = ['IID']        
        ataques = ['INVERTE_CONVEGENCIA']
        data_set = ['MNIST']                        
        alpha_dirichlet = [0.0]
        noise_gaussiano = [0.0]
        round_inicio = [4]
        per_cents_atacantes = [40]        

        combinacoes_unicas = set() 

        for i, j, k, l, m, n, o, p, q, r in product(niid_iid, ataques, data_set, modelos, round_inicio, per_cents_atacantes, noise_gaussiano, alpha_dirichlet, num_round, total_clients):
            combinacao = (i, j, k, l, m, n, o, p, q, r) 
            
            if i == 'NIID' and p == 0:                             
                continue
            elif i == 'IID' and p > 0:               
                continue
            elif j != 'RUIDO_GAUSSIANO' and o > 0:   
                continue
            elif j == 'RUIDO_GAUSSIANO' and o != 0:
                continue            
            elif (k == 'MNIST' and l == 'CNN') or (k == 'CIFAR10' and l == 'DNN'):
                continue
            else:
                if combinacao not in combinacoes_unicas:           
                    combinacoes_unicas.add(combinacao)
         

            print(f'Executando {arquivo}')                
            comando = f'python3 {arquivo} --iid_niid {i} --modo_ataque {j} --dataset {k} --modelo_definido {l} --round_inicio {m} --per_cents_atacantes {n} --noise_gaussiano {o} --alpha_dirichlet {p} --num_rounds {q}  --total_clients {r}'
                    
            print(f'\n\n################################################################################################')
            print(f'\n\n{comando}\n\n')
            print(f'################################################################################################\n\n')
            subprocess.run(shlex.split(comando), check=True)

            print(f'Executou com sucesso: {arquivo}')

    except subprocess.CalledProcessError as e:
        print(f'Erro ao executar o arquivo: {e}')
    except Exception as e:
        print(f'Erro inesperado: {e}')


def processar_dados(arquivos):
    for arquivo in arquivos:
        try:
            subprocess.run(['python3', arquivo], check=True)
        except subprocess.CalledProcessError as e:
            print(f"Erro ao executar {arquivo}: {e}")


max_threads = 1

for _ in range(100):
    with concurrent.futures.ThreadPoolExecutor(max_threads) as executor:
        resultados = list(executor.map(executar_arquivo, arquivos_teste))