TeamPGM/PagerMaid-Modify
3
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

✨sb support delete all messages. (#134)

Browse Source 
✨ sb 命令支持默认删除所有消息。

Co-authored-by: xtaodada <xtao@xtaolink.cn>
This commit is contained in:
vesugier 2021-09-26 23:00:58 +08:00 committed by GitHub
parent d2f224d9cc
commit ae0642fd16
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
pagermaid/modules/sb.py
View File

@ -4,6 +4,7 @@ from pagermaid.utils import lang, alias_command
from struct import error as StructError from struct import error as StructError
from telethon.tl.functions.messages import GetCommonChatsRequest from telethon.tl.functions.messages import GetCommonChatsRequest
from telethon.tl.functions.users import GetFullUserRequest from telethon.tl.functions.users import GetFullUserRequest
from telethon.tl.functions.channels import DeleteUserHistoryRequest
from telethon.tl.types import MessageEntityMentionName, ChannelParticipantsAdmins from telethon.tl.types import MessageEntityMentionName, ChannelParticipantsAdmins
from telethon.errors.rpcerrorlist import UserAdminInvalidError, ChatAdminRequiredError, FloodWaitError from telethon.errors.rpcerrorlist import UserAdminInvalidError, ChatAdminRequiredError, FloodWaitError
from asyncio import sleep from asyncio import sleep
@ -34,7 +35,7 @@ def mention_group(chat):
@listener(is_plugin=False, outgoing=True, command=alias_command("sb"), @listener(is_plugin=False, outgoing=True, command=alias_command("sb"),
description=lang('sb_des'), description=lang('sb_des'),
parameters="<reply|id|username>") parameters="<reply|id|username> <do_not_del_all>")
async def span_ban(context): async def span_ban(context):
if context.reply_to_msg_id: if context.reply_to_msg_id:
reply_message = await context.get_reply_message() reply_message = await context.get_reply_message()
@ -79,6 +80,9 @@ async def span_ban(context):
await context.edit(f"{lang('error_prefix')}{lang('profile_e_long')}") await context.edit(f"{lang('error_prefix')}{lang('profile_e_long')}")
return return
raise exception raise exception
chat = await context.get_chat()
if len(context.parameter) == 0:
await context.client(DeleteUserHistoryRequest(channel=chat, user_id=target_user))
myself = await context.client.get_me() myself = await context.client.get_me()
self_user_id = myself.id self_user_id = myself.id
if target_user.user.id == self_user_id: if target_user.user.id == self_user_id:
@ -89,6 +93,9 @@ async def span_ban(context):
groups = [] groups = []
for i in result.chats: for i in result.chats:
try: try:
chat = await context.client.get_entity(i.id)
if len(context.parameter) == 0:
await context.client(DeleteUserHistoryRequest(channel=chat, user_id=target_user))
await context.client.edit_permissions(i.id, target_user, view_messages=False) await context.client.edit_permissions(i.id, target_user, view_messages=False)
groups.append(mention_group(i)) groups.append(mention_group(i))
count += 1 count += 1
@ -120,6 +127,8 @@ async def span_ban(context):
for i in sb_groups: for i in sb_groups:
try: try:
chat = await context.client.get_entity(int(i)) chat = await context.client.get_entity(int(i))
if len(context.parameter) == 0:
await context.client(DeleteUserHistoryRequest(channel=chat, user_id=target_user))
await context.client.edit_permissions(chat, target_user, view_messages=False) await context.client.edit_permissions(chat, target_user, view_messages=False)
groups.append(mention_group(chat)) groups.append(mention_group(chat))
count += 1 count += 1