def age(self, chat_id, msg):
tokens = msg["text"].split()
if len(tokens) > 1:
full_name = msg["text"].replace("/age ", "")
unidecode_name = unidecode.unidecode(full_name).lower().replace(' ', '')
result_query = self.database.execute_query(QueryBuilder.get_query(chat_id, unidecode_name))
if len(result_query) == 1:
age = result_query[0][2]
self.bot.sendMessage(chat_id, full_name + " is " + str(age) + " years old")
elif len(result_query) > 1:
message = full_name + " is ambigious\n you could have meant:"
for row in result_query:
message += "\t" + row[2] + "\n"
self.bot.sendMessage(chat_id, message)
else:
self.bot.sendMessage(chat_id,
full_name + " does not exist yet \nUse /birth for " + full_name + " to exists")
else:
self.bot.sendMessage(chat_id, "usage /age <name of person>")
def birthday(self, chat_id, msg):
tokens = msg["text"].split()
if len(tokens) > 1:
full_name = msg["text"].replace("/birthday ", "")
unidecode_name = unidecode.unidecode(full_name).lower().replace(' ', '')
result_query = self.database.execute_query(QueryBuilder.get_query(chat_id, unidecode_name))
if len(result_query) == 1:
if time.time() - result_query[0][3] < 60 and result_query[0][3] is not 0:
if not result_query[0][4]:
self.bot.sendMessage(chat_id, "You already asked, try again in one minute")
self.database.execute_query(QueryBuilder.update_already_asked(chat_id, unidecode_name, True))
return
elif len(result_query) > 1:
message = full_name + " is ambigious\n you could have meant:"
for row in result_query:
message += "\t" + row[2] + "\n"
self.bot.sendMessage(chat_id, message)
return
else:
self.bot.sendMessage(chat_id,
full_name + " does not exist yet \nUse /birth for " + full_name + " to exist")
return
self.database.execute_query(QueryBuilder.update_already_asked(chat_id, unidecode_name, False))
self.database.execute_query(QueryBuilder.update_last_modified(chat_id, unidecode_name, int(time.time())))
result_query = self.database.execute_query(QueryBuilder.update_age(chat_id, unidecode_name, 1))
if len(result_query) == 1:
age = result_query[0][2]
birthday_msg = "Whoooohoooooo! Happy birthday! " + full_name + " is now " + str(age) + " years old"
self.bot.sendMessage(chat_id, birthday_msg)
elif len(result_query) > 1:
message = full_name + " is ambigious\n you could have meant:"
for row in result_query:
message += "\t" + row[2] + "\n"
self.bot.sendMessage(chat_id, message)
else:
self.bot.sendMessage(chat_id,
full_name + " does not exist yet \nUse /birth for " + full_name + " to exist")
else:
self.bot.sendMessage(chat_id, "usage /birthday <name of person>")
