def get_name(self):
"""
Sends messages to user to get their name
:return: client's name
"""
count = 0
max_limit = 3
if DEBUG_ENABLED:
print(
"1.1.1 - Write introductory message to user. Enter while count < max_limit"
)
self.in_out.write_message(self._random_choice(INTRODUCTION_MESSAGE))
while count < max_limit:
count += 1
if DEBUG_ENABLED:
print("1.1.2 - Get users message as messageType object")
message = MessageTypeAnalysis(self.in_out.read_message())
if message.is_miscellaneous() == Miscellaneous.GREETING:
self.in_out.write_message(
self._random_choice(GREETING_RESPONSE).format(
message.name))
if DEBUG_ENABLED:
print(
"1.1.3 - If user's message is a name, greet user referencing their name"
)
if message.is_name():
self.in_out.write_message(
self._random_choice(FIRST_GREETING).format(message.name))
if DEBUG_ENABLED:
print("1.1.4 - RETURN name")
return message.is_name()
if DEBUG_ENABLED:
print(
"1.1.4 - Otherwise, repeat requests until count is at max limit. count= {}. Go back to - 1.1.2"
.format(count))
if count == 1:
self.in_out.write_message(self._random_choice(REQUEST_NAME))
elif count == 2:
self.in_out.write_message(
self._random_choice(REPEAT_REQUEST_NAME))
else:
self.in_out.write_message(
self._random_choice(URGENT_REQUEST_NAME))
if DEBUG_ENABLED:
print(
"1.1.5 - No name given and max limit reached. Tell user -> RETURN None"
)
self.in_out.write_message(self._random_choice(NO_NAME_GIVEN))
return None
def test_smoking():
"""
> pytest -s
:return:
"""
#DEBUG_ENABLED = True
InputOutput.CACHE_MODE = True
in_out = InputOutput(interface_type=InputOutput.INTERFACE_TYPE_SIMPLE)
InputOutput.INPUT_CACHE_DATA_USER = "******"
therapist = Therapist(in_out=in_out)
assert therapist.interview.ask_for_name()
print("name: {}".format(therapist.client.get_name()))
assert therapist.client.get_name() == "Alex"
InputOutput.INPUT_CACHE_DATA_USER = "******"
#InputOutput.INPUT_CACHE_DATA_USER = "******"
message_for_dream = MessageTypeAnalysis(
InputOutput.INPUT_CACHE_DATA_USER).is_dream(Request.SUBJECT)
a_dream = Dream(message_for_dream)
AnalyseDream(a_dream)
assert a_dream.get_subject() == Subject.SOMEONE
assert a_dream.get_direct_object() == "Car"
assert a_dream.get_ratings() == [Rating.NEGATIVE]
def test_dream_message():
#An Input which contains a dream
for message in VALID_DREAMS:
# message_nlp as processed by messageTypeAnalysis
message_nlp = MessageTypeAnalysis(message).is_dream()
#Create a dream
a_dream = Dream(message_nlp)
#Test the contents of the dream match the message provided by user
assert a_dream.get_latest_message() == message_nlp
def test_dream_subject_client():
#An Input which contains a dream
message = "I am the subject of the dream"
# message_nlp as processed by messageTypeAnalysis
message_nlp = MessageTypeAnalysis(message).is_dream()
#Create a dream
a_dream = Dream(message_nlp)
analyse_dream = AnalyseDream(a_dream)
assert a_dream.get_subject() == Subject.CLIENT
def test_dream_subject_other():
#An Input which contains a dream
message = "The dream is about my car."
# message_nlp as processed by messageTypeAnalysis
message_nlp = MessageTypeAnalysis(message).is_dream()
#Create a dream
a_dream = Dream(message_nlp)
analyse_dream = AnalyseDream(a_dream)
assert a_dream.get_subject() == Subject.OTHER
def test_dream_subject_someone():
#An Input which contains a dream
message = "The subject of the dream is a friend"
# message_nlp as processed by messageTypeAnalysis
message_nlp = MessageTypeAnalysis(message).is_dream()
#Create a dream
a_dream = Dream(message_nlp)
analyse_dream = AnalyseDream(a_dream)
assert a_dream.get_subject() == Subject.SOMEONE
def get_post_analysis_option(self):
"""
Get options from user
:return: user's option
"""
if DEBUG_ENABLED:
print(
"3.1.1 - Send message to user requesting their option: new dream or get analysis"
)
self.in_out.write_message(
self._random_choice(REQUEST_POST_ANALYSIS_OPTION))
done = False
while not done:
if DEBUG_ENABLED:
print(
"3.1.2 - Read users response and check if is_option() -> MessageTypeAnalysis"
)
pos_opt = MessageTypeAnalysis(
self.in_out.read_message()).is_option()
if pos_opt == PostOption.NEW_DREAM:
if DEBUG_ENABLED:
print(
"3.1.3 - User has requested to give another dream - RETURN NEW_DREAM"
)
done = True
return PostOption.NEW_DREAM
elif pos_opt == PostOption.GET_ANALYSIS:
if DEBUG_ENABLED:
print(
"3.1.3 - User has requested to get_analysis - RETURN GET_ANALYSIS "
)
done = True
return PostOption.GET_ANALYSIS
else:
if DEBUG_ENABLED:
print(
"3.1.3 - No potion has been provide. Go back to - 3.1.3"
)
self.in_out.write_message(self._random_choice(NO_OPTION))
if DEBUG_ENABLED:
print("TODO: Do something with the - RETURN")
return
def test_dream_rating():
#An Input which contains a Dream.
message = VALID_DREAMS[3]
# message_nlp as processed by MessageTypeAnalysis.
message_nlp = MessageTypeAnalysis(message).is_dream()
#Create a Dream.
a_dream = Dream(message_nlp)
#Assign features of Dream.
analyse_dream = AnalyseDream(a_dream)
assert a_dream.get_ratings() == [Rating.POSITIVE, Rating.POSITIVE]
def test_dream_topic():
#An Input which contains a Dream.
message = VALID_DREAMS[2]
# message_nlp as processed by MessageTypeAnalysis.
message_nlp = MessageTypeAnalysis(message).is_dream()
#Create a Dream.
a_dream = Dream(message_nlp)
#Assign features of Dream.
analyse_dream = AnalyseDream(a_dream)
assert a_dream.get_topics() == [Topic.SEX_SYMBOL, Topic.PSYCHE]
def test_dream_dir_obj():
#An Input which contains a Dream.
message = VALID_DREAMS[2]
# message_nlp as processed by MessageTypeAnalysis.
message_nlp = MessageTypeAnalysis(message).is_dream()
#Create a Dream.
a_dream = Dream(message_nlp)
#Assign features of Dream.
analyse_dream = AnalyseDream(a_dream)
assert a_dream.get_direct_object() == "Mountain"
def test_dream_valid():
#message = VALID_DREAMS[3]
#message = VALID_DREAMS[4]
message = VALID_DREAMS[5]
# message_nlp as processed by MessageTypeAnalysis.
message_nlp = MessageTypeAnalysis(message).is_dream()
#Create a Dream.
a_dream = Dream(message_nlp)
#Assign features of Dream.
analyse_dream = AnalyseDream(a_dream)
dream_validity = DreamValidator(a_dream)
print(str(dream_validity.outstanding_features()))
assert dream_validity.is_valid() == False
def get_dream(self, post_opt: PostOption = None):
"""
Sends messages to user to get details on their dream.
Details invole the subject, topic and rating of the dream.
:type post_opt: PostOption
:param post_opt: whether they are providing their first dream, a further dream, or more information on a dream
:return: client's dream
"""
count = 0
max_limit = 3
validate_dream = None
if DEBUG_ENABLED:
print("2.1.1 - Write a request for a dream")
self.in_out.write_message(self._random_choice(REQUEST_DREAM))
if DEBUG_ENABLED:
print("2.1.2 - While count doesn't breach max_limit:")
while count < max_limit:
count += 1
print("count: {}".format(count))
if DEBUG_ENABLED:
print("2.1.3 - Get message from user and check if dream:")
message = MessageTypeAnalysis(self.in_out.read_message())
#TODO: This compensates for min_limit error. Should be fixed and method redundant
if message in NO_MORE_INFO:
if DEBUG_ENABLED:
print(
"2.1.4 - Message = {}. don't analyse message, but increase count"
)
count += 1
message_nlp = message.is_dream()
if DEBUG_ENABLED:
print(
"2.1.4 - If message is_dream: Check if dream already has a dream. If so, add message to dream messages. If not, create dream with users message"
)
if (message_nlp):
if (self.dream is not None):
if DEBUG_ENABLED:
print(
"2.1.5 - Dream exists Add message to existing dream. count should be 2+ count = {}"
.format(count))
self.dream.add_message(message_nlp)
else:
if DEBUG_ENABLED:
print(
"2.1.5 - No dream found. Create new one -> Dream(user msg)"
)
self.dream = Dream(message_nlp)
AnalyseDream(self.dream)
validate_dream = DreamValidator(self.dream)
missing_features = validate_dream.outstanding_features()
if DEBUG_ENABLED:
print(
"2.1.6 - Analyse and Validate dream. Create list of missing features. missing_features = {}"
.format(missing_features))
#TODO: This should really be a sequence of if/else because each time it gets a feature it should go back to the start incase topic and rating given straight after subject request.
# But recalling missing feautres seems to work so I'll leave it for now.
if DEBUG_ENABLED:
print("2.1.7 - Check if Subject is missing")
if Request.SUBJECT in missing_features:
self.get_subject()
# if DEBUG_ENABLED:
# print("2.1.8 - If count is below min_limit: request more info from user. If message provides more info (doesn't contain: no and is_dream), add message to dream")
# if count < min_limit:
self.in_out.write_message(
self._random_choice(MORE_INFO_SUBJECT))
#TODO: This may be messing up if user doesn't provie more info and says 'no' instead.
more_info_response = self.in_out.read_message()
if (MessageTypeAnalysis(more_info_response).is_dream(
Request.SUBJECT)) and (more_info_response
not in NO_MORE_INFO):
if DEBUG_ENABLED:
print(
"2.1.9 - more info response from client does provide more info, and message has been added to dream"
)
more_info_about_subject = MessageTypeAnalysis(
more_info_response).is_dream(Request.SUBJECT)
if more_info_about_subject is not None:
self.dream.add_message(more_info_about_subject)
AnalyseDream(self.dream)
else:
pdb.set_trace()
if DEBUG_ENABLED:
print("2.1.8 - Check if Topic is missing.")
if Request.TOPIC in missing_features:
if DEBUG_ENABLED:
print("2.1.9 - Topic is missing - get_topic()")
if self.get_topic() is None:
return None
# if DEBUG_ENABLED:
# print("2.1.11 - If count is below min_limit: request more info from user. If message provides more info (doesn't contian: no and is_dream), add message to dream")
# if count < min_limit:
self.in_out.write_message(
self._random_choice(MORE_INFO_TOPIC))
#TODO: This may be messing up if user doesn't provie more info and says 'no' instead.
more_info_response = self.in_out.read_message()
if (MessageTypeAnalysis(more_info_response).is_dream(
Request.TOPIC)) and (more_info_response
not in NO_MORE_INFO):
if DEBUG_ENABLED:
print(
"2.1.12 - More info response from client does provide more info, and message has been added to dream and analysed"
)
self.dream.add_message(
MessageTypeAnalysis(more_info_response).is_dream(
Request.TOPIC))
AnalyseDream(self.dream)
if DEBUG_ENABLED:
print(
"2.1.12 - Analyse and Re-validate dream with new messages. Get updated list of missing features"
)
print(
"TODO: Lets figure out what all our differen't instances of Dream contain."
)
print(" -self.dream = {}".format(self.dream))
#print(" -AnalyseDream(self.dream) = {}".format(AnalyseDream(self.dream)))
print(
" -validate_dream = {}".format(validate_dream))
#AnalyseDream(self.dream)
missing_features = validate_dream.outstanding_features()
if DEBUG_ENABLED:
print("2.1.9 - Check if Rating is missing.")
if Request.RATING in missing_features:
if DEBUG_ENABLED:
print("2.1.13 - Rating is missing - get_rating()")
if self.get_rating() is None:
return None
# if DEBUG_ENABLED:
# print("2.1.14 - If count is below min_limit: request more info from user. If message provides more info (doesn't contian: no or essential tokens), add message to dream")
# if count < min_limit:
self.in_out.write_message(
self._random_choice(MORE_INFO_RATNG))
#TODO: This may be messing up if user doesn't provie more info and says 'no' instead.
more_info_response = self.in_out.read_message()
if (MessageTypeAnalysis(more_info_response).is_dream(
Request.RATING)) and (more_info_response.lower()
not in NO_MORE_INFO):
if DEBUG_ENABLED:
print(
"2.1.15 - more info response from client does provide more info, and message has been added to dream"
)
self.dream.add_message(
MessageTypeAnalysis(more_info_response).is_dream(
Request.RATING))
AnalyseDream(self.dream)
if DEBUG_ENABLED:
print(
"TODO: Should Re-Analyse latest message and validate dream. Shouldn't be looking at previous message features."
)
validate_dream = DreamValidator(self.dream)
if validate_dream is not None:
if DEBUG_ENABLED:
print(
"2.1.16 - Validate dream. checking to see if updated missing features = {} and count should be greater than {}:"
.format(validate_dream.outstanding_features(), count))
if (validate_dream.is_valid()):
self.in_out.write_message(self._random_choice(FINAL_INFO))
self.in_out.read_message()
if DEBUG_ENABLED:
print(
"2.1.16 - A complete dream has been found (no missing features) and count exceeds min_limit. Dream: {} . count: {} - RETURN complete dream"
.format(self.dream, count))
return self.dream
if count == 1:
self.in_out.write_message(
self._random_choice(REPEAT_REQUEST_DREAM))
elif count == 2:
self.in_out.write_message(
self._random_choice(REPEAT_REQUEST_DREAM))
else:
self.in_out.write_message(
self._random_choice(URGENT_REQUEST_DREAM))
if DEBUG_ENABLED:
print("2.1.18 - max limit exceeded. No dream found - RETURN None")
self.in_out.write_message(self._random_choice(NO_DREAM_GIVEN))
return None
def get_rating(self):
"""
Get sentiment of user's dream
:return Rating: sentiment of dream
"""
count = 0
max_limit = 3
if DEBUG_ENABLED:
print("2.2.13.1 - Request rating from user")
self.in_out.write_message(
self._random_choice(REQUEST_RATING).format(
self.dream.get_direct_object()))
if DEBUG_ENABLED:
print("2.2.13.2 - While count doesn't breach max_limit:")
while count < max_limit:
count += 1
message = MessageTypeAnalysis(self.in_out.read_message())
if DEBUG_ENABLED:
print(
"2.2.13.3 - Read users response. If its a valid dream, add message to dream. message = {}. Analyse and Validate"
.format(message.message))
message_nlp = message.is_dream(Request.RATING)
#if message_nlp is not None:
if message_nlp:
self.dream.add_message(message_nlp)
AnalyseDream(self.dream)
validate_dream = DreamValidator(self.dream)
if DEBUG_ENABLED:
print(
"2.2.13.4 - If Dream no longer has a Rating missing. Return the dreams Rating - RETURN dreams Rating"
)
if Request.RATING not in validate_dream.outstanding_features():
self.in_out.write_message(
self._random_choice(RATING_RESPONSE))
return self.dream.get_ratings()
if DEBUG_ENABLED:
print(
"2.2.10.5 - Dream still doesn't have a Rating: {} - repeat request for Rating of dream depending on count. count = {} go to 2.2.3"
.format(self.dream.get_ratings(), count))
if count == 1:
self.in_out.write_message(
self._random_choice(REPEAT_REQUEST_RATING))
elif count == 2:
self.in_out.write_message(
self._random_choice(REPEAT_REQUEST_RATING))
else:
self.in_out.write_message(
self._random_choice(URGENT_REQUEST_RATING))
if DEBUG_ENABLED:
print(
"2.2.13.6 - No Rating has been given. Rating = {} - RETURN None"
.format(self.dream.get_ratings()))
self.in_out.write_message(self._random_choice(NO_RATING_GIVEN))
return None
def get_topic(self):
"""
Get symbol of user's dream
:return Topic: symbolism of dream
"""
count = 0
max_limit = 3
if DEBUG_ENABLED:
print("2.2.10.1 - Request topic from user")
self.in_out.write_message(self._random_choice(REQUEST_TOPIC))
if DEBUG_ENABLED:
print("2.2.10.2 - While count doesn't breach max_limit:")
while count < max_limit:
count += 1
if DEBUG_ENABLED:
print(
"2.2.10.3 - Read users response. If its a valid dream, add message to dream. Analyse and Validate"
)
message = MessageTypeAnalysis(self.in_out.read_message())
message_nlp = message.is_dream(Request.TOPIC)
if message_nlp is not None:
self.dream.add_message(message_nlp)
AnalyseDream(self.dream)
validate_dream = DreamValidator(self.dream)
if DEBUG_ENABLED:
print(
"2.2.10.4 - If Dream no longer has a Topic missing. Return the dreams topic - RETURN dreams topic's"
)
if Request.TOPIC not in validate_dream.outstanding_features():
self.in_out.write_message(
self._random_choice(TOPIC_RESPONSE))
return self.dream.get_topics()
if DEBUG_ENABLED:
print(
"2.2.10.5 - Dream still doesn't have a topic: {} - repeat request for Subject of dream depending on count. count = {} go to 2.2.3"
.format(self.dream.get_topics(), count))
if count == 1:
self.in_out.write_message(
self._random_choice(REPEAT_REQUEST_TOPIC))
elif count == 2:
self.in_out.write_message(
self._random_choice(REPEAT_REQUEST_TOPIC))
else:
self.in_out.write_message(
self._random_choice(URGENT_REQUEST_TOPIC))
if DEBUG_ENABLED:
print(
"2.2.10.6 - No Topic has been given. topics = {} - RETURN None"
.format(self.dream.get_topics()))
self.in_out.write_message(self._random_choice(NO_TOPIC_GIVEN))
return None
def get_subject(self):
"""
Get subject of user's dream
:return: subject of dream
"""
count = 0
#min_limit = 2
max_limit = 3
if DEBUG_ENABLED:
print("2.2.8.1 - Send message to user requesting Subject")
self.in_out.write_message(self._random_choice(REQUEST_SUBJECT))
if DEBUG_ENABLED:
print("2.2.8.2 - While hasnt exceeded max_limit")
while count < max_limit:
count += 1
if DEBUG_ENABLED:
print(
"2.2.8.3 - Read users response. If its a valid dream, add message to dream. Analyse and Validate"
)
message = MessageTypeAnalysis(self.in_out.read_message())
message_nlp = message.is_dream(Request.SUBJECT)
if message_nlp is not None:
self.dream.add_message(message_nlp)
AnalyseDream(self.dream)
validate_dream = DreamValidator(self.dream)
if DEBUG_ENABLED:
print(
"2.2.8.4 - If Dream no longer has a Subject missing. Return the dreams subject - RETURN dream's subject"
)
if Request.SUBJECT not in validate_dream.outstanding_features(
):
if self.dream.get_subject() == Subject.CLIENT:
self.in_out.write_message(
self._random_choice(CLIENT_SUBJECT_RESPONSE))
if self.dream.get_subject() == Subject.SOMEONE:
self.in_out.write_message(
self._random_choice(SOMEONE_SUBJECT_RESPONSE))
#TODO: could be crashing because of topic
if self.dream.get_subject() == Subject.OTHER:
self.in_out.write_message(
self._random_choice(
OTHER_SUBJECT_RESPONSE.format(
self.dream.get_topics)))
return self.dream.get_subject()
if DEBUG_ENABLED:
print(
"2.2.8.5 - Dream still doesn't have a sunject: {} - repeat request for Subject of dream depending on count. count = {} go to 2.2.3"
.format(self.dream.get_subject(), count))
if count == 1:
self.in_out.write_message(
self._random_choice(REPEAT_REQUEST_SUBJECT))
elif count == 2:
self.in_out.write_message(
self._random_choice(REPEAT_REQUEST_SUBJECT))
else:
self.in_out.write_message(
self._random_choice(URGENT_REQUEST_SUBJECT))
if DEBUG_ENABLED:
print(
"2.2.8.6 - No Subject has been given. Subject = {} - RETURN None"
.format(self.dream.get_subject()))
self.in_out.write_message(self._random_choice(NO_SUBJECT_GIVEN))
return None