|
@@ -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
|