task_3_parallelization.sage

import os
import sys
import argparse
from subprocess import Popen

import constants

sys.path.insert(0, "/home/sage/tii-claasp")

parser = argparse.ArgumentParser(description='parallelization of task 1')
parser.add_argument('--models', '-m', action="extend", nargs='*',
                    default=['sat', 'smt', 'milp', 'cp'])
args = parser.parse_args()

if not os.path.exists('scripts/LOGS/task_3'):
    os.makedirs('scripts/LOGS/task_3')

commands = []
for cipher in ('speck_block_cipher.py', 'midori_block_cipher.py'):
    for model in args.models:
        for solver in constants.MODEL_LIST[model]['solver_list']:
            cmd = ['sage', 'scripts/task_3.sage', '-m', model, '-s', solver,
                   '-c', cipher, '-r', str(constants.fixed_differential[cipher][4])]
            commands.append(cmd)

processes = [
    Popen(
        i,
        stdout=open(f'scripts/LOGS/task_3/{i[7].split("_")[0]}.txt', 'a'),
        stderr=open(f'scripts/LOGS/task_3/{i[7].split("_")[0]}_ERR.txt', 'a')) for i in commands]

for process in processes:
    process.communicate()