def load_words(inputstr):
    words = []
    for line in inputstr.splitlines():
        words.append(line)
    return words


# PART 1


def is_vowel(char):
    return char in "aeiou"


def has_double_letter(word):
    # This would look nicer im haskell:
    return count(zip(word, word[1:]), lambda x: x[0] == x[1]) > 0


def is_nice(word):
    if "ab" in word or "cd" in word or "pq" in word or "xy" in word:
        return False
    if len(list(filter(is_vowel, word))) < 3:
        return False
    return has_double_letter(word)


def count(what, function):
    # This would also look nicer in haskell
    return len(list(filter(function, what)))


# PART 2


def has_pair(word):
    for i in range(len(word)):
        if word[i : i + 2] in word[i + 2 :]:
            return True
    return False


def has_repeat_letter(word):
    return count(zip(word, word[2:]), lambda x: x[0] == x[1]) > 0


def is_nice_2(word):
    return has_pair(word) and has_repeat_letter(word)


def solve(inputstr):
    words = load_words(inputstr)
    amount = count(words, is_nice)
    print(f"Part 1: {amount}")
    amount_2 = count(words, is_nice_2)
    print(f"Part 2: {amount_2}")