mirror of
https://github.com/lifegpc/pythonscript.git
synced 2026-06-06 11:28:58 +08:00
91 lines
3.7 KiB
Python
91 lines
3.7 KiB
Python
from argparse import ArgumentParser
|
|
import asyncio
|
|
from json import load
|
|
import sys
|
|
from traceback import print_exc
|
|
from tdlib import TdLib
|
|
from util import tparse
|
|
|
|
|
|
async def get_chat_id_from_name(lib: TdLib, name: str) -> int:
|
|
re = await lib.searchChatsOnServer(name)
|
|
if re is None:
|
|
raise ValueError('Can not found chat.')
|
|
le = len(re['chat_ids'])
|
|
if le == 0:
|
|
re = await lib.searchPublicChats(name)
|
|
if re is None:
|
|
raise ValueError('Can not search public chats.')
|
|
le = len(re['chat_ids'])
|
|
if le == 0:
|
|
raise ValueError('No chat found.')
|
|
if le == 1:
|
|
return re['chat_ids'][0]
|
|
else:
|
|
k = 1
|
|
for i in re['chat_ids']:
|
|
c = await lib.getChat(i)
|
|
if c is None:
|
|
raise ValueError('Chat not found.')
|
|
print(f"{k:02}. Chat ID: \t{i}")
|
|
print(f" Chat Title: {c['title']}")
|
|
print(f" Chat Type: \t{c['type']}")
|
|
k += 1
|
|
while True:
|
|
inp = input('Please select one chat: ')
|
|
try:
|
|
i = int(inp)
|
|
if i >= 1 and i <= le:
|
|
return re['chat_ids'][i - 1]
|
|
if i in re['chat_ids']:
|
|
return i
|
|
except ValueError:
|
|
pass
|
|
|
|
|
|
async def main(lib: TdLib, arg):
|
|
with open(arg.config, 'r', encoding='UTF-8') as f:
|
|
se = load(f)
|
|
if not await lib.login(se['TdlibParameters'], se['encryption_key'],
|
|
se.get("proxy"), se.get("phone_number")):
|
|
raise ValueError('Can not login')
|
|
chat_id = arg.chat_id if arg.chat_id is not None else await get_chat_id_from_name(lib, arg.chat_name) # noqa: E501
|
|
chat = await lib.getChat(chat_id)
|
|
if chat is None:
|
|
print('Chat not found.')
|
|
return -1
|
|
print('Chat information:')
|
|
print(f"Chat ID: {chat['id']}")
|
|
print(f"Chat Title: {chat['title']}")
|
|
print(f"Chat Type: {chat['type']}")
|
|
yes = False
|
|
if not yes:
|
|
inp = input('Do you want to delete messages in this chat?(y/n)')
|
|
if inp[0].lower() == 'y':
|
|
yes = True
|
|
if not yes:
|
|
return 0
|
|
re = await lib.deleteAllMyMessageInChat(chat_id, arg.start_time, arg.end_time) # noqa: E501
|
|
print(re)
|
|
return 0 if re is not None else -1
|
|
|
|
|
|
p = ArgumentParser(description='Delete all my messages in a chat.', add_help=True) # noqa: E501
|
|
p.add_argument('-c', '--config', default='tdlib.json', metavar='CONFIG', dest='config', help='Specify the location of config file. Default: tdlib.json') # noqa: E501
|
|
p.add_argument('chat_id', nargs='?', type=int, help="Specify the chat's ID.")
|
|
p.add_argument('-n', '--chat-name', help="Specify chat's name. Will used if chat_id is not sepcified.", metavar='NAME', dest='chat_name') # noqa: E501
|
|
p.add_argument('-s', '--start-time', type=tparse, metavar='TIME', help="The messages which are sended before this time will not be deleted.", dest='start_time') # noqa: E501
|
|
p.add_argument('-e', '--end-time', type=tparse, metavar='TIME', help="The messages which are sended after this time will not be deleted.", dest="end_time") # noqa: E501
|
|
p.add_argument('-v', '--verbose', action='store_true', help="Enable verbose logging.", dest="verbose") # noqa: E501
|
|
p.add_argument('--max-cache', metavar='COUNT', type=int, help='Specify the max count of cache messages. At least 100.', dest="max_cache") # noqa: E501
|
|
arg = p.parse_intermixed_args()
|
|
if arg.chat_id is None and arg.chat_name is None:
|
|
raise ValueError('chat_id or chat_name is needed.')
|
|
try:
|
|
with TdLib(arg.verbose, arg.max_cache) as lib:
|
|
re = asyncio.run(main(lib, arg))
|
|
sys.exit(re)
|
|
except Exception:
|
|
print_exc()
|
|
sys.exit(-1)
|