Hoping for input on the clarity of the docstrings, syntax and performance of these two functions both of which should be self-explanatory:
def tell(paragraph, duration=.1):
"""Print paragraph line by line with pause between each line."""
lines = paragraph.splitlines()
for line in lines:
print line.lstrip()
time.sleep(duration)
def yinyang(yes, no, prompt = "y/n: ", tries = 3):
"""Return result based on parsed user input with specified number of tries."""
input = raw_input(prompt)
word_list = scan(input)
repeat_prompt = """Does not compute.
Please try again to express yes or no."""
player_decision = parse_sentence(word_list).decision
try:
assert(input and tries > 0)
input = None
if player_decision == "affirmative":
return yes
elif player_decision == "negative":
return no
else:
tries += 1
tell(repeat_prompt)
return yinyang(yes, no, prompt, tries)
except AssertionError:
return no
The functions, scan() and parse_sentence() are from a module so their docstring (within the module code) should suffice for any needed explanation, right?
Will include them, too:
"""Accept a word string and return a yes, no or error"""
WORD_TYPES = {'affirmative': ['yes', 'y', 'do', 'will', 'sure', 'will', 'ok', 'can', 'would'],
'negative': ['no', 'n', 'won\'t', 'not', 'nyet', 'can\'t']}
type_words = {word : label for label, words in WORD_TYPES.items() for word in words}
PUNCTUATION = ['.', '?', '!', ',', ':', ';']
def scan(sentence):
"""Strip punctuation and return list containing word and type"""
sentence_list = sentence.split()
for idx, word in enumerate(sentence_list):
for p in PUNCTUATION:
if p in word:
word = word.strip(p)
try:
lc_word = word.lower()
word_type = type_words.get(lc_word, None)
assert word_type is not None
sentence_list[idx] = (word_type, lc_word)
except AssertionError:
lc_word = word.lower()
sentence_list[idx] = ('error', lc_word)
return sentence_list
and parse_sentence is here:
class Sentence(object):
def __init__(self, decision):
# remember we take ('noun','princess') tuples and convert them
self.decision = decision[0]
def peek(word_list):
if word_list:
word = word_list[0]
print word[0]
print word[1]
return word[0]
else:
return None
def match(word_list, expecting):
if word_list:
word = word_list.pop(0)
if word[0] == expecting:
return word
else:
return None
else:
return None
def skip(word_list, word_type):
while peek(word_list) == word_type:
match(word_list, word_type)
def parse_decision(word_list):
next_word = peek(word_list)
if next_word == 'affirmative':
return match(word_list, 'affirmative')
elif next_word == 'negative':
return match(word_list, 'negative')
else:
return ('error', 0)
def parse_sentence(word_list):
decision = parse_decision(word_list)
return Sentence(decision)
