Changes for L119
This commit is contained in:
parent
5ee932b326
commit
4632879ee4
@ -539,7 +539,7 @@ class Client(Methods, Scaffold):
|
||||
getattr(
|
||||
getattr(
|
||||
update, "message", None
|
||||
), "to_id", None
|
||||
), "peer_id", None
|
||||
), "channel_id", None
|
||||
) or getattr(update, "channel_id", None)
|
||||
|
||||
|
@ -89,15 +89,15 @@ class GetDialogs(Scaffold):
|
||||
messages = {}
|
||||
|
||||
for message in r.messages:
|
||||
to_id = message.to_id
|
||||
peer_id = message.peer_id
|
||||
|
||||
if isinstance(to_id, raw.types.PeerUser):
|
||||
if isinstance(peer_id, raw.types.PeerUser):
|
||||
if message.out:
|
||||
chat_id = to_id.user_id
|
||||
chat_id = peer_id.user_id
|
||||
else:
|
||||
chat_id = message.from_id
|
||||
chat_id = utils.get_raw_peer_id(message.from_id)
|
||||
else:
|
||||
chat_id = utils.get_peer_id(to_id)
|
||||
chat_id = utils.get_peer_id(peer_id)
|
||||
|
||||
messages[chat_id] = await types.Message._parse(self, message, users, chats)
|
||||
|
||||
|
@ -17,7 +17,7 @@
|
||||
# along with Pyrogram. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
import pyrogram
|
||||
from pyrogram import raw
|
||||
from pyrogram import raw, utils
|
||||
from pyrogram import types
|
||||
from ..object import Object
|
||||
|
||||
@ -64,7 +64,7 @@ class GameHighScore(Object):
|
||||
@staticmethod
|
||||
def _parse_action(client, service: raw.types.MessageService, users: dict):
|
||||
return GameHighScore(
|
||||
user=types.User._parse(client, users[service.from_id]),
|
||||
user=types.User._parse(client, users[utils.get_raw_peer_id(service.from_id)]),
|
||||
score=service.action.score,
|
||||
client=client
|
||||
)
|
||||
|
@ -422,7 +422,7 @@ class Message(Object, Update):
|
||||
if isinstance(action, raw.types.MessageActionChatAddUser):
|
||||
new_chat_members = [types.User._parse(client, users[i]) for i in action.users]
|
||||
elif isinstance(action, raw.types.MessageActionChatJoinedByLink):
|
||||
new_chat_members = [types.User._parse(client, users[message.from_id])]
|
||||
new_chat_members = [types.User._parse(client, users[utils.get_raw_peer_id(message.from_id)])]
|
||||
elif isinstance(action, raw.types.MessageActionChatDeleteUser):
|
||||
left_chat_member = types.User._parse(client, users[action.user_id])
|
||||
elif isinstance(action, raw.types.MessageActionChatEditTitle):
|
||||
@ -444,7 +444,7 @@ class Message(Object, Update):
|
||||
message_id=message.id,
|
||||
date=message.date,
|
||||
chat=types.Chat._parse(client, message, users, chats),
|
||||
from_user=types.User._parse(client, users.get(message.from_id, None)),
|
||||
from_user=types.User._parse(client, users.get(utils.get_raw_peer_id(message.from_id), None)),
|
||||
service=True,
|
||||
new_chat_members=new_chat_members,
|
||||
left_chat_member=left_chat_member,
|
||||
@ -472,7 +472,7 @@ class Message(Object, Update):
|
||||
if isinstance(action, raw.types.MessageActionGameScore):
|
||||
parsed_message.game_high_score = types.GameHighScore._parse_action(client, message, users)
|
||||
|
||||
if message.reply_to_msg_id and replies:
|
||||
if message.reply_to and replies:
|
||||
try:
|
||||
parsed_message.reply_to_message = await client.get_messages(
|
||||
parsed_message.chat.id,
|
||||
@ -501,13 +501,17 @@ class Message(Object, Update):
|
||||
forward_date = forward_header.date
|
||||
|
||||
if forward_header.from_id:
|
||||
forward_from = types.User._parse(client, users[forward_header.from_id])
|
||||
elif forward_header.from_name:
|
||||
forward_sender_name = forward_header.from_name
|
||||
raw_peer_id = utils.get_raw_peer_id(forward_header.from_id)
|
||||
peer_id = utils.get_peer_id(forward_header.from_id)
|
||||
|
||||
if peer_id > 0:
|
||||
forward_from = types.User._parse(client, users[raw_peer_id])
|
||||
else:
|
||||
forward_from_chat = types.Chat._parse_channel_chat(client, chats[forward_header.channel_id])
|
||||
forward_from_chat = types.Chat._parse_channel_chat(client, chats[raw_peer_id])
|
||||
forward_from_message_id = forward_header.channel_post
|
||||
forward_signature = forward_header.post_author
|
||||
elif forward_header.from_name:
|
||||
forward_sender_name = forward_header.from_name
|
||||
|
||||
photo = None
|
||||
location = None
|
||||
@ -608,7 +612,7 @@ class Message(Object, Update):
|
||||
message_id=message.id,
|
||||
date=message.date,
|
||||
chat=types.Chat._parse(client, message, users, chats),
|
||||
from_user=types.User._parse(client, users.get(message.from_id, None)),
|
||||
from_user=types.User._parse(client, users.get(utils.get_raw_peer_id(message.from_id), None)),
|
||||
text=(
|
||||
Str(message.message).init(entities) or None
|
||||
if media is None or web_page is not None
|
||||
@ -664,7 +668,7 @@ class Message(Object, Update):
|
||||
client=client
|
||||
)
|
||||
|
||||
if message.reply_to_msg_id and replies:
|
||||
if message.reply_to and replies:
|
||||
try:
|
||||
parsed_message.reply_to_message = await client.get_messages(
|
||||
parsed_message.chat.id,
|
||||
|
@ -224,13 +224,20 @@ class Chat(Object):
|
||||
|
||||
@staticmethod
|
||||
def _parse(client, message: raw.types.Message or raw.types.MessageService, users: dict, chats: dict) -> "Chat":
|
||||
if isinstance(message.to_id, raw.types.PeerUser):
|
||||
return Chat._parse_user_chat(client, users[message.to_id.user_id if message.out else message.from_id])
|
||||
if isinstance(message.peer_id, raw.types.PeerUser):
|
||||
return Chat._parse_user_chat(
|
||||
client,
|
||||
users[
|
||||
message.peer_id.user_id
|
||||
if message.out
|
||||
else utils.get_raw_peer_id(message.from_id)
|
||||
]
|
||||
)
|
||||
|
||||
if isinstance(message.to_id, raw.types.PeerChat):
|
||||
return Chat._parse_chat_chat(client, chats[message.to_id.chat_id])
|
||||
if isinstance(message.peer_id, raw.types.PeerChat):
|
||||
return Chat._parse_chat_chat(client, chats[message.peer_id.chat_id])
|
||||
|
||||
return Chat._parse_channel_chat(client, chats[message.to_id.channel_id])
|
||||
return Chat._parse_channel_chat(client, chats[message.peer_id.channel_id])
|
||||
|
||||
@staticmethod
|
||||
def _parse_dialog(client, peer, users: dict, chats: dict):
|
||||
|
@ -211,11 +211,10 @@ class ChatMember(Object):
|
||||
client=client
|
||||
)
|
||||
|
||||
if isinstance(member, (raw.types.ChannelParticipantCreator, raw.types.ChatParticipantCreator)):
|
||||
if isinstance(member, raw.types.ChatParticipantCreator):
|
||||
return ChatMember(
|
||||
user=user,
|
||||
status="creator",
|
||||
title=getattr(member, "rank", None),
|
||||
client=client
|
||||
)
|
||||
|
||||
@ -228,6 +227,25 @@ class ChatMember(Object):
|
||||
client=client
|
||||
)
|
||||
|
||||
if isinstance(member, raw.types.ChannelParticipantCreator):
|
||||
permissions = member.admin_rights
|
||||
|
||||
return ChatMember(
|
||||
user=user,
|
||||
status="creator",
|
||||
title=member.rank,
|
||||
invited_by=invited_by,
|
||||
can_change_info=permissions.change_info,
|
||||
can_post_messages=permissions.post_messages,
|
||||
can_edit_messages=permissions.edit_messages,
|
||||
can_delete_messages=permissions.delete_messages,
|
||||
can_restrict_members=permissions.ban_users,
|
||||
can_invite_users=permissions.invite_users,
|
||||
can_pin_messages=permissions.pin_messages,
|
||||
can_promote_members=permissions.add_admins,
|
||||
client=client
|
||||
)
|
||||
|
||||
if isinstance(member, raw.types.ChannelParticipantAdmin):
|
||||
permissions = member.admin_rights
|
||||
|
||||
|
@ -231,7 +231,22 @@ MIN_CHAT_ID = -2147483647
|
||||
MAX_USER_ID = 2147483647
|
||||
|
||||
|
||||
def get_raw_peer_id(peer: raw.base.Peer) -> Union[int, None]:
|
||||
"""Get the raw peer id from a Peer object"""
|
||||
if isinstance(peer, raw.types.PeerUser):
|
||||
return peer.user_id
|
||||
|
||||
if isinstance(peer, raw.types.PeerChat):
|
||||
return peer.chat_id
|
||||
|
||||
if isinstance(peer, raw.types.PeerChannel):
|
||||
return peer.channel_id
|
||||
|
||||
return None
|
||||
|
||||
|
||||
def get_peer_id(peer: raw.base.Peer) -> int:
|
||||
"""Get the non-raw peer id from a Peer object"""
|
||||
if isinstance(peer, raw.types.PeerUser):
|
||||
return peer.user_id
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user