mir/pyrogram
2
0
Fork 0
mirror of https://github.com/pyrogram/pyrogram synced 2025-08-29 05:18:10 +00:00
Code Issues Releases Wiki Activity

Add photo field

Browse Source
This commit is contained in:
Dan 2018-03-31 12:20:22 +02:00
parent 692539f6a2
commit 846fb0b637
1 changed files with 42 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

43
pyrogram/client/utils.py
View File

@ -1,4 +1,5 @@
from base64 import b64encode, b64decode
from struct import pack
import pyrogram
from pyrogram.api import types
@ -103,6 +104,45 @@ def parse_message(message: types.Message, users: dict, chats: dict):
forward_from_message_id = forward_header.channel_post
forward_signature = forward_header.post_author
photo = None
media = message.media
if media:
if isinstance(media, types.MessageMediaPhoto):
photo = media.photo
if isinstance(photo, types.Photo):
sizes = photo.sizes
photo_sizes = []
for size in sizes:
if isinstance(size, (types.PhotoSize, types.PhotoCachedSize)):
location = size.location
if isinstance(location, types.FileLocation):
photo_size = pyrogram.PhotoSize(
file_id=encode(
pack(
"<iiqqqqi",
2,
location.dc_id,
photo.id,
photo.access_hash,
location.volume_id,
location.secret,
location.local_id
)
),
width=size.w,
height=size.h,
file_size=size.size
)
photo_sizes.append(photo_size)
photo = photo_sizes
return pyrogram.Message(
message_id=message.id,
date=message.date,
@ -118,7 +158,8 @@ def parse_message(message: types.Message, users: dict, chats: dict):
forward_from_message_id=forward_from_message_id,
forward_signature=forward_signature,
forward_date=forward_date,
edit_date=message.edit_date
edit_date=message.edit_date,
photo=photo
)