from flask import Flask, request
import flask
import logging
from bot1 import process_the_update as bot1_process
from bot2 import process_the_update as bot2_process
from bot3 import process_the_update as bot3_process
import telebot

app = Flask(__name__)

gunicorn_error_logger = logging.getLogger('gunicorn.error')
app.logger.handlers.extend(gunicorn_error_logger.handlers)
app.logger.setLevel(logging.DEBUG)


# Read messages from telegram servers
@app.route('/route_for_bot1', methods=['POST'])
def bot1():
    if flask.request.headers.get('content-type') == 'application/json':
        app.logger.debug(request.data)
        json_string = flask.request.get_data().decode('utf-8')
        update = telebot.types.Update.de_json(json_string)
        bo1_process([update])
        return ''
    else:
        flask.abort(403)


@app.route('/route_for_bot2', methods=['POST'])
def bot2():
    if flask.request.headers.get('content-type') == 'application/json':
        app.logger.debug(request.data)
        json_string = flask.request.get_data().decode('utf-8')
        update = telebot.types.Update.de_json(json_string)
        bot2_process([update])
        return ''
    else:
        flask.abort(403)


@app.route('/route_for_bot3', methods=['POST'])
def bot3():
    if flask.request.headers.get('content-type') == 'application/json':
        app.logger.debug(request.data)
        json_string = flask.request.get_data().decode('utf-8')
        update = telebot.types.Update.de_json(json_string)
        bot3_process([update])
        return ''
    else:
        flask.abort(403)

def process_the_update(update):
    bot.process_new_updates(update)

LOG_FILENAME = "BOT_NAME.log"
LOG_FORMAT = '%(asctime)s (%(filename)-12s:%(lineno)-5d %(threadName)-15s) %(levelname)-8s %(name)-12s: "%(message)s"'
DATE_FMT = "%c"
logging.basicConfig(level=logging.INFO,
                    format=LOG_FORMAT,
                    datefmt=DATE_FMT)
logger = logging.getLogger(__name__)
formatter = logging.Formatter(LOG_FORMAT, datefmt=DATE_FMT)
handler = RotatingFileHandler(LOG_FILENAME, maxBytes=104857600, backupCount=7)  # 100mb
handler.setFormatter(formatter)
telebot.logger.setLevel(logging.DEBUG)
telebot.console_output_handler.setFormatter(formatter)
telebot.logger.addHandler(handler)