welcomecenterbot/handlers/handle_feedback.py

import json

from bot.api import (
    send_message,
    forward_message,
    get_chat_administrators,
)
from storage import storage
from bot.config import FEEDBACK_CHAT_ID
import logging


logger = logging.getLogger(__name__)
logging.basicConfig(level=logging.INFO)


async def handle_feedback(msg, state):
    mid = msg["message_id"]
    cid = msg["chat"]["id"]
    uid = msg["from"]["id"]
    if msg["text"] == "/start":
        r = await send_message(cid, "Напишите своё сообщение для администрации чата")
        logger.debug(r)
    elif state.is_talking(uid):
        r = await forward_message(cid, mid, state.talking[uid])
        logger.debug(r)
        state.aho(uid)
    else:
        r = await forward_message(cid, mid, FEEDBACK_CHAT_ID)
        logger.debug(r)
        support_msg_id = r["result"]["message_id"]
        # сохранение айди сообщения в приватной переписке с ботом
        storage.set(
            f"fbk-{support_msg_id}",
            json.dumps(
                {"author_id": uid, "message_id": mid, "chat_id": cid}
            ),
        )


async def handle_answer(msg):
    logger.info("handle answering feedback")
    logger.debug(msg)
    answered_msg = msg.get("reply_to_message")
    if answered_msg:
        if "from" not in answered_msg:
            answered_msg["from"] = msg.get("from_user")
        r = await get_chat_administrators(msg["chat"]["id"])
        logger.debug(r)
        admins = []
        for a in r["result"]:
            admins.append(a["user"]["id"])
        if answered_msg["from"]["is_bot"] and msg["from"]["id"] in admins:
            support_msg_id = str(answered_msg["message_id"])
            # получение сохраненного информации о сообщении для ответа
            stored_feedback = storage.get(f"fbk-{support_msg_id}")
            if stored_feedback:
                logger.info("handle an answer from feedback group")
                stored_feedback = json.loads(stored_feedback)
                r = await send_message(
                    f'{stored_feedback["chat_id"]}',
                    msg["text"],
                    reply_to=stored_feedback["message_id"],
                )
                logger.debug(r)