modules/inline.py

@@ -11,7 +11,7 @@ from pyrogram.enums.chat_members_filter import ChatMembersFilter
 from dateutil.relativedelta import relativedelta
 from classes.errors.holo_user import UserNotFoundError, UserInvalidError
 from classes.holo_user import HoloUser
-from modules.utils import configGet, locale
+from modules.utils import configGet, jsonLoad, locale
 from modules.database import col_applications, col_spoilers
 from bson.objectid import ObjectId
 from bson.errors import InvalidId
@@ -64,19 +64,27 @@ async def inline_answer(client: Client, inline_query: InlineQuery):
         )
         return
 
+    results_forbidden = [
+        InlineQueryResultArticle(
+            title=locale("title", "inline", "forbidden", locale=inline_query.from_user),
+            input_message_content=InputTextMessageContent(
+                locale("message_content", "inline", "forbidden", locale=inline_query.from_user)
+            ),
+            description=locale("description", "inline", "forbidden", locale=inline_query.from_user)
+        )
+    ]
+
     try:
         holo_user = HoloUser(inline_query.from_user)
     except (UserNotFoundError, UserInvalidError):
         await inline_query.answer(
-        results=[
-            InlineQueryResultArticle(
-                title=locale("title", "inline", "forbidden", locale=inline_query.from_user),
-                input_message_content=InputTextMessageContent(
-                    locale("message_content", "inline", "forbidden", locale=inline_query.from_user)
-                ),
-                description=locale("description", "inline", "forbidden", locale=inline_query.from_user)
-                )
-            ]
+            results=results_forbidden
+        )
+        return
+
+    if path.exists(path.join(configGet("cache", "locations"), "group_members")) and (inline_query.from_user.id not in jsonLoad(path.join(configGet("cache", "locations"), "group_members"))):
+        await inline_query.answer(
+            results=results_forbidden
         )
         return