This commit closes #13
This commit is contained in:
@@ -1,18 +1,21 @@
|
||||
from pyrogram.types import Message
|
||||
|
||||
from classes.pyroclient import PyroClient
|
||||
from classes.pyrogroup import PyroGroup
|
||||
|
||||
|
||||
async def kick_unstarted(
|
||||
app: PyroClient, user_id: int, group_id: int, message_id: int
|
||||
) -> None:
|
||||
user = await app.find_user(user_id, group_id)
|
||||
group = await PyroGroup.create_if_not_exists(group_id)
|
||||
|
||||
if user.score == 0 and user.failed == 0:
|
||||
banned = await app.ban_chat_member(group_id, user_id)
|
||||
if group.ban_failed:
|
||||
banned = await app.ban_chat_member(group_id, user_id)
|
||||
|
||||
if isinstance(banned, Message):
|
||||
await banned.delete()
|
||||
if isinstance(banned, Message):
|
||||
await banned.delete()
|
||||
|
||||
await app.delete_messages(group_id, message_id)
|
||||
|
||||
@@ -21,11 +24,13 @@ async def kick_unverified(
|
||||
app: PyroClient, user_id: int, group_id: int, message_id: int
|
||||
) -> None:
|
||||
user = await app.find_user(user_id, group_id)
|
||||
group = await PyroGroup.create_if_not_exists(group_id)
|
||||
|
||||
if user.score < 6 or user.failed:
|
||||
banned = await app.ban_chat_member(group_id, user_id)
|
||||
if group.ban_failed:
|
||||
banned = await app.ban_chat_member(group_id, user_id)
|
||||
|
||||
if isinstance(banned, Message):
|
||||
await banned.delete()
|
||||
if isinstance(banned, Message):
|
||||
await banned.delete()
|
||||
|
||||
await app.delete_messages(group_id, message_id)
|
||||
|
Reference in New Issue
Block a user