init

app/modules/telegram/__init__.py

@@ -0,0 +1,111 @@
+from typing import Optional, Union, List, Tuple
+
+from fastapi import Request
+
+from app.core import MediaInfo, TorrentInfo, settings
+from app.log import logger
+from app.modules import _ModuleBase
+from app.modules.telegram.telegram import Telegram
+
+
+class TelegramModule(_ModuleBase):
+
+    telegram: Telegram = None
+
+    def init_module(self) -> None:
+        self.telegram = Telegram()
+
+    def init_setting(self) -> Tuple[str, Union[str, bool]]:
+        return "MESSAGER", "telegram"
+
+    async def message_parser(self, request: Request) -> Optional[dict]:
+        """
+        解析消息内容，返回字典，注意以下约定值：
+        userid: 用户ID
+        username: 用户名
+        text: 内容
+        :param request:  请求体
+        :return: 消息内容、用户ID
+        """
+        """
+            {
+                'update_id': ,
+                'message': {
+                    'message_id': ,
+                    'from': {
+                        'id': ,
+                        'is_bot': False,
+                        'first_name': '',
+                        'username': '',
+                        'language_code': 'zh-hans'
+                    },
+                    'chat': {
+                        'id': ,
+                        'first_name': '',
+                        'username': '',
+                        'type': 'private'
+                    },
+                    'date': ,
+                    'text': ''
+                }
+            }
+        """
+        msg_json: dict = await request.json()
+        if msg_json:
+            message = msg_json.get("message", {})
+            text = message.get("text")
+            user_id = message.get("from", {}).get("id")
+            # 获取用户名
+            user_name = message.get("from", {}).get("username")
+            if text:
+                logger.info(f"收到Telegram消息：userid={user_id}, username={user_name}, text={text}")
+                # 检查权限
+                if text.startswith("/"):
+                    if str(user_id) not in settings.TELEGRAM_ADMINS.split(',') \
+                            and str(user_id) != settings.TELEGRAM_CHAT_ID:
+                        self.telegram.send_msg(title="只有管理员才有权限执行此命令", userid=user_id)
+                        return {}
+                else:
+                    if not str(user_id) in settings.TELEGRAM_USERS.split(','):
+                        self.telegram.send_msg(title="你不在用户白名单中，无法使用此机器人", userid=user_id)
+                        return {}
+                return {
+                    "userid": user_id,
+                    "username": user_name,
+                    "text": text
+                }
+        return None
+
+    def post_message(self, title: str,
+                     text: str = None, image: str = None, userid: Union[str, int] = None) -> Optional[bool]:
+        """
+        发送消息
+        :param title:  标题
+        :param text: 内容
+        :param image: 图片
+        :param userid:  用户ID
+        :return: 成功或失败
+        """
+        return self.telegram.send_msg(title=title, text=text, image=image, userid=userid)
+
+    def post_medias_message(self, title: str, items: List[MediaInfo],
+                            userid: Union[str, int] = None) -> Optional[bool]:
+        """
+        发送媒体信息选择列表
+        :param title:  标题
+        :param items:  消息列表
+        :param userid:  用户ID
+        :return: 成功或失败
+        """
+        return self.telegram.send_meidas_msg(title=title, medias=items, userid=userid)
+
+    def post_torrents_message(self, title: str, items: List[TorrentInfo],
+                              userid: Union[str, int] = None) -> Optional[bool]:
+        """
+        TODO 发送种子信息选择列表
+        :param title: 标题
+        :param items:  消息列表
+        :param userid:  用户ID
+        :return: 成功或失败
+        """
+        pass

app/modules/telegram/telegram.py

@@ -0,0 +1,194 @@
+from threading import Event, Thread
+from typing import Optional, List
+from urllib.parse import urlencode
+
+from app.core import settings, MediaInfo
+from app.log import logger
+from app.utils.http import RequestUtils
+from app.utils.singleton import Singleton
+
+
+class Telegram(metaclass=Singleton):
+
+    _poll_timeout: int = 5
+    _event = Event()
+
+    def __init__(self):
+        """
+        初始化参数
+        """
+        # Token
+        self._telegram_token = settings.TELEGRAM_TOKEN
+        # Chat Id
+        self._telegram_chat_id = settings.TELEGRAM_CHAT_ID
+        # 用户Chat Id列表
+        self._telegram_user_ids = settings.TELEGRAM_USERS.split(",")
+        # 管理员Chat Id列表
+        self._telegram_admin_ids = settings.TELEGRAM_ADMINS.split(",")
+        # 消息轮循
+        if self._telegram_token and self._telegram_chat_id:
+            self._thread = Thread(target=self.__start_telegram_message_proxy)
+
+    def send_msg(self, title: str, text: str = "", image: str = "", userid: str = "") -> Optional[bool]:
+        """
+        发送Telegram消息
+        :param title: 消息标题
+        :param text: 消息内容
+        :param image: 消息图片地址
+        :param userid: 用户ID，如有则只发消息给该用户
+        :userid: 发送消息的目标用户ID，为空则发给管理员
+        """
+        if not self._telegram_token or not self._telegram_chat_id:
+            return None
+
+        if not title and not text:
+            logger.warn("标题和内容不能同时为空")
+            return False
+
+        try:
+            # text中的Markdown特殊字符转义
+            text = text.replace("[", r"\[").replace("_", r"\_").replace("*", r"\*").replace("`", r"\`")
+            # 拼装消息内容
+            titles = str(title).split('\n')
+            if len(titles) > 1:
+                title = titles[0]
+                if not text:
+                    text = "\n".join(titles[1:])
+                else:
+                    text = "%s\n%s" % ("\n".join(titles[1:]), text)
+
+            if text:
+                caption = "*%s*\n%s" % (title, text.replace("\n\n", "\n"))
+            else:
+                caption = title
+
+            if userid:
+                chat_id = userid
+            else:
+                chat_id = self._telegram_chat_id
+
+            return self.__send_request(chat_id=chat_id, image=image, caption=caption)
+
+        except Exception as msg_e:
+            logger.error(f"发送消息失败：{msg_e}")
+            return False
+
+    def send_meidas_msg(self, medias: List[MediaInfo], userid: str = "", title: str = "") -> Optional[bool]:
+        """
+        发送媒体列表消息
+        """
+        if not self._telegram_token or not self._telegram_chat_id:
+            return None
+
+        try:
+            index, image, caption = 1, "", "*%s*" % title
+            for media in medias:
+                if not image:
+                    image = media.get_message_image()
+                if media.get_vote_string():
+                    caption = "%s\n%s. [%s](%s)\n%s，%s" % (caption,
+                                                           index,
+                                                           media.get_title_string(),
+                                                           media.get_detail_url(),
+                                                           media.get_type_string(),
+                                                           media.get_vote_string())
+                else:
+                    caption = "%s\n%s. [%s](%s)\n%s" % (caption,
+                                                        index,
+                                                        media.get_title_string(),
+                                                        media.get_detail_url(),
+                                                        media.get_type_string())
+                index += 1
+
+            if userid:
+                chat_id = userid
+            else:
+                chat_id = self._telegram_chat_id
+
+            return self.__send_request(chat_id=chat_id, image=image, caption=caption)
+
+        except Exception as msg_e:
+            logger.error(f"发送消息失败：{msg_e}")
+            return False
+
+    def __send_request(self, chat_id="", image="", caption="") -> bool:
+        """
+        向Telegram发送报文
+        """
+
+        def __res_parse(result):
+            if result and result.status_code == 200:
+                ret_json = result.json()
+                status = ret_json.get("ok")
+                if status:
+                    return True
+                else:
+                    logger.error(
+                        f"发送消息错误，错误码：{ret_json.get('error_code')}，错误原因：{ret_json.get('description')}")
+                    return False
+            elif result is not None:
+                logger.error(f"发送消息错误，错误码：{result.status_code}，错误原因：{result.reason}")
+                return False
+            else:
+                logger.error("发送消息错误，未知错误")
+                return False
+
+        # 请求
+        request = RequestUtils(proxies=settings.PROXY)
+
+        # 发送图文消息
+        if image:
+            res = request.get_res("https://api.telegram.org/bot%s/sendPhoto?" % self._telegram_token + urlencode(
+                {"chat_id": chat_id, "photo": image, "caption": caption, "parse_mode": "Markdown"}))
+            if __res_parse(res):
+                return True
+            else:
+                photo_req = request.get_res(image)
+                if photo_req and photo_req.content:
+                    res = request.post_res("https://api.telegram.org/bot%s/sendPhoto" % self._telegram_token,
+                                           data={"chat_id": chat_id, "caption": caption, "parse_mode": "Markdown"},
+                                           files={"photo": photo_req.content})
+                    if __res_parse(res):
+                        return True
+        # 发送文本消息
+        res = request.get_res("https://api.telegram.org/bot%s/sendMessage?" % self._telegram_token + urlencode(
+            {"chat_id": chat_id, "text": caption, "parse_mode": "Markdown"}))
+        return __res_parse(res)
+
+    def __start_telegram_message_proxy(self):
+        logger.info("Telegram消息接收服务启动")
+
+        def consume_messages(_offset: int, _sc_url: str, _ds_url: str) -> int:
+            try:
+                res = RequestUtils(proxies=settings.PROXY).get_res(
+                    _sc_url + urlencode({"timeout": self._poll_timeout, "offset": _offset}))
+                if res and res.json():
+                    for msg in res.json().get("result", []):
+                        # 无论本地是否成功，先更新offset，即消息最多成功消费一次
+                        _offset = msg["update_id"] + 1
+                        logger.debug("Telegram接收到消息: %s" % msg)
+                        local_res = RequestUtils(timeout=10).post_res(_ds_url, json=msg)
+                        logger.debug("Telegram message: %s processed, response is: %s" % (msg, local_res.text))
+            except Exception as e:
+                logger.error("Telegram 消息接收出现错误: %s" % e)
+            return _offset
+
+        offset = 0
+
+        while True:
+            if self._event.is_set():
+                logger.info("Telegram消息接收服务已停止")
+                break
+            index = 0
+            while index < 20 and not self._event.is_set():
+                offset = consume_messages(_offset=offset,
+                                          _sc_url="https://api.telegram.org/bot%s/getUpdates?" % self._telegram_token,
+                                          _ds_url="http://127.0.0.1:%s/api/v1/messages?token=%s" % (
+                                              settings.PORT, settings.API_TOKEN))
+                index += 1
+
+    def stop(self):
+        """
+        停止Telegram消息接收服务
+        """
+        self._event.set()