2021-08-14 15:26:10 +00:00
|
|
|
import re
|
|
|
|
import traceback
|
2021-12-31 14:44:34 +00:00
|
|
|
from typing import List, Union
|
2021-08-14 15:26:10 +00:00
|
|
|
|
2023-05-02 12:07:19 +00:00
|
|
|
from bots.aiogram.client import dp, bot, token
|
2022-07-14 13:23:15 +00:00
|
|
|
from config import Config
|
2023-03-05 11:15:44 +00:00
|
|
|
from core.builtins import Bot, Plain, Image, Voice, MessageSession as MS, ErrorMessage
|
2023-02-05 14:33:33 +00:00
|
|
|
from core.builtins.message.chain import MessageChain
|
2022-01-18 12:32:43 +00:00
|
|
|
from core.logger import Logger
|
2023-02-05 14:33:33 +00:00
|
|
|
from core.types import MsgInfo, Session, FetchTarget as FT, FetchedSession as FS, \
|
|
|
|
FinishedSession as FinS
|
|
|
|
from core.utils.image import image_split
|
2021-08-30 18:53:39 +00:00
|
|
|
from database import BotDBUtil
|
2021-08-14 15:26:10 +00:00
|
|
|
|
2022-07-14 13:23:15 +00:00
|
|
|
enable_analytics = Config('enable_analytics')
|
|
|
|
|
|
|
|
|
2022-01-15 14:28:35 +00:00
|
|
|
class FinishedSession(FinS):
|
|
|
|
async def delete(self):
|
|
|
|
"""
|
|
|
|
用于删除这条消息。
|
|
|
|
"""
|
|
|
|
try:
|
|
|
|
for x in self.result:
|
|
|
|
await x.delete()
|
|
|
|
except Exception:
|
2022-01-20 13:31:50 +00:00
|
|
|
Logger.error(traceback.format_exc())
|
2021-09-10 18:05:27 +00:00
|
|
|
|
|
|
|
|
2021-08-14 15:26:10 +00:00
|
|
|
class MessageSession(MS):
|
|
|
|
class Feature:
|
|
|
|
image = True
|
2021-08-18 13:11:14 +00:00
|
|
|
voice = True
|
2022-01-16 13:24:15 +00:00
|
|
|
embed = False
|
2021-10-11 14:45:28 +00:00
|
|
|
forward = False
|
2021-11-16 14:19:48 +00:00
|
|
|
delete = True
|
2022-01-16 13:24:15 +00:00
|
|
|
quote = True
|
2022-01-16 13:25:39 +00:00
|
|
|
wait = True
|
2021-08-14 15:26:10 +00:00
|
|
|
|
2022-08-27 14:02:26 +00:00
|
|
|
async def sendMessage(self, msgchain, quote=True, disable_secret_check=False,
|
|
|
|
allow_split_image=True) -> FinishedSession:
|
2021-12-31 14:44:34 +00:00
|
|
|
msgchain = MessageChain(msgchain)
|
2022-01-08 08:08:25 +00:00
|
|
|
if not msgchain.is_safe and not disable_secret_check:
|
2023-03-05 11:15:44 +00:00
|
|
|
return await self.sendMessage(Plain(ErrorMessage(self.locale.t("error.message.chain.unsafe"))))
|
2022-06-26 06:06:26 +00:00
|
|
|
self.sent.append(msgchain)
|
2021-12-31 14:44:34 +00:00
|
|
|
count = 0
|
|
|
|
send = []
|
|
|
|
for x in msgchain.asSendable(embed=False):
|
|
|
|
if isinstance(x, Plain):
|
|
|
|
send_ = await bot.send_message(self.session.target, x.text,
|
|
|
|
reply_to_message_id=self.session.message.message_id if quote
|
2023-04-30 03:30:59 +00:00
|
|
|
and count == 0 and self.session.message else None)
|
2022-01-18 12:32:43 +00:00
|
|
|
Logger.info(f'[Bot] -> [{self.target.targetId}]: {x.text}')
|
2022-08-27 14:02:26 +00:00
|
|
|
send.append(send_)
|
|
|
|
count += 1
|
2021-12-31 14:44:34 +00:00
|
|
|
elif isinstance(x, Image):
|
2022-08-27 14:02:26 +00:00
|
|
|
if allow_split_image:
|
|
|
|
split = await image_split(x)
|
|
|
|
for xs in split:
|
|
|
|
with open(await xs.get(), 'rb') as image:
|
|
|
|
send_ = await bot.send_photo(self.session.target, image,
|
|
|
|
reply_to_message_id=self.session.message.message_id
|
|
|
|
if quote
|
2023-04-30 03:30:59 +00:00
|
|
|
and count == 0
|
|
|
|
and self.session.message else None)
|
2022-08-27 14:02:26 +00:00
|
|
|
Logger.info(f'[Bot] -> [{self.target.targetId}]: Image: {str(xs.__dict__)}')
|
|
|
|
send.append(send_)
|
|
|
|
count += 1
|
|
|
|
else:
|
|
|
|
with open(await x.get(), 'rb') as image:
|
|
|
|
send_ = await bot.send_photo(self.session.target, image,
|
|
|
|
reply_to_message_id=self.session.message.message_id
|
|
|
|
if quote
|
2023-04-30 03:30:59 +00:00
|
|
|
and count == 0
|
|
|
|
and self.session.message else None)
|
2022-08-27 14:02:26 +00:00
|
|
|
Logger.info(f'[Bot] -> [{self.target.targetId}]: Image: {str(x.__dict__)}')
|
|
|
|
send.append(send_)
|
|
|
|
count += 1
|
2021-12-31 14:44:34 +00:00
|
|
|
elif isinstance(x, Voice):
|
|
|
|
with open(x.path, 'rb') as voice:
|
|
|
|
send_ = await bot.send_audio(self.session.target, voice,
|
|
|
|
reply_to_message_id=self.session.message.message_id if quote
|
2023-04-30 03:30:59 +00:00
|
|
|
and count == 0 and self.session.message else None)
|
2022-01-18 12:32:43 +00:00
|
|
|
Logger.info(f'[Bot] -> [{self.target.targetId}]: Voice: {str(x.__dict__)}')
|
2022-08-27 14:02:26 +00:00
|
|
|
send.append(send_)
|
|
|
|
count += 1
|
|
|
|
|
2022-06-28 06:11:03 +00:00
|
|
|
msgIds = []
|
|
|
|
for x in send:
|
|
|
|
msgIds.append(x.message_id)
|
2023-02-06 10:48:09 +00:00
|
|
|
return FinishedSession(self, msgIds, send)
|
2021-08-14 15:26:10 +00:00
|
|
|
|
|
|
|
async def checkPermission(self):
|
2022-08-27 07:38:13 +00:00
|
|
|
if self.session.message.chat.type == 'private' or self.target.senderId in self.custom_admins \
|
2023-04-30 03:30:59 +00:00
|
|
|
or self.target.senderInfo.query.isSuperUser:
|
2021-08-14 15:26:10 +00:00
|
|
|
return True
|
|
|
|
admins = [member.user.id for member in await dp.bot.get_chat_administrators(self.session.message.chat.id)]
|
2021-08-18 13:11:14 +00:00
|
|
|
if self.session.sender in admins:
|
2021-08-14 15:26:10 +00:00
|
|
|
return True
|
|
|
|
return False
|
|
|
|
|
2022-01-08 08:37:03 +00:00
|
|
|
async def checkNativePermission(self):
|
|
|
|
if self.session.message.chat.type == 'private':
|
|
|
|
return True
|
|
|
|
admins = [member.user.id for member in await dp.bot.get_chat_administrators(self.session.message.chat.id)]
|
|
|
|
if self.session.sender in admins:
|
|
|
|
return True
|
|
|
|
return False
|
|
|
|
|
2023-03-04 09:40:38 +00:00
|
|
|
def asDisplay(self, text_only=False):
|
2023-05-18 10:11:22 +00:00
|
|
|
if self.session.message.text:
|
|
|
|
return self.session.message.text
|
|
|
|
return ''
|
2021-08-14 15:26:10 +00:00
|
|
|
|
2023-05-02 12:07:19 +00:00
|
|
|
async def toMessageChain(self):
|
|
|
|
lst = []
|
|
|
|
if self.session.message.photo:
|
|
|
|
file = await bot.get_file(self.session.message.photo[-1]['file_id'])
|
|
|
|
lst.append(Image(f'https://api.telegram.org/file/bot{token}/{file.file_path}'))
|
|
|
|
if self.session.message.caption:
|
|
|
|
lst.append(Plain(self.session.message.caption))
|
|
|
|
if self.session.message.text:
|
|
|
|
lst.append(Plain(self.session.message.text))
|
|
|
|
return MessageChain(lst)
|
|
|
|
|
2021-08-14 15:26:10 +00:00
|
|
|
async def delete(self):
|
|
|
|
try:
|
2022-01-15 14:28:35 +00:00
|
|
|
for x in self.session.message:
|
|
|
|
await x.delete()
|
2021-08-23 12:44:31 +00:00
|
|
|
except Exception:
|
2022-01-20 13:31:50 +00:00
|
|
|
Logger.error(traceback.format_exc())
|
2021-08-14 15:26:10 +00:00
|
|
|
|
|
|
|
class Typing:
|
|
|
|
def __init__(self, msg: MS):
|
|
|
|
self.msg = msg
|
|
|
|
|
|
|
|
async def __aenter__(self):
|
2021-11-12 14:25:53 +00:00
|
|
|
# await bot.answer_chat_action(self.msg.session.target, 'typing')
|
2021-08-23 16:49:54 +00:00
|
|
|
pass
|
2021-08-14 15:26:10 +00:00
|
|
|
|
|
|
|
async def __aexit__(self, exc_type, exc_val, exc_tb):
|
|
|
|
pass
|
|
|
|
|
|
|
|
|
2022-01-15 13:46:23 +00:00
|
|
|
class FetchedSession(FS):
|
|
|
|
def __init__(self, targetFrom, targetId):
|
|
|
|
self.target = MsgInfo(targetId=f'{targetFrom}|{targetId}',
|
|
|
|
senderId=f'{targetFrom}|{targetId}',
|
|
|
|
targetFrom=targetFrom,
|
|
|
|
senderFrom=targetFrom,
|
2022-06-12 14:30:02 +00:00
|
|
|
senderName='',
|
2022-06-28 06:11:03 +00:00
|
|
|
clientName='Telegram', messageId=0, replyId=None)
|
2022-01-15 13:46:23 +00:00
|
|
|
self.session = Session(message=False, target=targetId, sender=targetId)
|
|
|
|
self.parent = MessageSession(self.target, self.session)
|
|
|
|
|
|
|
|
|
2021-08-21 15:58:07 +00:00
|
|
|
class FetchTarget(FT):
|
2022-01-05 11:14:45 +00:00
|
|
|
name = 'Telegram'
|
|
|
|
|
2021-08-14 15:26:10 +00:00
|
|
|
@staticmethod
|
2022-01-15 13:46:23 +00:00
|
|
|
async def fetch_target(targetId) -> Union[FetchedSession, bool]:
|
2021-08-18 13:11:14 +00:00
|
|
|
matchChannel = re.match(r'^(Telegram\|.*?)\|(.*)', targetId)
|
2021-08-14 15:26:10 +00:00
|
|
|
if matchChannel:
|
2022-01-15 13:46:23 +00:00
|
|
|
return FetchedSession(matchChannel.group(1), matchChannel.group(2))
|
2021-08-14 15:26:10 +00:00
|
|
|
else:
|
2021-08-18 13:11:14 +00:00
|
|
|
return False
|
2021-08-30 18:53:39 +00:00
|
|
|
|
2021-10-08 11:54:27 +00:00
|
|
|
@staticmethod
|
2022-01-15 13:46:23 +00:00
|
|
|
async def fetch_target_list(targetList: list) -> List[FetchedSession]:
|
2021-10-08 11:54:27 +00:00
|
|
|
lst = []
|
|
|
|
for x in targetList:
|
|
|
|
fet = await FetchTarget.fetch_target(x)
|
|
|
|
if fet:
|
|
|
|
lst.append(fet)
|
|
|
|
return lst
|
|
|
|
|
2021-08-30 18:53:39 +00:00
|
|
|
@staticmethod
|
2023-03-29 14:42:53 +00:00
|
|
|
async def post_message(module_name, message, user_list: List[FetchedSession] = None, i18n=False, **kwargs):
|
2021-08-30 18:53:39 +00:00
|
|
|
if user_list is not None:
|
|
|
|
for x in user_list:
|
|
|
|
try:
|
2023-03-29 14:42:53 +00:00
|
|
|
if i18n:
|
2023-08-07 05:04:13 +00:00
|
|
|
await x.sendDirectMessage(x.parent.locale.t(message, **kwargs))
|
2023-04-05 05:36:11 +00:00
|
|
|
|
2023-03-29 14:42:53 +00:00
|
|
|
else:
|
|
|
|
await x.sendDirectMessage(message)
|
2022-07-14 13:23:15 +00:00
|
|
|
if enable_analytics:
|
|
|
|
BotDBUtil.Analytics(x).add('', module_name, 'schedule')
|
2021-08-30 18:53:39 +00:00
|
|
|
except Exception:
|
2022-01-20 13:31:50 +00:00
|
|
|
Logger.error(traceback.format_exc())
|
2021-08-30 18:53:39 +00:00
|
|
|
else:
|
2022-08-29 15:54:20 +00:00
|
|
|
get_target_id = BotDBUtil.TargetInfo.get_enabled_this(module_name, "Telegram")
|
2021-08-30 18:53:39 +00:00
|
|
|
for x in get_target_id:
|
2022-08-29 15:54:20 +00:00
|
|
|
fetch = await FetchTarget.fetch_target(x.targetId)
|
2021-08-30 18:53:39 +00:00
|
|
|
if fetch:
|
|
|
|
try:
|
2023-03-29 14:42:53 +00:00
|
|
|
if i18n:
|
2023-08-07 05:04:13 +00:00
|
|
|
await fetch.sendDirectMessage(fetch.parent.locale.t(message, **kwargs))
|
2023-04-05 05:36:11 +00:00
|
|
|
|
2023-03-29 14:42:53 +00:00
|
|
|
else:
|
|
|
|
await fetch.sendDirectMessage(message)
|
2022-07-14 13:23:15 +00:00
|
|
|
if enable_analytics:
|
|
|
|
BotDBUtil.Analytics(fetch).add('', module_name, 'schedule')
|
2021-08-30 18:53:39 +00:00
|
|
|
except Exception:
|
2022-01-20 13:31:50 +00:00
|
|
|
Logger.error(traceback.format_exc())
|
2023-02-04 14:42:21 +00:00
|
|
|
|
|
|
|
|
|
|
|
Bot.MessageSession = MessageSession
|
|
|
|
Bot.FetchTarget = FetchTarget
|