main.py

import os
import sys
import logging
import telegram
from telegram_bot.bot import execute_bot_command
from telegram_bot.mocks import bot as bot_mock, update as update_mock

def parse_instructions(update):
    if update and update.message and isinstance(update.message.text, str):
        instructions = update.message.text.split()
        command = instructions[0].split('@', 1)[0]
        args = instructions[1:]

        return command, args

    return None, None

def telegram_bot(request):
    logging.basicConfig(
        format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
        level=logging.INFO
    )

    bot = telegram.Bot(token=os.getenv('TOKEN'))

    if request and request.method == "POST":
        update = telegram.Update.de_json(request.get_json(force=True), bot)
        command, args = parse_instructions(update)

        execute_bot_command(command, args, bot, update)

# Used to test the bot on commandline by: python main.py /command [args]
if __name__ == '__main__':
    command = sys.argv[1]
    args = sys.argv[2:]

    execute_bot_command(command, args, bot_mock, update_mock)