Oh shit

wbb/modules/reverse.py

@@ -10,7 +10,7 @@ from pyrogram import filters
 
 from wbb import app
 from wbb.core.decorators.errors import capture_err
-from wbb.modules.nsfw import get_file_id_from_message
+from wbb.utils.functions import get_file_id_from_message
 
 
 @app.on_message(filters.command("reverse"))

wbb/utils/dbfunctions.py

@@ -45,7 +45,6 @@ captchadb = db.captcha
 antiservicedb = db.antiservice
 pmpermitdb = db.pmpermit
 welcomedb = db.welcome_text
-nsfwdb = db.nsfw
 captcha_cachedb = db.captcha_cache
 blacklist_filtersdb = db.blacklistFilters
 pipesdb = db.pipes
@@ -544,30 +543,6 @@ async def del_welcome(chat_id: int):
     return await welcomedb.delete_one({"chat_id": chat_id})
 
 
-"""NSFW System"""
-
-
-async def is_nsfw_on(chat_id: int) -> bool:
-    chat = await nsfwdb.find_one({"chat_id": chat_id})
-    if not chat:
-        return True
-    return False
-
-
-async def nsfw_on(chat_id: int):
-    is_nsfw = await is_nsfw_on(chat_id)
-    if is_nsfw:
-        return
-    return await nsfwdb.delete_one({"chat_id": chat_id})
-
-
-async def nsfw_off(chat_id: int):
-    is_nsfw = await is_nsfw_on(chat_id)
-    if not is_nsfw:
-        return
-    return await nsfwdb.insert_one({"chat_id": chat_id})
-
-
 """ CAPTCHA CACHE SYSTEM """
 
 
@@ -772,7 +747,6 @@ async def clean_restart_stage() -> dict:
     }
 
 
-
 """FLOOD System"""
 
 

wbb/utils/filter_groups.py

@@ -28,10 +28,8 @@ karma_negative_group = 4
 regex_group = 5
 welcome_captcha_group = 6
 antiflood_group = 7
-nsfw_detect_group = 8
-blacklist_filters_group = 9
-pipes_group = 10
-taglog_group = 11
-chat_watcher_group = 12
-flood_group = 13
-autocorrect_group = 14
+blacklist_filters_group = 8
+taglog_group = 9
+chat_watcher_group = 10
+flood_group = 11
+autocorrect_group = 12

wbb/utils/functions.py

@@ -301,3 +301,36 @@ async def test_ARQ(message):
         except Exception:
             results += f"**{key.capitalize()}:** `Failed`\n"
     return results
+
+
+async def get_file_id_from_message(message):
+    file_id = None
+    if message.document:
+        if int(message.document.file_size) > 3145728:
+            return
+        mime_type = message.document.mime_type
+        if mime_type != "image/png" and mime_type != "image/jpeg":
+            return
+        file_id = message.document.file_id
+
+    if message.sticker:
+        if message.sticker.is_animated:
+            if not message.sticker.thumbs:
+                return
+            file_id = message.sticker.thumbs[0].file_id
+        else:
+            file_id = message.sticker.file_id
+
+    if message.photo:
+        file_id = message.photo.file_id
+
+    if message.animation:
+        if not message.animation.thumbs:
+            return
+        file_id = message.animation.thumbs[0].file_id
+
+    if message.video:
+        if not message.video.thumbs:
+            return
+        file_id = message.video.thumbs[0].file_id
+    return file_id