mirror of
https://github.com/eternnoir/pyTelegramBotAPI.git
synced 2023-08-10 21:12:57 +03:00
Fix Things on copyMessage
This commit is contained in:
parent
58281f0a10
commit
b684c4f60d
@ -7,6 +7,8 @@ import sys
|
|||||||
import threading
|
import threading
|
||||||
import time
|
import time
|
||||||
|
|
||||||
|
from telebot.types import MessageID
|
||||||
|
|
||||||
logger = logging.getLogger('TeleBot')
|
logger = logging.getLogger('TeleBot')
|
||||||
formatter = logging.Formatter(
|
formatter = logging.Formatter(
|
||||||
'%(asctime)s (%(filename)s:%(lineno)d %(threadName)s) %(levelname)s - %(name)s: "%(message)s"'
|
'%(asctime)s (%(filename)s:%(lineno)d %(threadName)s) %(levelname)s - %(name)s: "%(message)s"'
|
||||||
@ -751,19 +753,24 @@ class TeleBot:
|
|||||||
apihelper.forward_message(self.token, chat_id, from_chat_id, message_id, disable_notification, timeout))
|
apihelper.forward_message(self.token, chat_id, from_chat_id, message_id, disable_notification, timeout))
|
||||||
|
|
||||||
def copy_message(self, chat_id, from_chat_id, message_id, caption=None, parse_mode=None, caption_entities=None,
|
def copy_message(self, chat_id, from_chat_id, message_id, caption=None, parse_mode=None, caption_entities=None,
|
||||||
reply_to_message_id=None, allow_sending_without_reply=None, reply_markup=None,
|
disable_notification=None, reply_to_message_id=None, allow_sending_without_reply=None, reply_markup=None,
|
||||||
disable_notification=None, timeout=None):
|
timeout=None):
|
||||||
# FIXME: rewrite the docstring
|
|
||||||
"""
|
"""
|
||||||
Use this method to copy messages of any kind.
|
Use this method to copy messages of any kind.
|
||||||
:param disable_notification:
|
|
||||||
:param chat_id: which chat to forward
|
:param chat_id: which chat to forward
|
||||||
:param from_chat_id: which chat message from
|
:param from_chat_id: which chat message from
|
||||||
:param message_id: message id
|
:param message_id: message id
|
||||||
|
:param caption:
|
||||||
|
:param parse_mode:
|
||||||
|
:param caption_entities:
|
||||||
|
:param disable_notification:
|
||||||
|
:param reply_to_message_id:
|
||||||
|
:param allow_sending_without_reply:
|
||||||
|
:param reply_markup:
|
||||||
:param timeout:
|
:param timeout:
|
||||||
:return: API reply.
|
:return: API reply.
|
||||||
"""
|
"""
|
||||||
return types.Message.de_json(
|
return MessageID.de_json(
|
||||||
apihelper.copy_message(self.token, chat_id, from_chat_id, message_id, caption, parse_mode, caption_entities,
|
apihelper.copy_message(self.token, chat_id, from_chat_id, message_id, caption, parse_mode, caption_entities,
|
||||||
reply_to_message_id, allow_sending_without_reply, reply_markup,
|
reply_to_message_id, allow_sending_without_reply, reply_markup,
|
||||||
disable_notification, timeout))
|
disable_notification, timeout))
|
||||||
|
@ -336,11 +336,11 @@ def copy_message(token, chat_id, from_chat_id, message_id, caption=None, parse_m
|
|||||||
if parse_mode is not None:
|
if parse_mode is not None:
|
||||||
payload['parse_mode'] = parse_mode
|
payload['parse_mode'] = parse_mode
|
||||||
if caption_entities is not None:
|
if caption_entities is not None:
|
||||||
payload['caption_entities'] = caption_entities
|
payload['caption_entities'] = _convert_entites(caption_entities)
|
||||||
if reply_to_message_id is not None:
|
if reply_to_message_id is not None:
|
||||||
payload['reply_to_message_id'] = reply_to_message_id
|
payload['reply_to_message_id'] = reply_to_message_id
|
||||||
if reply_markup is not None:
|
if reply_markup is not None:
|
||||||
payload['reply_markup'] = reply_markup
|
payload['reply_markup'] = _convert_markup(reply_markup)
|
||||||
if allow_sending_without_reply is not None:
|
if allow_sending_without_reply is not None:
|
||||||
payload['allow_sending_without_reply'] = allow_sending_without_reply
|
payload['allow_sending_without_reply'] = allow_sending_without_reply
|
||||||
if disable_notification is not None:
|
if disable_notification is not None:
|
||||||
@ -1309,6 +1309,12 @@ def _convert_markup(markup):
|
|||||||
return markup
|
return markup
|
||||||
|
|
||||||
|
|
||||||
|
def _convert_entites(entites):
|
||||||
|
if isinstance(entites[0], types.JsonSerializable):
|
||||||
|
return [entity.to_json() for entity in entites]
|
||||||
|
return entites
|
||||||
|
|
||||||
|
|
||||||
def convert_input_media(media):
|
def convert_input_media(media):
|
||||||
if isinstance(media, types.InputMedia):
|
if isinstance(media, types.InputMedia):
|
||||||
return media.convert_input_media()
|
return media.convert_input_media()
|
||||||
|
@ -261,6 +261,19 @@ class Chat(JsonDeserializable):
|
|||||||
self.location = location
|
self.location = location
|
||||||
|
|
||||||
|
|
||||||
|
class MessageID(JsonDeserializable):
|
||||||
|
@classmethod
|
||||||
|
def de_json(cls, json_string):
|
||||||
|
if(json_string is None):
|
||||||
|
return None
|
||||||
|
obj = cls.check_json(json_string)
|
||||||
|
message_id = obj['message_id']
|
||||||
|
return cls(message_id)
|
||||||
|
|
||||||
|
def __init__(self, message_id):
|
||||||
|
self.message_id = message_id
|
||||||
|
|
||||||
|
|
||||||
class Message(JsonDeserializable):
|
class Message(JsonDeserializable):
|
||||||
@classmethod
|
@classmethod
|
||||||
def de_json(cls, json_string):
|
def de_json(cls, json_string):
|
||||||
@ -548,7 +561,7 @@ class Message(JsonDeserializable):
|
|||||||
return self.__html_text(self.caption, self.caption_entities)
|
return self.__html_text(self.caption, self.caption_entities)
|
||||||
|
|
||||||
|
|
||||||
class MessageEntity(JsonDeserializable):
|
class MessageEntity(Dictionaryable, JsonSerializable, JsonDeserializable):
|
||||||
@classmethod
|
@classmethod
|
||||||
def de_json(cls, json_string):
|
def de_json(cls, json_string):
|
||||||
if (json_string is None): return None
|
if (json_string is None): return None
|
||||||
@ -569,6 +582,17 @@ class MessageEntity(JsonDeserializable):
|
|||||||
self.user = user
|
self.user = user
|
||||||
self.language = language
|
self.language = language
|
||||||
|
|
||||||
|
def to_json(self):
|
||||||
|
return json.dumps(self.to_dict())
|
||||||
|
|
||||||
|
def to_dict(self):
|
||||||
|
return {"type": self.type,
|
||||||
|
"offset": self.offset,
|
||||||
|
"length": self.length,
|
||||||
|
"url": self.url,
|
||||||
|
"user": self.user,
|
||||||
|
"language": self.language}
|
||||||
|
|
||||||
|
|
||||||
class Dice(JsonSerializable, Dictionaryable, JsonDeserializable):
|
class Dice(JsonSerializable, Dictionaryable, JsonDeserializable):
|
||||||
@classmethod
|
@classmethod
|
||||||
|
@ -293,6 +293,13 @@ class TestTeleBot:
|
|||||||
ret_msg = tb.forward_message(CHAT_ID, CHAT_ID, msg.message_id)
|
ret_msg = tb.forward_message(CHAT_ID, CHAT_ID, msg.message_id)
|
||||||
assert ret_msg.forward_from
|
assert ret_msg.forward_from
|
||||||
|
|
||||||
|
def test_copy_message(self):
|
||||||
|
text = 'CI copy_message Test Message'
|
||||||
|
tb = telebot.TeleBot(TOKEN)
|
||||||
|
msg = tb.send_message(CHAT_ID, text)
|
||||||
|
ret_msg = tb.copy_message(CHAT_ID, CHAT_ID, msg.message_id)
|
||||||
|
assert ret_msg
|
||||||
|
|
||||||
def test_forward_message_dis_noti(self):
|
def test_forward_message_dis_noti(self):
|
||||||
text = 'CI forward_message Test Message'
|
text = 'CI forward_message Test Message'
|
||||||
tb = telebot.TeleBot(TOKEN)
|
tb = telebot.TeleBot(TOKEN)
|
||||||
|
Loading…
Reference in New Issue
Block a user