main.py

# -*- coding: utf-8 -*-

import os
import json
import logging

from flask import Flask, request, jsonify

from faces import faces


app = Flask(__name__)


@app.route('/')
def hello():
  """Return a friendly HTTP greeting."""
  return "Hello!"


@app.route('/{}'.format(os.environ['TELEGRAM_BOT_TOKEN']), methods=['POST'])
def webhook():
  req = request.get_json()
  logging.info(req)
  inline_query = req.get('inline_query')
  message = req.get('message')
  if inline_query:
    offset = inline_query.get('offset')
    offset = 0 if not offset else int(offset)
    query = inline_query.get('query')
    results = [{
        'type': 'article',
        'id': str(i + offset),
        'title': face,
        'description': u'{} {}'.format(query, face) if query else face,
        'message_text': u'{} {}'.format(query, face) if query else face,
    } for i, face in enumerate(faces[offset:offset + 50])]
    resp = {'method': 'answerInlineQuery',
            'inline_query_id': inline_query.get('id'),
            'results': results,
            'next_offset': str(offset + len(results))}
    return jsonify(**resp)
  elif message:
    text = message.get('text')
    if text == '/list':
      resp = {'method': 'sendMessage',
              'text': '\n'.join(faces),
              'chat_id': message.get('from').get('id')}
    elif text == '/start':
      resp = {'method': 'sendMessage',
              'text': '''I allow you to append funny ascii faces to your messages.
You can contribute to the code here:
https://github.com/phil-r/asciifacesbot''',
              'chat_id': message.get('from').get('id')}
    else:
      resp = {'method': 'sendMessage',
              'text': 'I know only /list command ¯\_(ツ)_/¯',
              'chat_id': message.get('from').get('id')}
    return jsonify(**resp)
  return 'OK'


@app.errorhandler(404)
def page_not_found(e):
  return 'Sorry, Nothing at this URL.', 404


@app.errorhandler(500)
def application_error(e):
  return 'Sorry, unexpected error: {}'.format(e), 500