from typing import Dict, Any, Callable, Iterable

DataType = Iterable[Dict[str, Any]]
ModifierFunc = Callable[[DataType], DataType]


def query(data: DataType, selector: ModifierFunc,
          *filters: ModifierFunc) -> DataType:
    """
    Query data with column selection and filters

    :param data: List of dictionaries with columns and values
    :param selector: result of `select` function call
    :param filters: Any number of results of `field_filter` function calls
    :return: Filtered data
    """
    filtered_data = selector(data)
    for filter in filters:
        filtered_data = filter(filtered_data)
    print(filtered_data)
    return filtered_data


def select(*columns: str) -> ModifierFunc:
    """Return function that selects only specific columns from dataset"""
    search_fields = columns

    def selector(data: DataType) -> DataType:
        selected_list = []
        for i in data:
            selected_dict = {key: value for (key, value) in i.items() if key in search_fields}
            selected_list.append(selected_dict)

        data = selected_list
        return data

    return selector


def field_filter(column: str, *values: Any) -> ModifierFunc:
    """Return function that filters specific column to be one of `values`"""
    flt_key = column
    flt_values = values

    def filter(data: DataType) -> DataType:
        filtered_list = []
        for i in data:
            if (flt_key in i.keys() and i[flt_key] in flt_values) or not (flt_key in i.keys()):
                filtered_list.append(i)

        data = filtered_list
        return data

    return filter


def test_query():
    friends = [
        {'name': 'Sam', 'gender': 'male', 'sport': 'Basketball'}
    ]
    value = query(
        friends,
        select(*('name', 'gender', 'sport')),
        field_filter(*('sport', *('Basketball', 'volleyball'))),
        field_filter(*('gender', *('male',))),
    )
    assert [{'gender': 'male', 'name': 'Sam', 'sport': 'Basketball'}] == value


if __name__ == "__main__":
    test_query()