txlyre 2 сар өмнө
parent
commit
350ed860c0
1 өөрчлөгдсөн 8 нэмэгдсэн , 1 устгасан
  1. 8 1
      main.py

+ 8 - 1
main.py

@@ -38,7 +38,7 @@ load_dotenv()
 API_ID = int(os.getenv("API_ID"))
 API_HASH = os.getenv("API_HASH")
 BOT_TOKEN = os.getenv("BOT_TOKEN")
-ALLOWED_GROUPS = set(map(int, os.getenv("ALLOWED_GROUPS").split(",")))
+ALLOWED_GROUPS = set(map(lambda cid: int("-100" + cid), os.getenv("ALLOWED_GROUPS").split(",")))
 CAPTCHA_WAIT_TIME = int(os.getenv("CAPTCHA_WAIT_TIME", 80))
 MAX_CAPTCHA_ATTEMPTS = int(os.getenv("MAX_CAPTCHA_ATTEMPTS", 8))
 
@@ -133,6 +133,9 @@ class PendingUser:
 
 @bot.on(events.ChatAction)
 async def handler(event):
+    if event.chat_id not in ALLOWED_GROUPS:
+        return
+
     if (event.user_left or event.user_kicked) and event.user_id in pending:
         await pending[event.user_id].remove()
 
@@ -148,6 +151,10 @@ async def handler(event):
 
 @bot.on(events.NewMessage)
 async def handler(event):
+    peer_id = get_peer_id(event.peer_id)
+    if peer_id not in ALLOWED_GROUPS:
+        return
+
     pending_user = pending.get(event.sender.id)
     if not pending_user:
         return