#!/usr/bin/env python3

"""Make rhyming words"""

import argparse

import re

import string

# --------------------------------------------------

def get_args():

"""get command-line arguments"""

parser = argparse.ArgumentParser(

description='Make rhyming "words"',

formatter_class=argparse.ArgumentDefaultsHelpFormatter)

parser.add_argument('word', metavar='str', help='A word to rhyme')

return parser.parse_args()

# --------------------------------------------------

def main():

"""Make a jazz noise here"""

args = get_args()

prefixes = list('bcdfghjklmnpqrstvwxyz') + (

'bl br ch cl cr dr fl fr gl gr pl pr sc '

'sh sk sl sm sn sp st sw th tr tw thw wh wr '

'sch scr shr sph spl spr squ str thr').split()

start, rest = stemmer(args.word)

if rest:

print('\n'.join(sorted([p + rest for p in prefixes if p != start])))

else:

print(f'Cannot rhyme "{args.word}"')

# --------------------------------------------------

def stemmer(word):

"""Return leading consonants (if any), and 'stem' of word"""

word = word.lower()

pos = list(

filter(lambda v: v >= 0,

map(lambda c: word.index(c) if c in word else -1, 'aeiou')))

if pos:

first = min(pos)

return (word[:first], word[first:])

else:

return (word, '')

# --------------------------------------------------

def test_stemmer():

"""test the stemmer"""

assert stemmer('') == ('', '')

assert stemmer('cake') == ('c', 'ake')

assert stemmer('chair') == ('ch', 'air')

assert stemmer('APPLE') == ('', 'apple')

assert stemmer('RDNZL') == ('rdnzl', '')

# --------------------------------------------------

if __name__ == '__main__':

main()