update nsfw_check.py

AeonOrg · Jul 15, 2024 · 8dc992d · 8dc992d
1 parent 259e46e
commit 8dc992d
Showing 1 changed file with 10 additions and 7 deletions.
diff --git a/bot/helper/aeon_utils/nsfw_check.py b/bot/helper/aeon_utils/nsfw_check.py
@@ -59,15 +59,18 @@ async def nsfw_precheck(message):
     if not reply_to:
         return False
 
-    if any(
-        isNSFW(getattr(reply_to, attr).file_name)
-        for attr in ['document', 'video']
-        if hasattr(reply_to, attr) and getattr(reply_to, attr)
-    ):
-        return True
+    try:
+        if any(
+            isNSFW(getattr(reply_to, attr).file_name)
+            for attr in ['document', 'video']
+            if hasattr(reply_to, attr) and getattr(reply_to, attr)
+        ):
+            return True
+    except AttributeError:
+        pass
 
     return any(
         isNSFW(getattr(reply_to, attr))
         for attr in ['caption', 'text']
         if hasattr(reply_to, attr) and getattr(reply_to, attr)
-    )
+    )