telegram.py

import logging

from module.models import Notification
from module.network import RequestContent
from module.utils import load_image

logger = logging.getLogger(__name__)


class TelegramNotification(RequestContent):
    def __init__(self, token, chat_id):
        super().__init__()
        self.photo_url = f"https://api.telegram.org/bot{token}/sendPhoto"
        self.message_url = f"https://api.telegram.org/bot{token}/sendMessage"
        self.chat_id = chat_id

    @staticmethod
    def gen_message(notify: Notification) -> str:
        text = f"""
        番剧名称：{notify.official_title}\n季度： 第{notify.season}季\n更新集数： 第{notify.episode}集
        """
        return text.strip()

    def post_msg(self, notify: Notification) -> bool:
        text = self.gen_message(notify)
        data = {
            "chat_id": self.chat_id,
            "caption": text,
            "text": text,
            "disable_notification": True,
        }
        photo = load_image(notify.poster_path)
        if photo:
            resp = self.post_files(self.photo_url, data, files={"photo": photo})
        else:
            resp = self.post_data(self.message_url, data)
        logger.debug(f"Telegram notification: {resp.status_code}")
        return resp.status_code == 200