def save_plots_to_os()

in src/save_plots.py [0:0]

• 9 lines of code
• 5 McCabe index (conditional complexity)

def save_plots_to_os(figures, figure_names, dirname, relaxed=False):
    base_dir = os.path.dirname(__file__)[:-4]  # we use -4 to take off the src/ from the end to go back a directory
    results_dir = os.path.join(base_dir, f'{dirname}' + ('/Relaxation_Curves/' if relaxed else '/Plots/'))

    # Create the directory, if needed
    if not os.path.isdir(results_dir):
        print(f'making directory: {results_dir}')
        os.makedirs(results_dir)

    # Save all figures
    for fig, name in zip(figures, figure_names):
        fig.savefig(results_dir + name)

    print(f'Successfully saved plots to {dirname}')