tiny_python_projects/14_rhymer/solution2_no_regex.py at master · t-rezwana/tiny_python_projects · GitHub

History

executable file

65 lines (46 loc) · 1.73 KB

Raw

#!/usr/bin/env python3

"""Make rhyming words"""

import argparse

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

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()

vowel_pos = list(map(word.index, filter(lambda v: v in word, 'aeiou')))

if vowel_pos:

first_vowel = min(vowel_pos)

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

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', '')

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

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

if __name__ == '__main__':

main()

tiny_python_projects/14_rhymer/solution2_no_regex.py at master · t-rezwana/tiny_python_projects · GitHub

Expand file tree

Breadcrumbs

solution2_no_regex.py

Latest commit

History

Breadcrumbs

solution2_no_regex.py

File metadata and controls