Enhance and use .get_input_photo on the generated code

This commit is contained in:
Lonami Exo
2017-10-08 13:45:14 +02:00
parent 62aec947c0
commit 83677fc927
2 changed files with 8 additions and 3 deletions

View File

@@ -20,8 +20,8 @@ from .tl.types import (
GeoPointEmpty, InputGeoPointEmpty, Photo, InputPhoto, PhotoEmpty,
InputPhotoEmpty, FileLocation, ChatPhotoEmpty, UserProfilePhotoEmpty,
FileLocationUnavailable, InputMediaUploadedDocument,
InputMediaUploadedPhoto,
DocumentAttributeFilename)
InputMediaUploadedPhoto, DocumentAttributeFilename, photos
)
def get_display_name(entity):
@@ -188,6 +188,9 @@ def get_input_photo(photo):
if type(photo).SUBCLASS_OF_ID == 0x846363e0: # crc32(b'InputPhoto')
return photo
if isinstance(photo, photos.Photo):
photo = photo.photo
if isinstance(photo, Photo):
return InputPhoto(id=photo.id, access_hash=photo.access_hash)