import json
import os
import warnings

from gedi.utils.io_helpers import sort_files
from tqdm import tqdm
from utils.param_keys import INPUT_NAME, FILENAME, FOLDER_PATH, PARAMS

def get_model_params_list(alg_json_file: str) :#-> list[dict]:
    """
    Loads the list of model configurations given from a json file or the default list of dictionary from the code.
    @param alg_json_file: str
        Path to the json data with the running configuration
    @return: list[dict]
        list of model configurations
    """
    if alg_json_file is not None:
        return json.load(open(alg_json_file))
    else:
        warnings.warn('The default model parameter list is used instead of a .json-file.\n'
                      '  Use a configuration from the `config_files`-folder together with the args `-a`.')
        return [
            {ALGORITHM_NAME: 'pca', NDIM: TENSOR_NDIM},
            ]
def get_run_params(alg_params_json: str) -> dict:
    """
    Loads the running configuration given from a json file or the default dictionary from the code.
    @param alg_params_json: str
        Path to the json data with the running configuration
    @return: dict
        Running Configuration
    """
    if alg_params_json is not None:
        return json.load(open(alg_params_json))
    else:
        warnings.warn('The default run option is used instead of a .json-file.\n'
                      '  Use a configuration from the `config_files`-folder together with the args `-o`.')
        return {
            RUN_OPTION: COMPARE,
            PLOT_TYPE: COLOR_MAP,  # 'heat_map', 'color_map', '3d_map', 'explained_var_plot'
            PLOT_TICS: True,
            N_COMPONENTS: 2,
            INPUT_NAME: 'runningExample',
            SAVE_RESULTS: True,
            LOAD_RESULTS: True
        }

def get_files_and_kwargs(params: dict):
    """
    This method returns the filename list of the trajectory and generates the kwargs for the DataTrajectory.
    The method is individually created for the available data set.
    Add new trajectory options, if different data set are used.
    @param params: dict
        running configuration
    @return: tuple
        list of filenames of the trajectories AND
        kwargs with the important arguments for the classes
    """
    try:
        input_name = params[INPUT_NAME]
    except KeyError as e:
        raise KeyError(f'Run option parameter is missing the key: `{e}`. This parameter is mandatory.')

    #TODO: generate parent directories if they don't exist
    if input_name == 'test':
        filename_list = list(tqdm(sort_files(os.listdir('data/test'))))
        kwargs = {FILENAME: filename_list, FOLDER_PATH: 'data/test'}
    elif input_name == 'realLogs':
        filename_list = list(tqdm(sort_files(os.listdir('data/real_event_logs'))))
        kwargs = {FILENAME: filename_list, FOLDER_PATH: 'data/real_event_logs'}
    elif input_name == 'gen5':
        filename_list = list(tqdm(sort_files(os.listdir('data/event_log'))))[:5]
        kwargs = {FILENAME: filename_list, FOLDER_PATH: 'data/event_log'}
    elif input_name == 'gen20':
        filename_list = list(tqdm(sort_files(os.listdir('data/event_log'))))[:20]
        kwargs = {FILENAME: filename_list, FOLDER_PATH: 'data/event_log'}
    elif input_name == 'runningExample':
        filename_list = ['running-example.xes']
        kwargs = {FILENAME: filename_list[0], FOLDER_PATH: 'data/'}
    elif input_name == 'metaFeatures':
        filename_list = ['log_features.csv']
        kwargs = {FILENAME: filename_list[0], FOLDER_PATH: 'results/'}
    else:
        raise ValueError(f'No data trajectory was found with the name `{input_name}`.')

    #filename_list.pop(file_element)
    kwargs[PARAMS] = params
    return filename_list, kwargs