import torch
import numpy as np
import string
import shutil
import os


def kwargify(**kwargs):
    return kwargs


def show_lr(optimizer):
    lr = []
    for param_group in optimizer.param_groups:
        lr += [param_group['lr']]

    return np.array(lr).mean()


def contains_nan_or_inf(tensor):
    return torch.isnan(tensor).any() or torch.isinf(tensor).any()


def map_phonemes(phonemes):
    new_phonemes = []
    charset = string.printable.strip()

    for phoneme in phonemes:
        if phoneme == ' ':
            new_phonemes.append(phoneme)
        else:
            index = phonemes.index(phoneme) - 1
            new_phonemes.append(charset[index])

    return new_phonemes


def empty_dir(directory):
    for filename in os.listdir(directory):
        file_path = os.path.join(directory, filename)

        try:
            if os.path.isfile(file_path):
                os.remove(file_path)
            elif os.path.islink(file_path):
                os.unlink(file_path)
            elif os.path.isdir(file_path):
                shutil.rmtree(file_path)
        except Exception as e:
            print('Failed to delete %s. Reason: %s' % (file_path, e))