EmotAI.py

from MemoryData import MemoryData
from ReactionStrategies.LaughStrategy import LaughStrategy
from ReactionStrategies.SmileStrategy import SmileStrategy
from ReactionStrategies.AgressiveStrategy import AgressiveStrategy
from ReactionStrategies.CryStrategy import CryStrategy
from ReactionStrategies.DefaultStrategy import DefaultStrategy
from PersonalityParser import  PersonalityParser


class EmotAI:
    NAME_OF_MEMORY_INDEX = 0
    VALUE_OF_MEMORY_INDEX = 1
    MEMORY_SIZE = 6

    def __init__(self, personality):
        self.personality_name = personality
        self.personality = PersonalityParser.parse(personality)
        self.knowledge = self.take_knowledge()
        self.actions_memory = MemoryData(self.personality["memory"], self.MEMORY_SIZE)
        self.actions_memory.onMemoryChange = self.onMemoryChange

    def onMemoryChange(self):
        self.personality["memory"] = self.actions_memory.values
        PersonalityParser.save(self.personality_name,self.personality)


    def take_knowledge(self):
        f = open("memoryOfEmotAI.txt", "r")
        my_dict = {}
        for memoryLine in f:
            splitted_memories = memoryLine.replace("\n", "").split('=')
            my_dict[splitted_memories[self.NAME_OF_MEMORY_INDEX]] = splitted_memories[self.VALUE_OF_MEMORY_INDEX]
        return my_dict

    def say(self, sentence):
        print(sentence)

    def introduce(self):
        self.say("Hello I am " + str(self.knowledge.get("myName")) + " ! I was born on the " + str(
            self.knowledge.get("myBirthDate")))

    # return True to continue
    # return False to stop
    def hear_user(self):
        to_analyze = input("Please send me a word : ")
        if str.lower(to_analyze.strip()) == "stop" or str.lower(to_analyze.strip()) == "bye":
            self.say("Ok, bye !")
            return False
        elif str.lower(to_analyze) == "hit":
            return self.hit()
        elif str.lower(to_analyze) == "joke":
            self.joke()
            return True
        elif str.lower(to_analyze) == "look":
            self.look()
            return True
        else:
            self.say("You said \"" + to_analyze + "\" but I did not understand. If you want to stop, just tell me STOP")
            return True

    def how_do_you_feel(self):
        self.say("For the moment...")

    def hit(self):
        personalities_strategies = [
            LaughStrategy(self),
            SmileStrategy(self),
            AgressiveStrategy(self),
            CryStrategy(self),
            DefaultStrategy(self)
        ]
        for personality in personalities_strategies:
            if personality.is_personality["hit"]:
                return personality.hit()

    def joke(self):
        personalities_strategies = [
            LaughStrategy(self),
            SmileStrategy(self),
            CryStrategy(self),
            AgressiveStrategy(self),
            DefaultStrategy(self)
        ]
        for personality in personalities_strategies:
            if personality.is_personality["joke"]:
                return personality.joke()

    def look(self):
        personalities_strategies = [
            AgressiveStrategy(self),
            DefaultStrategy(self)
        ]
        for personality in personalities_strategies:
            if personality.is_personality["look"]:
                return personality.look()