some fixes

This commit is contained in:
levina 2022-02-11 22:40:15 +07:00 committed by GitHub
parent 34690ed1ec
commit 82f5102595
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -1,4 +1,5 @@
import asyncio import asyncio
from driver.core import user from driver.core import user
from pyrogram.types import Message from pyrogram.types import Message
from pyrogram import Client, filters from pyrogram import Client, filters
@ -6,6 +7,7 @@ from config import BOT_USERNAME, SUDO_USERS
from driver.filters import command, other_filters from driver.filters import command, other_filters
from driver.database.dbchat import remove_served_chat from driver.database.dbchat import remove_served_chat
from driver.database.dbqueue import remove_active_chat from driver.database.dbqueue import remove_active_chat
from driver.database.dbpunish import is_gbanned_user
from pyrogram.errors import UserAlreadyParticipant, UserNotParticipant from pyrogram.errors import UserAlreadyParticipant, UserNotParticipant
from driver.decorators import authorized_users_only, bot_creator from driver.decorators import authorized_users_only, bot_creator
@ -16,6 +18,10 @@ from driver.decorators import authorized_users_only, bot_creator
@authorized_users_only @authorized_users_only
async def join_chat(c: Client, m: Message): async def join_chat(c: Client, m: Message):
chat_id = m.chat.id chat_id = m.chat.id
user_id = m.from_user.id
if await is_gbanned_user(user_id):
await m.reply_text("❗️ **You've blocked from using this bot!**")
return
try: try:
invitelink = await c.export_chat_invite_link(chat_id) invitelink = await c.export_chat_invite_link(chat_id)
if invitelink.startswith("https://t.me/+"): if invitelink.startswith("https://t.me/+"):
@ -35,6 +41,10 @@ async def join_chat(c: Client, m: Message):
@authorized_users_only @authorized_users_only
async def leave_chat(_, m: Message): async def leave_chat(_, m: Message):
chat_id = m.chat.id chat_id = m.chat.id
user_id = m.from_user.id
if await is_gbanned_user(user_id):
await m.reply_text("❗️ **You've blocked from using this bot!**")
return
try: try:
await user.leave_chat(chat_id) await user.leave_chat(chat_id)
await remove_active_chat(chat_id) await remove_active_chat(chat_id)