This commit is contained in:
levina 2022-02-11 23:10:25 +07:00 committed by GitHub
parent a0587c19d6
commit 9d1f3ccc30
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -1,6 +1,6 @@
# Copyright (C) 2021 By VeezMusicProject # Copyright (C) 2021 By VeezMusicProject
from driver.core import user from driver.core import user, bot
from driver.queues import QUEUE from driver.queues import QUEUE
from driver.database.dbpunish import is_gbanned_user from driver.database.dbpunish import is_gbanned_user
from pyrogram import Client, filters from pyrogram import Client, filters
@ -8,7 +8,6 @@ from program.utils.inline import menu_markup, stream_markup
from pyrogram.types import CallbackQuery, InlineKeyboardButton, InlineKeyboardMarkup from pyrogram.types import CallbackQuery, InlineKeyboardButton, InlineKeyboardMarkup
from config import ( from config import (
BOT_NAME,
BOT_USERNAME, BOT_USERNAME,
GROUP_SUPPORT, GROUP_SUPPORT,
OWNER_USERNAME, OWNER_USERNAME,
@ -21,6 +20,7 @@ from config import (
@Client.on_callback_query(filters.regex("home_start")) @Client.on_callback_query(filters.regex("home_start"))
async def set_start(_, query: CallbackQuery): async def set_start(_, query: CallbackQuery):
user_id = query.from_user.id user_id = query.from_user.id
BOT_NAME = (await bot.get_me()).first_name
if await is_gbanned_user(user_id): if await is_gbanned_user(user_id):
await query.answer("❗️ You've blocked from using this bot!", show_alert=True) await query.answer("❗️ You've blocked from using this bot!", show_alert=True)
return return