from simdeep.config import PATH_TO_SAVE_MODEL

from os.path import isfile

from os.path import isdir

from os import mkdir

# from sys import version_info

# if version_info > (3, 0, 0):

#     import pickle as cPickle

# else:

#     import cPickle

import dill

from time import time

def save_model(boosting, path_to_save_model=PATH_TO_SAVE_MODEL):

    """ """

    if not isdir(path_to_save_model):

        mkdir(path_to_save_model)

    boosting._convert_logs()

    t = time()

    with open('{0}/{1}.pickle'.format(

            path_to_save_model,

            boosting._project_name), 'wb') as f_pick:

        dill.dump(boosting, f_pick)

    print('model saved in %2.1f s at %s/%s.pickle' % (

        time() - t, path_to_save_model, boosting._project_name))

def load_model(project_name, path_model=PATH_TO_SAVE_MODEL):

    """ """

    t = time()

    project_name = project_name.replace('.pickle', '') + '.pickle'

    assert(isfile('{0}/{1}'.format(path_model, project_name)))

    with open('{0}/{1}'.format(path_model, project_name), 'rb') as f_pick:

        boosting = dill.load(f_pick)

    print('model loaded in %2.1f s' % (time() - t))

    return boosting

def metadata_usage_type(value):

    """ """

    if value not in {None,

                     False,

                     'labels',

                     'new-features',

                     'test-labels',

                     'all', True}:

        raise Exception(

            "metadata_usage_type: {0} should be from the following choices:" \

            " [None, False, 'labels', 'new-features', 'all', True]" \

            .format(value))

    if value == True:

        return 'all'

    return value

def feature_selection_usage_type(value):

    """ """

    if value not in {'individual',

                     'lasso',

                     None}:

        raise Exception(

            "feature_selection_usage_type: {0} should be from the following choices:" \

            " ['individual', 'lasso', None]" \

            .format(value))

    return value

def load_labels_file(path_labels, sep="\t"):

    """

    """

    labels_dict = {}

    for line in open(path_labels):

        split = line.strip().split(sep)

        if len(split) < 2:

            raise Exception(

                '## Errorfor file in load_labels_file: {0} for line{1}' \

                ' line cannot be splitted in more than 2'.format(

                    line, path_labels))

        patient, label = split[0], split[1]

        try:

            label = int(float(label))

        except Exception:

            raise Exception(

                '## Error: in load_labels_file {0} for line {1}' \

                'labels should be an int'.format(

                    path_labels, line))

        if len(split) > 2:

            try:

                proba = float(split[2])

            except Exception:

                raise Exception(

                    '## Error: in load_labels_file {0} for line {1}' \

                    'label proba in column 3 should be a float'.format(

                        path_labels, line))

            else:

                proba = label

        labels_dict[patient] = (label, proba)

    return labels_dict