mirror of
https://github.com/whyour/qinglong.git
synced 2025-07-30 00:16:07 +08:00
58 lines
2.7 KiB
Python
58 lines
2.7 KiB
Python
from telethon import events, Button
|
|
from .utils import split_list, press_event, cmd
|
|
import subprocess
|
|
from asyncio import exceptions
|
|
from .. import jdbot, chat_id, _shortcut, logger
|
|
|
|
|
|
@jdbot.on(events.NewMessage(from_users=chat_id, pattern=r'^/a$'))
|
|
async def shortcut(event):
|
|
markup = []
|
|
SENDER = event.sender_id
|
|
msg = await jdbot.send_message(chat_id, '正在查询您的常用命令，请稍后')
|
|
with open(_shortcut, 'r', encoding='utf-8') as f:
|
|
shortcuts = f.readlines()
|
|
try:
|
|
async with jdbot.conversation(SENDER, timeout=60) as conv:
|
|
markup = [Button.inline(shortcut.split(
|
|
'-->')[0], data=str(shortcut.split('-->')[-1])) for shortcut in shortcuts if '-->' in shortcut]
|
|
markup = split_list(markup, 3)
|
|
markup.append([Button.inline('取消', data='cancel')])
|
|
msg = await jdbot.edit_message(msg, '请做出您的选择：', buttons=markup)
|
|
convdata = await conv.wait_event(press_event(SENDER))
|
|
res = bytes.decode(convdata.data)
|
|
if res == 'cancel':
|
|
msg = await jdbot.edit_message(msg, '对话已取消')
|
|
conv.cancel()
|
|
elif 'nohup ' in res:
|
|
msg = await jdbot.edit_message(msg, '即将执行您的操作'+res)
|
|
cmdtext = res.replace('nohup ', '')
|
|
subprocess.Popen(
|
|
cmdtext, shell=True, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
|
|
msg = await jdbot.edit_message(msg, '已在后台执行您的操作'+res.replace('nohup ', ''))
|
|
conv.cancel()
|
|
else:
|
|
await jdbot.delete_messages(chat_id, msg)
|
|
await cmd(res)
|
|
conv.cancel()
|
|
except exceptions.TimeoutError:
|
|
msg = await jdbot.edit_message(msg, '选择已超时，对话已停止')
|
|
except Exception as e:
|
|
await jdbot.edit_message(msg, 'something wrong,I\'m sorry\n'+str(e))
|
|
logger.error('something wrong,I\'m sorry\n'+str(e))
|
|
|
|
@jdbot.on(events.NewMessage(from_users=chat_id, pattern=r'^/b$'))
|
|
async def shortcut(event):
|
|
markup = []
|
|
msg = await jdbot.send_message(chat_id, '正在查询您的常用命令，请稍后')
|
|
with open(_shortcut, 'r', encoding='utf-8') as f:
|
|
shortcuts = f.readlines()
|
|
try:
|
|
await jdbot.delete_messages(chat_id,msg)
|
|
markup = [Button.text(shortcut,single_use=True) for shortcut in shortcuts if '-->' not in shortcut]
|
|
markup = split_list(markup, 3)
|
|
await jdbot.send_message(chat_id, '请做出您的选择：', buttons=markup)
|
|
except Exception as e:
|
|
await jdbot.edit_message(msg, 'something wrong,I\'m sorry\n'+str(e))
|
|
logger.error('something wrong,I\'m sorry\n'+str(e))
|