main.py

import timeit

from bruteforce import bruteforce
from file import get_info_from_file
from optimized_dynamic import optimized_dynamic
from optimized_naive import optimized_naive


def menu():
    print('Menu Principal:')
    print('1. Force brute')
    print('2. Optimisé Naïve')
    print('3. Optimisé Dynamique')
    print('4. Quitter')


while True:
    menu()
    print()
    try:
        choice = int(input('Sélectionner une option: '))
        if choice == 1:
            choice = input('Sélectionner un fichier CSV: ')
            print()
            start = timeit.default_timer()
            result = bruteforce(500, get_info_from_file(choice))
            end_time = round(timeit.default_timer() - start, 2)
            print(f"Voici la liste d'action les plus rentable pour un budget maximum de 500€")
            print()
            for action in result[1]:
                print(action)
            print()
            print(f"Notre bénéfice total s'élève à {result[0]}€")
            print()
            print(f"Temps d'exécution: {end_time} secondes")
            print()
        if choice == 2:
            choice = input('Sélectionner un fichier CSV: ')
            print()
            start = timeit.default_timer()
            result = optimized_naive(500, get_info_from_file(choice))
            end_time = round(timeit.default_timer() - start, 4)
            print(f"Voici la liste d'action les plus rentable pour un budget maximum de 500€")
            print()
            for action in result:
                print(action)
            benefice = sum(map(lambda action_list: action_list.benefice, result))
            print()
            print(f"Notre bénéfice total s'élève à {benefice:>.2f}€")
            print()
            print(f"Temps d'exécution: {end_time} secondes")
            print()
        if choice == 3:
            choice = input('Sélectionner un fichier CSV: ')
            print()
            start = timeit.default_timer()
            result = optimized_dynamic(500, get_info_from_file(choice))
            end_time = round(timeit.default_timer() - start, 2)
            print(f"Voici la liste d'action les plus rentable pour un budget maximum de 500€")
            print()
            for action in result[1]:
                print(action)
            print()
            print(f"Notre bénéfice total s'élève à {result[0]:>.2f}€")
            print()
            print(f"Temps d'exécution: {end_time} secondes")
            print()
        if choice == 4:
            quit()
    except ValueError:
        print("Sélectionner un numéro valide")