app.telegram_bot.routes

Complexity

Total Complexity

4

Size/Duplication

Total Lines	53
Duplicated Lines	0 %

Importance

Changes

0

4 Functions

Rating	Name	Duplication	Size	Complexity
A	get_message()	0	8	1
A	delete()	0	8	1
A	index()	0	8	1
A	webhook()	0	9	1

# -*- coding:utf-8 -*-
"""
Providing routes for our application
"""

from flask import Blueprint, request as rq
from telebot import types

from app.telegram_bot.bot import bot, APP_SITE, TOKEN

bot_routes = Blueprint("route", __name__)


@bot_routes.route(f"/telegram_bot/{TOKEN}", methods=["POST"])
def get_message():
    """
    Get all messages
    :return:
    """
    bot.process_new_updates([types.Update.de_json(rq.stream.read().decode("utf-8"))])
    return "Ok", 200


@bot_routes.route("/telegram_bot/")
@bot_routes.route("/telegram_bot/heartbeat")
def index():
    """
    Test functionality
    :return:
    """
    return {k: v for k, v in bot.get_me().__dict__.items() if v}, 200


@bot_routes.route("/telegram_bot/set")
def webhook():
    """
    Set telegram webhook
    :return:
    """
    bot.remove_webhook()
    bot.set_webhook(url=f"https://{APP_SITE}/{TOKEN}")
    return "⚓ Webhook was set.", 200


@bot_routes.route("/telegram_bot/delete")
def delete():
    """
    Delete telegram webhook
    :return:
    """
    bot.remove_webhook()
    return "⚓ Webhook was deleted.", 200