mirror of
https://github.com/EstrellaXD/Auto_Bangumi.git
synced 2026-03-20 03:46:40 +08:00
39 lines
1.3 KiB
Python
39 lines
1.3 KiB
Python
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()
|
|
|
|
async 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 = await self.post_files(self.photo_url, data, files={"photo": photo})
|
|
else:
|
|
resp = await self.post_data(self.message_url, data)
|
|
logger.debug("Telegram notification: %s", resp.status_code)
|
|
return resp.status_code == 200
|