def cancel(bot, update):
user = update.message.from_user
logger.info("User %s canceled the conversation." % user.first_name)
update.message.reply_text('Bye! I hope we can talk again some day.',
reply_markup=ReplyKeyboardRemove())
return ConversationHandler.END
def pickDay(bot, update):
daysOfWeek = [
'Monday', 'Tuesday', 'Wednesday', 'Thursday', 'Friday', 'Saturday',
'Sunday'
]
userId = str(update.message.chat.id)
if update.message.text == '/weekly':
mapUserCurrent[userId] = {'typeOrWeekly': []}
elif update.message.text in daysOfWeek:
mapUserCurrent[userId]['typeOrWeekly'].append(update.message.text)
logger.info("%s" % (update.message.text))
if update.message.text == 'Done':
reply_keyboard = [['Proceed to next step']]
update.message.reply_text('Please',
reply_markup=ReplyKeyboardMarkup(
reply_keyboard, one_time_keyboard=True))
return FROM_PROPOSTION
else:
reply_keyboard = [[i] for i in daysOfWeek
if i not in mapUserCurrent[userId]['typeOrWeekly']]
reply_keyboard.append(['Done'])
update.message.reply_text('On which day of the week ?',
reply_markup=ReplyKeyboardMarkup(
reply_keyboard, one_time_keyboard=True))
return PICK_DAY
def toProposition(bot, update):
mapUserCurrent[str(update.message.chat.id)]['from'] = update.message.text
user = update.message.from_user
logger.info("From of %s: %s" % (user.first_name, update.message.text))
update.message.reply_text("Where will you be going to?")
return TO_CONFIRMATION
def getConnection(bot, update):
idReq = str(update.message.chat.id)
logger.info("Wanting to %s at %s " %
(mapUserCurrent[idReq]['by'], update.message.text))
time = datetime.datetime.now()
timeStr = ''
# if re.match('^now$', update.message.text):
if re.match('^\d[:h]\d{0,2}$', update.message.text, re.IGNORECASE):
h = int(update.message.text[0:1])
m = int(update.message.text[2:])
if h > time.hour or (h == time.hour and m >= time.minute):
time = time.replace(hour=h, minute=m)
else:
time = time.replace(hour=h, minute=m) + datetime.timedelta(days=1)
timeStr = str(int(time.timestamp() * 1000))
elif re.match('^\d\d[:h]\d{0,2}$', update.message.text, re.IGNORECASE):
h = int(update.message.text[0:2])
m = int(update.message.text[3:])
if h > time.hour or (h == time.hour and m >= time.minute):
time = time.replace(hour=h, minute=m)
else:
time = time.replace(hour=h, minute=m) + datetime.timedelta(days=1)
timeStr = str(int(time.timestamp() * 1000))
elif re.match('^(?i)now$', update.message.text):
timeStr = None
else:
logger.warn("problem with time regex")
return -1
mapUserCurrent[idReq]['time'] = timeStr
# print(mapUserCurrent[idReq])
# try:
qhandler = qh.QueryHandler()
queryResponse = qhandler.getConnexion(mapUserCurrent[idReq]['from'],
mapUserCurrent[idReq]['to'],
mapUserCurrent[idReq]['via'],
mapUserCurrent[idReq]['time'],
mapUserCurrent[idReq]['by'])
mapUserCurrent[idReq]['fullJsons'] = queryResponse
parsedResponse = [
parser.Parser().parseConnexion(i) for i in queryResponse['connections']
]
mapUserCurrent[idReq]['jsons'] = parsedResponse
response = helper.Helper().getConnexionsStrings(
queryResponse['connections'][:NUMBER_OF_TRAINS])
update.message.reply_text("Which train do you want?",
reply_markup=ReplyKeyboardMarkup(
response, one_time_keyboard=True))
# except:
# logger.warn("error somewhere")
# update.message.reply_text("No connection available?",
# reply_markup=ReplyKeyboardMarkup([[NO_CONNECTION_MESSAGE]], one_time_keyboard=True))
return CHOOSE_TRAIN
def fromConfirm(bot, update):
logger.info("Trying to leave from %s" % update.message.text)
listFrom = qh.QueryHandler().getStationsFromName(update.message.text)
listFrom = parser.Parser().parseStations(listFrom)
reply_keyboard = [[el] for el in listFrom]
update.message.reply_text("Please choose one of the following:",
reply_markup=ReplyKeyboardMarkup(
reply_keyboard, one_time_keyboard=True))
return TO_PROPOSTION
def connectionType(bot, update):
user = update.message.chat
logger.info("User %s started a new connection." % (user.first_name))
reply_keyboard = [['/unique', '/weekly']]
update.message.reply_text(
'Hi, do you want to add a unique connection or a weekly one?',
reply_markup=ReplyKeyboardMarkup(reply_keyboard,
one_time_keyboard=True))
def viaConfirm(bot, update):
logger.info("Trying to go via %s" % update.message.text)
listTo = qh.QueryHandler().getStationsFromName(update.message.text)
listTo = parser.Parser().parseStations(listTo)
reply_keyboard = [[el] for el in listTo]
update.message.reply_text("Please choose one of the following:",
reply_markup=ReplyKeyboardMarkup(
reply_keyboard, one_time_keyboard=True))
return ARRIVE_DEPART
def viaProposition(bot, update):
mapUserCurrent[str(update.message.chat.id)]['to'] = update.message.text
user = update.message.from_user
logger.info("To of %s: %s" % (user.first_name, update.message.text))
update.message.reply_text(
"Use /skip if you do not need to stop in between" +
" (via), otherwise please enter where you'd wish to stop")
return VIA_CONFIRMATION
def newFriend(bot, update):
user = update.message.from_user
logger.info("User %s started a new connection." % (user.first_name))
reply_keyboard = [[
KeyboardButton(text="send_contact", mapUserCurrent_contact=True)
]]
update.message.reply_text('Choose a contact to add?',
reply_markup=ReplyKeyboardMarkup(
reply_keyboard, one_time_keyboard=True))
return PROCESS_FRIEND
def whenTime(bot, update):
mapUserCurrent[str(
update.message.chat.id
)]['by'] = update.message.text == DEPART_ARRIVE_KEYBOARD[0][0]
logger.info("Wanting to \"%s\" " % update.message.text)
update.message.reply_text(
"Please type the time you want to %s. Possible input ..:.., now" %
update.message.text[:-8])
return GET_CONNECTION
def fromProposition(bot, update):
userId = str(update.message.chat.id)
if update.message.text == '/unique':
mapUserCurrent[userId] = {'typeOrWeekly': []}
user = update.message.from_user
logger.info("New connection day of %s: %s" %
(user.first_name, mapUserCurrent[userId]['typeOrWeekly']))
update.message.reply_text("Where will you be leaving from?")
return FROM_CONFIRMATION
def whenStartArrive(bot, update):
res = update.message.text
if res == 'Yes' or res == 'No': #only if not conrifmed
mapUserCurrent[str(update.message.chat.id)]['via'] = None
else:
mapUserCurrent[str(
update.message.chat.id)]['via'] = update.message.text
logger.info("Going via %s" % update.message.text)
reply_keyboard = DEPART_ARRIVE_KEYBOARD
update.message.reply_text(
"Do you want to arrive or depart by a certain time?",
reply_markup=ReplyKeyboardMarkup(reply_keyboard,
one_time_keyboard=True))
return TIME
