From 51eb2f90b983ae53f5116c2a7bfab07875efd133 Mon Sep 17 00:00:00 2001
From: Dan <14043624+delivrance@users.noreply.github.com>
Date: Fri, 3 Aug 2018 18:38:26 +0200
Subject: [PATCH] Add support for editing photousing file IDs

---
 .../methods/messages/edit_message_media.py    | 25 +++++++++++++++++++
 1 file changed, 25 insertions(+)

diff --git a/pyrogram/client/methods/messages/edit_message_media.py b/pyrogram/client/methods/messages/edit_message_media.py
index 35c24f1a..d79ecfbe 100644
--- a/pyrogram/client/methods/messages/edit_message_media.py
+++ b/pyrogram/client/methods/messages/edit_message_media.py
@@ -16,9 +16,12 @@
 # You should have received a copy of the GNU Lesser General Public License
 # along with Pyrogram.  If not, see <http://www.gnu.org/licenses/>.
 
+import binascii
 import os
+import struct
 
 from pyrogram.api import functions, types
+from pyrogram.api.errors import FileIdInvalid
 from pyrogram.client.ext import BaseClient, utils
 from pyrogram.client.types import (
     InputMediaPhoto
@@ -55,6 +58,28 @@ class EditMessageMedia(BaseClient):
                 media = types.InputMediaPhotoExternal(
                     url=media.media
                 )
+            else:
+                try:
+                    decoded = utils.decode(media.media)
+                    fmt = "<iiqqqqi" if len(decoded) > 24 else "<iiqq"
+                    unpacked = struct.unpack(fmt, decoded)
+                except (AssertionError, binascii.Error, struct.error):
+                    raise FileIdInvalid from None
+                else:
+                    if unpacked[0] != 2:
+                        media_type = BaseClient.MEDIA_TYPE_ID.get(unpacked[0], None)
+
+                        if media_type:
+                            raise FileIdInvalid("The file_id belongs to a {}".format(media_type))
+                        else:
+                            raise FileIdInvalid("Unknown media type: {}".format(unpacked[0]))
+
+                    media = types.InputMediaPhoto(
+                        id=types.InputPhoto(
+                            id=unpacked[2],
+                            access_hash=unpacked[3]
+                        )
+                    )
 
         r = self.send(
             functions.messages.EditMessage(