from collections import defaultdict
from itertools import product


def pipe(x):
    return x


def print_pipe(x):
    print(x)
    return x


def flattenDict(l):
    def tupalize(k, vs):
        l = []
        if isinstance(vs, list):
            for v in vs:
                l.append((k, v))
        else:
            l.append((k, vs))
        return l

    flat_list = [tupalize(k, vs) for k, vs in l.items()]
    flat_dict = [dict(items) for items in product(*flat_list)]
    return flat_dict


def flatten(l):
    if isinstance(l, list):
        return [item for sublist in l for item in sublist]
    elif isinstance(l, dict):
        return flattenDict(l)


def flatMap(f, collection):
    return flatten(list(map(f, collection)))


def dict_filter(dictionary, condition):
    return dict([(k, v) for k, v in dictionary.items() if condition(v)])


def contains_type(_collection, type):
    return any(isinstance(x, type) for x in _collection)


def drop_right(l, n):
    return l[:len(l) - n]


def key_filter(l, keyname):
    return [v[keyname] for k, v in l.items()]


def groupByKey(l):
    d = defaultdict(list)
    for key, value in l:
        d[key].append(value)
    return list(dict(d).items()).pop()


# @curried
def rename(new_name, f):
    f.__name__ = new_name
    return f


def curry_pot(f, *argv):
    sweep_ind = f.__name__[0:5] == 'sweep'
    arg_len = len(argv)
    if sweep_ind is True and arg_len == 4:
        return f(argv[0])(argv[1])(argv[2])(argv[3])
    elif sweep_ind is False and arg_len == 4:
        return f(argv[0], argv[1], argv[2], argv[3])
    elif sweep_ind is True and arg_len == 3:
        return f(argv[0])(argv[1])(argv[2])
    elif sweep_ind is False and arg_len == 3:
        return f(argv[0], argv[1], argv[2])
    else:
        raise TypeError('curry_pot() needs 3 or 4 positional arguments')

# def curry_pot(f, *argv):
#     sweep_ind = f.__name__[0:5] == 'sweep'
#     arg_len = len(argv)
#     if sweep_ind is True and arg_len == 4:
#         return f(argv[0])(argv[1])(argv[2])(argv[3])
#     elif sweep_ind is False and arg_len == 4:
#         return f(argv[0])(argv[1])(argv[2])(argv[3])
#     elif sweep_ind is True and arg_len == 3:
#         return f(argv[0])(argv[1])(argv[2])
#     elif sweep_ind is False and arg_len == 3:
#         return f(argv[0])(argv[1])(argv[2])
#     else:
#         raise TypeError('curry_pot() needs 3 or 4 positional arguments')

# def rename(newname):
#     def decorator(f):
#         f.__name__ = newname
#         return f
#     return decorator