iframe-proxy | Sunbelt Computer Software

83 lines (62 loc) · 2.07 KB

import json
import telegram
import logging
# Logging is cool!
logger = logging.getLogger()
if logger.handlers:
    for handler in logger.handlers:
        logger.removeHandler(handler)
logging.basicConfig(level=logging.INFO)
OK_RESPONSE = {
    'statusCode': 200,
    'headers': {'Content-Type': 'application/json'},
    'body': json.dumps('ok')
ERROR_RESPONSE = {
    'statusCode': 400,
    'body': json.dumps('Oops, something went wrong!')
def configure_telegram():
    Configures the bot with a Telegram Token.
    Returns a bot instance.
    TELEGRAM_TOKEN = os.environ.get('TELEGRAM_TOKEN')
    if not TELEGRAM_TOKEN:
        logger.error('The TELEGRAM_TOKEN must be set')
        raise NotImplementedError
    return telegram.Bot(TELEGRAM_TOKEN)
def webhook(event, context):
    Runs the Telegram webhook.
    bot = configure_telegram()
    logger.info('Event: {}'.format(event))
    if event.get('httpMethod') == 'POST' and event.get('body'): 
        logger.info('Message received')
        update = telegram.Update.de_json(json.loads(event.get('body')), bot)
        chat_id = update.message.chat.id
        text = update.message.text
        if text == '/start':
            text = """Hello, human! I am an echo bot, built with Python and the Serverless Framework.
            You can take a look at my source code here: https://github.com/jonatasbaldin/serverless-telegram-bot.
            If you have any issues, please drop a tweet to my creator: https://twitter.com/jonatsbaldin. Happy botting!"""
        bot.sendMessage(chat_id=chat_id, text=text)
        logger.info('Message sent')
        return OK_RESPONSE
    return ERROR_RESPONSE
def set_webhook(event, context):
    Sets the Telegram bot webhook.
    logger.info('Event: {}'.format(event))
    bot = configure_telegram()
    url = 'https://{}/{}/'.format(
        event.get('headers').get('Host'),
        event.get('requestContext').get('stage'),
    webhook = bot.set_webhook(url)
    if webhook:
        return OK_RESPONSE
    return ERROR_RESPONSE

Sunbelt Computer Software

PL/B Language Development and Support

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Uh oh!

Sunbelt Computer Software

PL/B Language Development and Support

Uh oh!

FilesExpand file tree

handler.py

Latest commit

History

handler.py

File metadata and controls