def __init__(self, auth_key):
#self.enable_msg_id = dict()
self.key = '198410'
self.chats = dict()
self.users = dict()
self.chat_user_actions = dict()
self.whereeat = WhereToEat('AIzaSyB18YNnS59cL8dVFpOoIGXfATCyvITSbFU')
self.bot = Updater(auth_key)
self.manualbot = Bot(token=auth_key)
# Get the dispatcher to register handlers
dp = self.bot.dispatcher
# Register commands
dp.addTelegramCommandHandler("start", self.start)
dp.addTelegramCommandHandler("enable", self.enable)
dp.addTelegramCommandHandler("eat", self.eat)
dp.addTelegramCommandHandler("help", self.help)
dp.addTelegramMessageHandler(self.echo)
dp.addErrorHandler(self.error)
class DeboscioBot:
def __init__(self, auth_key):
#self.enable_msg_id = dict()
self.key = '198410'
self.chats = dict()
self.users = dict()
self.chat_user_actions = dict()
self.whereeat = WhereToEat('AIzaSyB18YNnS59cL8dVFpOoIGXfATCyvITSbFU')
self.bot = Updater(auth_key)
self.manualbot = Bot(token=auth_key)
# Get the dispatcher to register handlers
dp = self.bot.dispatcher
# Register commands
dp.addTelegramCommandHandler("start", self.start)
dp.addTelegramCommandHandler("enable", self.enable)
dp.addTelegramCommandHandler("eat", self.eat)
dp.addTelegramCommandHandler("help", self.help)
dp.addTelegramMessageHandler(self.echo)
dp.addErrorHandler(self.error)
def is_security_cleared(self, user_id):
if user_id in self.users:
# I know this user but
return self.users[user_id].enabled
def message_event(self, bot, message):
# Is it a new chat
if (self.is_a_new_chat(message.chat.id)):
logging.debug("new chat")
self.chats[message.chat.id] = DeboscioChat(message)
result = bot.sendMessage(message.chat.id, text='Hi ' + message.from_user.first_name)
else:
logging.debug("old chat")
self.chats[message.chat.id].add_msg(message)
# Is it a new user
if (self.is_a_new_user(message.from_user.id)):
logging.debug("new user")
self.users[message.from_user.id] = DeboscioUser(message.from_user)
# Is it a msg with location
if (message.location is not None):
self.users[message.from_user.id].set_location(message.location.latitude, message.location.longitude)
result = bot.sendMessage(message.chat.id, text='I got the new position')
def is_a_new_chat(self, chat_id):
return not (chat_id in self.chats)
def is_a_new_user(self, user_id):
return not (user_id in self.users)
def start(self, bot, update):
self.message_event(bot,update.message)
bot.sendMessage(update.message.chat_id, text='Identify yourself!')
def help(self, bot, update):
self.message_event(bot,update.message)
bot.sendMessage(update.message.chat_id, text='Nobody can helps you!')
def eat(self, bot, update):
self.message_event(bot,update.message)
# do I've a location for this user?
user = self.users[update.message.from_user.id]
if not (isinstance(user.latitude, float) and isinstance(user.longitude,float)):
bot.sendMessage(update.message.chat_id, text=user.first_name + ' send me your location.')
return
bot.sendChatAction(chat_id=update.message.chat_id, action=telegram.ChatAction.TYPING)
logging.debug("Looking for a place...")
places = self.whereeat.find_places(update.message.chat_id, update.message.from_user.id, user.latitude, user.longitude, max_results = 3)
logging.debug(places)
reply_markup = ReplyKeyboardMarkup([places])
bot.sendMessage(update.message.chat_id, text='Choose', reply_markup = reply_markup)
self.chat_user_actions[(update.message.chat_id, update.message.from_user.id)] = self.whereeat.select_place
def enable(self, bot, update):
self.message_event(bot,update.message)
bot.sendMessage(update.message.chat_id, text='Tell me the code!') #,reply_markup=reply_markup)
# one command at time per chat
self.chat_user_actions[(update.message.chat_id, update.message.from_user.id)] = self.authenticate
# self.enable_msg_id[val.chat_id] = val.message_id
#print("Sent reply: " + str(self.enable_msg_id[val.chat_id]))
def echo(self, bot, update):
if not (self.is_a_new_chat(update.message.chat.id)):
if not (self.chats[update.message.chat.id].active):
return
self.message_event(bot,update.message)
logging.debug(update.message)
# Check if the msg is a reply to a previous command
if (update.message.chat_id, update.message.from_user.id) in self.chat_user_actions:
_call_method = self.chat_user_actions[(update.message.chat_id, update.message.from_user.id)]
ret = _call_method( chat_id = update.message.chat_id, user_id = update.message.from_user.id, is_admin = self.is_security_cleared(update.message.from_user.id), msg = update.message.text)
bot.sendLocation(update.message.chat_id, latitude=float(ret['lat']), longitude=float(ret['lng']))
bot.sendMessage(update.message.chat_id, text='Got it!', reply_markup=ReplyKeyboardHide())
del self.chat_user_actions[(update.message.chat_id, update.message.from_user.id)]
# if update.message.reply_to_message is not None:
# if (update.message.chat_id in self.enable_msg_id) & (update.message.reply_to_message.message_id == self.enable_msg_id[update.message.chat_id]):
# # We got reply for enable command
# if self.authenticate(update.message):
# bot.sendMessage(update.message.chat_id, text='Got it!')
if self.is_security_cleared(update.message.from_user.id):
bot.sendMessage(update.message.chat_id, text='Keep talking')
def authenticate(self, chat_id, user_id, is_admin, msg):
# check key and add user
if msg == self.key:
self.users[user_id].enabled = True
return True
else:
self.users[user_id].enabled = False
return False
def error(bot, update, error):
loggin.warn('Update "%s" caused error "%s"' % (update, error))
raise ValueError('Parameter should...')
