Update Akeno/plugins/antinsfw.py

Akeno/plugins/antinsfw.py

@@ -26,16 +26,14 @@ from pyrogram.types import Chat, ChatMember, ChatPrivileges, Message
 
 from Akeno.utils.database import db
 from Akeno.utils.handler import *
+from Akeno.utils.logger import LOGS
 from config import CMD_HANDLER
 
 ANTINSFW_GROUPS = 12
 
-async def can_delete(client: Client, bot_id: int) -> bool:
-    member = await client.get_member(bot_id)
-    if member.status in [enums.ChatMemberStatus.ADMINISTRATOR, enums.ChatMemberStatus.OWNER]:
-        return True
-    else:
-        return False
+async def can_delete(client: Client, chat_id: int, bot_id: int) -> bool:
+    member = await client.get_chat_member(chat_id, bot_id)
+    return member.status in [enums.ChatMemberStatus.ADMINISTRATOR, enums.ChatMemberStatus.OWNER]
 
 def check_anti_nsfw(media) -> bool:
     url = "https://akeno.randydev.my.id/akeno/anti-nsfw"
@@ -82,14 +80,20 @@ async def antinsfw_filter(client: Client, message: Message):
         return
     if not message.photo:
         return
-    me = await client.get_me()
-    if message.photo:
-        file_id = message.photo.file_id
+    file_id = message.photo.file_id
     media = await client.download_media(file_id)
-    if check_anti_nsfw(media):
-        if await can_delete(chat, me.id):
-            return await message.delete()
+    if not media:
+        return
+    try:
+        if check_anti_nsfw(media):
+            me = await client.get_me()
+            if await can_delete(client, chat.id, me.id):
+                await message.delete()
+    except Exception as e:
+        LOGS.info(f"Error processing NSFW check: {e}")
+    finally:
+        if os.path.exists(media):
             os.remove(media)
 
 module = modules_help.add_module("antinsfw", __file__)
-module.add_command("antinsfw", "to anti nsfw auto delete messages")
+module.add_command("antinsfw", "to anti nsfw auto delete messages")