def make_status_message(self,
msg_base: Message = None,
mid: MessageID = None) -> Message:
if mid is not None:
msg = self.message_cache[mid]
elif msg_base is not None:
msg = msg_base
else:
raise ValueError
reply = Message(
type=MsgType.Text,
chat=msg.chat,
author=msg.chat.make_system_member(uid=ChatID("filter_info"),
name="Filter middleware",
middleware=self),
deliver_to=coordinator.master,
)
if mid:
reply.uid = mid
else:
reply.uid = str(uuid.uuid4())
status = self.filter_reason(msg)
if not status:
# Blue circle emoji
status = "\U0001F535 This chat is not filtered."
else:
# Red circle emoji
status = "\U0001F534 " + status
reply.text = "Filter status for chat {chat_id} from {module_id}:\n" \
"\n" \
"{status}\n".format(
module_id=msg.chat.module_id,
chat_id=msg.chat.id,
status=status
)
command = MessageCommand(name="%COMMAND_NAME%",
callable_name="toggle_filter_by_chat_id",
kwargs={
"mid": reply.uid,
"module_id": msg.chat.module_id,
"chat_id": msg.chat.id
})
if self.is_chat_filtered_by_id(msg.chat):
command.name = "Unfilter by chat ID"
command.kwargs['value'] = False
else:
command.name = "Filter by chat ID"
command.kwargs['value'] = True
reply.commands = MessageCommands([command])
return reply
def test_pickle_message_removal(slave_channel, master_channel):
msg = Message()
msg.chat = slave_channel.alice
msg.uid = "uid"
msg_removal = MessageRemoval(source_channel=master_channel,
destination_channel=slave_channel,
message=msg)
msg_removal_dup = pickle.loads(pickle.dumps(msg_removal))
# Assume Message is picklable
for i in ("source_channel", "destination_channel"):
assert getattr(msg_removal, i) == getattr(msg_removal_dup, i)
def reply_message(self, message: Message, text: str):
reply = Message()
reply.text = text
# reply.chat = coordinator.slaves[message.chat.channel_id].get_chat(message.chat.chat_uid)
reply.chat = coordinator.slaves[message.chat.module_id].get_chat(
message.chat.uid)
reply.author = message.chat.make_system_member(
uid=self.middleware_id, name=self.middleware_name, middleware=self)
reply.type = MsgType.Text
# reply.deliver_to = coordinator.master
reply.deliver_to = coordinator.slaves[message.chat.module_id]
# reply.target = message
reply.uid = str(uuid.uuid4())
r2 = reply
coordinator.send_message(reply)
r2.deliver_to = coordinator.master
coordinator.send_message(r2)
def master_qr_code(self, uuid, status, qrcode=None):
status = int(status)
if self.qr_uuid == (uuid, status):
return
self.qr_uuid = (uuid, status)
msg = Message(
uid=f"ews_auth_{uuid}_{status}_{uuid4()}",
type=MsgType.Text,
chat=self.user_auth_chat,
author=self.user_auth_chat.other,
deliver_to=coordinator.master,
)
if status == 201:
msg.type = MsgType.Text
msg.text = self._('Confirm on your phone.')
self.master_qr_picture_id = None
elif status == 200:
msg.type = MsgType.Text
msg.text = self._("Successfully logged in.")
self.master_qr_picture_id = None
elif uuid != self.qr_uuid:
msg.type = MsgType.Image
file = NamedTemporaryFile(suffix=".png")
qr_url = "https://login.weixin.qq.com/l/" + uuid
QRCode(qr_url).png(file, scale=10)
msg.text = self._("QR code expired, please scan the new one.")
msg.path = Path(file.name)
msg.file = file
msg.mime = 'image/png'
if self.master_qr_picture_id is not None:
msg.edit = True
msg.edit_media = True
msg.uid = self.master_qr_picture_id
else:
self.master_qr_picture_id = msg.uid
if status in (200, 201) or uuid != self.qr_uuid:
coordinator.send_message(msg)
def reply_message_img(self, message: Message, im3: Image.Image):
reply = Message()
# reply.text = text
# reply.chat = coordinator.slaves[message.chat.channel_id].get_chat(message.chat.chat_uid)
reply.chat = coordinator.slaves[message.chat.module_id].get_chat(
message.chat.uid)
reply.author = message.chat.make_system_member(
uid=self.middleware_id, name=self.middleware_name, middleware=self)
reply.type = MsgType.Image
reply.mime = 'image/png'
f = tempfile.NamedTemporaryFile(suffix='.png')
img_data = io.BytesIO()
im3.save(img_data, format='png')
f.write(img_data.getvalue())
f.file.seek(0)
reply.file = f
reply.path = f.name
reply.filename = os.path.basename(reply.file.name)
# reply.deliver_to = coordinator.master
reply.deliver_to = coordinator.slaves[message.chat.module_id]
# reply.target = message
reply.uid = str(uuid.uuid4())
r2 = reply
coordinator.send_message(reply)
r2.type = MsgType.Image
r2.mime = 'image/png'
f = tempfile.NamedTemporaryFile(suffix='.png')
img_data = io.BytesIO()
im3.save(img_data, format='png')
f.write(img_data.getvalue())
f.file.seek(0)
r2.file = f
r2.path = f.name
r2.filename = os.path.basename(r2.file.name)
r2.deliver_to = coordinator.master
coordinator.send_message(r2)
def send_message(self, msg: Message) -> Message:
"""Send a message to WeChat.
Supports text, image, sticker, and file.
Args:
msg (channel.Message): Message Object to be sent.
Returns:
This method returns nothing.
Raises:
EFBChatNotFound:
Raised when a chat required is not found.
EFBMessageTypeNotSupported:
Raised when the message type sent is not supported by the
channel.
EFBOperationNotSupported:
Raised when an message edit request is sent, but not
supported by the channel.
EFBMessageNotFound:
Raised when an existing message indicated is not found.
E.g.: The message to be edited, the message referred
in the :attr:`msg.target <.Message.target>`
attribute.
EFBMessageError:
Raised when other error occurred while sending or editing the
message.
"""
if msg.chat == self.user_auth_chat:
raise EFBChatNotFound
chat: wxpy.Chat = self.chats.get_wxpy_chat_by_uid(msg.chat.uid)
# List of "SentMessage" response for all messages sent
r: List[wxpy.SentMessage] = []
self.logger.info("[%s] Sending message to WeChat:\n"
"uid: %s\n"
"UserName: %s\n"
"NickName: %s\n"
"Type: %s\n"
"Text: %s",
msg.uid,
msg.chat.uid, chat.user_name, chat.name, msg.type, msg.text)
try:
chat.mark_as_read()
except wxpy.ResponseError as e:
self.logger.exception(
"[%s] Error occurred while marking chat as read. (%s)", msg.uid, e)
send_text_only = False
self.logger.debug('[%s] Is edited: %s', msg.uid, msg.edit)
if msg.edit and msg.uid:
if self.flag('delete_on_edit'):
msg_ids = json.loads(msg.uid)
if msg.type in self.MEDIA_MSG_TYPES and not msg.edit_media:
# Treat message as text message to prevent resend of media
msg_ids = msg_ids[1:]
send_text_only = True
failed = 0
for i in msg_ids:
try:
ews_utils.message_id_to_dummy_message(i, self).recall()
except wxpy.ResponseError as e:
self.logger.error(
"[%s] Trying to recall message but failed: %s", msg.uid, e)
failed += 1
if failed:
raise EFBMessageError(
self.ngettext('Failed to recall {failed} out of {total} message, edited message was not sent.',
'Failed to recall {failed} out of {total} messages, edited message was not sent.',
len(msg_ids)).format(
failed=failed,
total=len(msg_ids)
))
# Not caching message ID as message recall feedback is not needed in edit mode
else:
raise EFBOperationNotSupported()
if send_text_only or msg.type in [MsgType.Text, MsgType.Link]:
if isinstance(msg.target, Message):
max_length = self.flag("max_quote_length")
qt_txt = msg.target.text or msg.target.type.name
if max_length > 0:
if len(qt_txt) >= max_length:
tgt_text = qt_txt[:max_length]
tgt_text += "…"
else:
tgt_text = qt_txt
elif max_length < 0:
tgt_text = qt_txt
else:
tgt_text = ""
if isinstance(chat, wxpy.Group) and not isinstance(msg.target.author, SelfChatMember):
tgt_alias = "@%s\u2005：" % msg.target.author.display_name
else:
tgt_alias = ""
msg.text = f"「{tgt_alias}{tgt_text}」\n- - - - - - - - - - - - - - -\n{msg.text}"
r.append(self._bot_send_msg(chat, msg.text))
self.logger.debug(
'[%s] Sent as a text message. %s', msg.uid, msg.text)
elif msg.type in (MsgType.Image, MsgType.Sticker, MsgType.Animation):
self.logger.info("[%s] Image/GIF/Sticker %s", msg.uid, msg.type)
convert_to = None
file = msg.file
assert file is not None
if self.flag('send_stickers_and_gif_as_jpeg'):
if msg.type == MsgType.Sticker or msg.mime == "image/gif":
convert_to = "image/jpeg"
else:
if msg.type == MsgType.Sticker:
convert_to = "image/gif"
if convert_to == "image/gif":
with NamedTemporaryFile(suffix=".gif") as f:
try:
img = Image.open(file)
try:
alpha = img.split()[3]
mask = Image.eval(
alpha, lambda a: 255 if a <= 128 else 0)
except IndexError:
mask = Image.eval(img.split()[0], lambda a: 0)
img = img.convert('RGB').convert(
'P', palette=Image.ADAPTIVE, colors=255)
img.paste(255, mask)
img.save(f, transparency=255)
msg.path = Path(f.name)
self.logger.debug(
'[%s] Image converted from %s to GIF', msg.uid, msg.mime)
file.close()
if f.seek(0, 2) > self.MAX_FILE_SIZE:
raise EFBMessageError(
self._("Image size is too large. (IS02)"))
f.seek(0)
r.append(self._bot_send_image(chat, f.name, f))
finally:
if not file.closed:
file.close()
elif convert_to == "image/jpeg":
with NamedTemporaryFile(suffix=".jpg") as f:
try:
img = Image.open(file).convert('RGBA')
out = Image.new("RGBA", img.size, (255, 255, 255, 255))
out.paste(img, img)
out.convert('RGB').save(f)
msg.path = Path(f.name)
self.logger.debug(
'[%s] Image converted from %s to JPEG', msg.uid, msg.mime)
file.close()
if f.seek(0, 2) > self.MAX_FILE_SIZE:
raise EFBMessageError(
self._("Image size is too large. (IS02)"))
f.seek(0)
r.append(self._bot_send_image(chat, f.name, f))
finally:
if not file.closed:
file.close()
else:
try:
if file.seek(0, 2) > self.MAX_FILE_SIZE:
raise EFBMessageError(
self._("Image size is too large. (IS01)"))
file.seek(0)
self.logger.debug(
"[%s] Sending %s (image) to WeChat.", msg.uid, msg.path)
filename = msg.filename or (msg.path and msg.path.name)
assert filename
r.append(self._bot_send_image(chat, filename, file))
finally:
if not file.closed:
file.close()
if msg.text:
r.append(self._bot_send_msg(chat, msg.text))
elif msg.type in (MsgType.File, MsgType.Audio):
self.logger.info("[%s] Sending %s to WeChat\nFileName: %s\nPath: %s\nFilename: %s",
msg.uid, msg.type, msg.text, msg.path, msg.filename)
filename = msg.filename or (msg.path and msg.path.name)
assert filename and msg.file
r.append(self._bot_send_file(chat, filename, file=msg.file))
if msg.text:
self._bot_send_msg(chat, msg.text)
if not msg.file.closed:
msg.file.close()
elif msg.type == MsgType.Video:
self.logger.info(
"[%s] Sending video to WeChat\nFileName: %s\nPath: %s", msg.uid, msg.text, msg.path)
filename = msg.filename or (msg.path and msg.path.name)
assert filename and msg.file
r.append(self._bot_send_video(chat, filename, file=msg.file))
if msg.text:
r.append(self._bot_send_msg(chat, msg.text))
if not msg.file.closed:
msg.file.close()
else:
raise EFBMessageTypeNotSupported()
msg.uid = ews_utils.generate_message_uid(r)
self.logger.debug(
'WeChat message is assigned with unique ID: %s', msg.uid)
return msg
def send_message(self, msg: Message) -> Message:
self.logger.debug("Received message: %r", msg)
self.messages.put(msg)
msg.uid = MessageID(str(uuid4()))
self.messages_sent[msg.uid] = msg
return msg
def handle_tg_img_preview(self, message: Message):
if not message or not message.file or not message.filename:
return
if message.author.uid == self.middleware_id: # trysh-middleware
# self.lg('self')
return
if message.type != MsgType.Image:
return
try:
message.file.seek(0)
fbs = message.file.read()
message.file.seek(0)
im: Image.Image = Image.open(io.BytesIO(fbs))
max_size = max(im.size)
min_size = min(im.size)
img_ratio = max_size / min_size
if img_ratio < 10.0:
return
im2 = im.copy()
for _ in range(100):
max_size = max(im.size)
min_size = min(im.size)
img_ratio = max_size / min_size
if img_ratio >= 10.0:
if im.width == min_size:
im = im.resize((im.width * 2, im.height),
box=(0, 0, 1, 1))
else:
im = im.resize((im.width, im.height * 2),
box=(0, 0, 1, 1))
continue
else:
break
im.paste(im2, (0, 0, im2.width, im2.height))
im3 = im.convert('RGB') # im.copy() #
reply = Message()
# reply.text = text
# reply.chat = coordinator.slaves[message.chat.channel_id].get_chat(message.chat.chat_uid)
reply.chat = coordinator.slaves[message.chat.module_id].get_chat(
message.chat.uid)
reply.author = message.chat.make_system_member(
uid=self.middleware_id,
name=self.middleware_name,
middleware=self)
reply.type = MsgType.Image
reply.mime = 'image/png'
f = tempfile.NamedTemporaryFile(suffix='.png')
img_data = io.BytesIO()
im3.save(img_data, format='png')
f.write(img_data.getvalue())
f.file.seek(0)
reply.file = f
reply.path = f.name
reply.filename = os.path.basename(reply.file.name)
# reply.deliver_to = coordinator.master
reply.deliver_to = coordinator.master
# reply.target = message
reply.uid = str(uuid.uuid4())
coordinator.send_message(reply)
except BaseException as e:
self.lg(f'handle_tg_img_preview e:{e}')
pass