def imageSearch(self, telegramBot, update): p = re.compile('([^ ]*) (.*)') query = (p.match(update.message.text).groups()[1] or '').strip() self.logDebug(u'Google image search (chat_id: %s, query: %s)' % (update.message.chat_id, query or 'None')) imageSources = ExternalAPI.searchGoogleImage(query) if imageSources: self.bot.sendMessage(chat_id=update.message.chat_id, text=imageSources[0]) else: self.bot.sendMessage(chat_id=update.message.chat_id, text=u'Não encontrei imagem relacionada')
def testSearchGoogleImage(self): result = ExternalAPI.searchGoogleImage('amigo') self.assertItemsEqual(result, filter(lambda img: re.match('http.*gstatic.*', img), result))