Update save_media.py

Author: PredatorHackerzZ

handlers/save_media.py

@@ -1,4 +1,4 @@
-# (c) @AbirHasan2005 | @PredatorHackerzZ
+# (c) @AbirHasan2005 | @PredatorHackerzZ | @MrAbhi2k3
 
 import asyncio
 import requests
@@ -48,11 +48,15 @@ async def forward_to_channel(bot: Client, message: Message, editable: Message):
             )
         return await forward_to_channel(bot, message, editable)
 
-
 async def save_batch_media_in_channel(bot: Client, editable: Message, message_ids: list):
     try:
+        if editable.reply_to_message.from_user.id not in Config.OTHER_USERS_CAN_SAVE_FILE:
+            await editable.reply_text("You are not authorized to save files.")
+            return
+
         message_ids_str = ""
-        for message in (await bot.get_messages(chat_id=editable.chat.id, message_ids=message_ids)):
+        for message_id in message_ids:
+            message = await bot.get_messages(chat_id=editable.chat.id, message_ids=message_id)
             sent_message = await forward_to_channel(bot, message, editable)
             if sent_message is None:
                 continue
@@ -102,6 +106,9 @@ async def save_batch_media_in_channel(bot: Client, editable: Message, message_id
 
 async def save_media_in_channel(bot: Client, editable: Message, message: Message):
     try:
+        if message.from_user.id not in Config.OTHER_USERS_CAN_SAVE_FILE:
+            await editable.reply_text("You are not authorized to save files.")
+            return
         forwarded_msg = await message.forward(Config.DB_CHANNEL)
         file_er_id = str(forwarded_msg.id)
         await forwarded_msg.reply_text(