def gban_check(client, message):
gbanned = sql.is_gbanned(message.from_user.id)
if gbanned:
try:
user_id = message.from_user.id
chat_id = message.chat.id
client.kick_chat_member(chat_id, user_id)
except BaseException as e:
send_log(get_translation('banError', ['`', '**', e]))
message.continue_propagation()
def gban_user(client, message):
args = extract_args(message)
reply = message.reply_to_message
edit(message, f'`{get_translation("banProcess")}`')
if args:
try:
user = client.get_users(args)
except Exception:
edit(message, f'`{get_translation("banFailUser")}`')
return
elif reply:
user_id = reply.from_user.id
user = client.get_users(user_id)
user_id = user.id
else:
edit(message, f'`{get_translation("banFailUser")}`')
return
try:
replied_user = reply.from_user
if replied_user.is_self:
return edit(message, f'`{get_translation("cannotBanMyself")}`')
except BaseException:
pass
if user.id in BRAIN:
return edit(
message,
get_translation('brainError',
['`', '**', user.first_name, user.id]))
try:
if sql.is_gbanned(user.id):
return edit(message, f'`{get_translation("alreadyBanned")}`')
chat_id = message.chat.id
sql.gban(user.id)
edit(
message,
get_translation('gbanResult',
['**', user.first_name, user.id, '`']))
sleep(1)
send_log(
get_translation('gbanLog', ['**', user.first_name, user.id, '`']))
except Exception as e:
edit(message, get_translation('banError', ['`', '**', e]))
return
def ungban_user(client, message):
args = extract_args(message)
reply = message.reply_to_message
edit(message, f'`{get_translation("unbanProcess")}`')
if args:
try:
user = client.get_users(args)
except Exception:
edit(message, f'`{get_translation("banFailUser")}`')
return
elif reply:
user_id = reply.from_user.id
user = client.get_users(user_id)
user_id = user.id
else:
edit(message, f'`{get_translation("banFailUser")}`')
return
try:
replied_user = reply.from_user
if replied_user.is_self:
return edit(message, f'`{get_translation("cannotUnbanMyself")}`')
except BaseException:
pass
try:
if not sql.is_gbanned(user.id):
return edit(message, f'`{get_translation("alreadyUnbanned")}`')
chat_id = message.chat.id
sql.ungban(user.id)
client.unban_chat_member(chat_id, user.id)
edit(
message,
get_translation('unbanResult',
['**', user.first_name, user.id, '`']))
except Exception as e:
edit(message, get_translation('banError', ['`', '**', e]))
return