pet_acts.py

from player import Player
from speech_recog import *
from text_to_speech import *
from story_node import StoryNode
import random
import animal
from nltk.stem.snowball import SnowballStemmer
#acts must pass player as a parameter

stemmer = SnowballStemmer('english')

flag = True
pet_name = ""
pet = None
pet_types = ['dog', 'cat', 'fish', 'bird', 'lizard']
pet_syns = [['dog', 'doggy', 'puppy', 'pooch', 'pup', 'canine'], ['cat', 'kitten', 'kitty', 'feline'], ['fish', 'fishes', 'fishies'], ['bird', 'fowl', 'birdie', 'chick', 'fledgling', 'nestling'], ['lizard']]
yes = ['yes']
yes_syns = [['yes', 'yup', 'yeah', 'yea', 'indeed', 'sure']]


def have_pets_act(player):
    current = player.location
    speak("Hi " + player.name)
    speak (current.description)
    s = getInputString()
    if player.get_target(s, yes, yes_syns) != 'yes': 
        return "done"
    return get_next(player, current, s)

def kinds_act(player):
    global pet
    current = player.location
    speak(current.description)
    s = getInputString()
    pet = animal.Animal(player.get_target(s, pet_types, pet_syns))
    return get_next(player, current, s)

def name_act(player):
    global pet_name
    current = player.location
    speak("What is your " + pet.type + "'s name?")
    s = getInputString()
    pet_name = s
    return get_next(player, current, s)

def breed_act(player):
    current = player.location
    speak("What breed is " + pet_name + "?")
    s = getInputString()
    return get_next(player, current, s)

def age_act(player):
    current = player.location
    speak("How old is your " + pet.type + "?")
    s = getInputString()
    return get_next(player, current, s)

def toy_act(player):
    current = player.location
    speak("What is your " + pet.type + "'s favorite toy or game?")
    s = getInputString()
    return get_next(player, current, s)

def color_act(player):
    current = player.location
    speak("What color " + pet.bc_plural+ " your " + pet.type + "'s " + pet.body_covering + "?")
    s = getInputString()
    return get_next(player, current, s)

def size_act(player):
    current = player.location
    speak("Is " + pet_name + " big or small?")
    s = getInputString()
    return get_next(player, current, s)

def sound_act(player):
    current = player.location
    speak("What sound does " + pet_name + " make?")
    s = getInputString()
    return get_next(player, current, s)

def mess_act(player):
    current = player.location
    speak("Is your " + pet.type + " messy?")
    s = getInputString()
    return get_next(player, current, s)

def sleep_act(player):
    current = player.location
    speak("Where does " + pet_name + " usually sleep?")
    s = getInputString()
    return get_next(player, current, s)

def food_act(player):
    current = player.location
    speak("Who in your family feeds your " + pet.type + " its " + pet.eats +"?")
    s = getInputString()
    return get_next(player, current, s)

def treat_act(player):
    current = player.location
    speak("What is " + pet_name + "'s favorite treat?")
    s = getInputString()
    return get_next(player, current, s)

def sibs_act(player):
    current = player.location
    speak("Does " + pet_name + " have any siblings?")
    s = getInputString()
    return get_next(player, current, s)

def friends_act(player):
    current = player.location
    speak("Does " + pet_name + " have animal friends?")
    s = getInputString()
    return get_next(player, current, s)
	
def temperment_act(player):
    current = player.location
    speak("Does your " + pet.type + " like strangers?")
    s = getInputString()
    return get_next(player, current, s)

def vet_act(player):
    current = player.location
    speak("Does " + pet_name + " like going to the vet?")
    s = getInputString()
    return get_next(player, current, s)

def walk_act(player):
    current = player.location
    speak("Do you take " + pet_name + " on walks?")
    s = getInputString()
    return get_next(player, current, s)

def tricks_act(player):
    current = player.location
    speak("Can " + pet_name + " do any tricks?")
    s = getInputString()
    return get_next(player, current, s)

def tail_act(player):
    current = player.location
    s = ""
    if pet.tail:
        speak("Is " + pet_name + "'s tail long or short?")
        s = getInputString()
    else:
        player.completed[current.name] = False
    return get_next(player, current, s)

def pet_act(player):
    current = player.location
    speak(current.description)
    s = getInputString()
    return get_next(player, current, s)

def check_completed(dct, s):
    global flag
    flag = True
    inLst = False
    for x in dct.keys():
        if dct[x] == False:
            flag = False
        if s.name.lower() == x.lower() and dct[x] == True:
            inLst = True

    return inLst

def done_act(player):
    speak (player.location.description)
    return "quit"

def get_next(player, current, s):
	split = s.lower().split()
	if 'goodbye' in split or 'done' in split:
		return 'done'
	if len(current.children) == 2:
		return current.children[0].name
	if convo_over(player):
		return "done"
	num_childs = len(current.children)
	x = random.randint(0,num_childs-2)
	while check_completed(player.completed, current.children[x]):
		if not flag:
			x = random.randint(0,num_childs-2)
		else:
			return "done"
	return current.children[x].name

def convo_over(player):
    count = 0
    for item in player.completed:
        if player.completed[item] == True:
            count += 1
    if count >= 10:
        return True
    return False