mirror of
https://github.com/whyour/qinglong.git
synced 2025-07-30 08:26:08 +08:00
259 lines
9.7 KiB
Python
259 lines
9.7 KiB
Python
import os
|
||
from telethon import events, Button
|
||
import re
|
||
from .. import jdbot, chat_id, _LogDir, logger, _JdDir, _OwnDir, _ConfigDir
|
||
import asyncio
|
||
import datetime
|
||
|
||
bean_log = _LogDir + '/jd_bean_change/'
|
||
_ConfigFile = _ConfigDir+'/config.sh'
|
||
V4, QL = False, False
|
||
if 'JD_DIR' in os.environ.keys():
|
||
V4 = True
|
||
_ConfigFile = _ConfigDir+'/config.sh'
|
||
_DiyDir = _OwnDir
|
||
jdcmd = 'jtask'
|
||
elif 'QL_DIR' in os.environ.keys():
|
||
QL = True
|
||
_ConfigFile = _ConfigDir+'/cookie.sh'
|
||
_DiyDir = None
|
||
jdcmd = 'task'
|
||
dirs = os.listdir(_LogDir)
|
||
for mydir in dirs:
|
||
if 'jd_bean_change' in mydir:
|
||
bean_log = _LogDir + '/' + mydir
|
||
break
|
||
else:
|
||
_DiyDir = None
|
||
jdcmd = 'node'
|
||
|
||
ckreg = re.compile(r'pt_key=\S*;pt_pin=\S*;')
|
||
with open(_ConfigFile, 'r', encoding='utf-8') as f:
|
||
lines = f.read()
|
||
cookies = ckreg.findall(lines)
|
||
for ck in cookies:
|
||
if ck == 'pt_key=xxxxxxxxxx;pt_pin=xxxx;':
|
||
cookies.remove(ck)
|
||
break
|
||
|
||
|
||
def split_list(datas, n, row: bool = True):
|
||
"""一维列表转二维列表，根据N不同，生成不同级别的列表"""
|
||
length = len(datas)
|
||
size = length / n + 1 if length % n else length/n
|
||
_datas = []
|
||
if not row:
|
||
size, n = n, size
|
||
for i in range(int(size)):
|
||
start = int(i * n)
|
||
end = int((i + 1) * n)
|
||
_datas.append(datas[start:end])
|
||
return _datas
|
||
|
||
|
||
async def backfile(file):
|
||
'''如果文件存在，则备份，并更新'''
|
||
if os.path.exists(file):
|
||
try:
|
||
os.rename(file, file+'.bak')
|
||
except WindowsError:
|
||
os.remove(file+'.bak')
|
||
os.rename(file, file+'.bak')
|
||
|
||
|
||
def press_event(user_id):
|
||
return events.CallbackQuery(func=lambda e: e.sender_id == user_id)
|
||
|
||
|
||
async def cmd(cmdtext):
|
||
'''定义执行cmd命令'''
|
||
try:
|
||
msg = await jdbot.send_message(chat_id, '开始执行命令')
|
||
p = await asyncio.create_subprocess_shell(
|
||
cmdtext, stdout=asyncio.subprocess.PIPE, stderr=asyncio.subprocess.PIPE)
|
||
res_bytes, res_err = await p.communicate()
|
||
res = res_bytes.decode('utf-8')
|
||
if len(res) == 0:
|
||
await jdbot.edit_message(msg, '已执行，但返回值为空')
|
||
elif len(res) <= 4000:
|
||
await jdbot.delete_messages(chat_id, msg)
|
||
await jdbot.send_message(chat_id, res)
|
||
elif len(res) > 4000:
|
||
_log = _LogDir + '/bot/'+cmdtext.split('/')[-1].split(
|
||
'.js')[0]+datetime.datetime.now().strftime('%H-%M-%S')+'.log'
|
||
with open(_log, 'w+', encoding='utf-8') as f:
|
||
f.write(res)
|
||
await jdbot.delete_messages(chat_id, msg)
|
||
await jdbot.send_message(chat_id, '执行结果较长，请查看日志', file=_log)
|
||
os.remove(_log)
|
||
except Exception as e:
|
||
await jdbot.send_message(chat_id, 'something wrong,I\'m sorry\n'+str(e))
|
||
logger.error('something wrong,I\'m sorry'+str(e))
|
||
|
||
|
||
async def getname(path, dir):
|
||
'''获取文件中文名称，如无则返回文件名'''
|
||
names = []
|
||
reg = r'new Env\(\'[\S]+?\'\)'
|
||
cname = False
|
||
for file in dir:
|
||
if os.path.isdir(path+'/'+file):
|
||
names.append(file)
|
||
elif file.endswith('.js') and file != 'jdCookie.js' and file != 'getJDCookie.js' and file != 'JD_extra_cookie.js' and 'ShareCode' not in file:
|
||
with open(path+'/'+file, 'r', encoding='utf-8') as f:
|
||
resdatas = f.readlines()
|
||
for data in resdatas:
|
||
if 'new Env' in data:
|
||
data = data.replace('\"', '\'')
|
||
res = re.findall(reg, data)
|
||
if len(res) != 0:
|
||
res = res[0].split('\'')[-2]
|
||
names.append(res+'--->'+file)
|
||
cname = True
|
||
break
|
||
if not cname:
|
||
names.append(file+'--->'+file)
|
||
cname = False
|
||
else:
|
||
continue
|
||
return names
|
||
|
||
|
||
async def logbtn(conv, SENDER, path, msg, page, filelist):
|
||
'''定义log日志按钮'''
|
||
mybtn = [Button.inline('上一页', data='up'), Button.inline(
|
||
'下一页', data='next'), Button.inline('上级', data='updir'), Button.inline('取消', data='cancel')]
|
||
try:
|
||
if filelist:
|
||
markup = filelist
|
||
newmarkup = markup[page]
|
||
if mybtn not in newmarkup:
|
||
newmarkup.append(mybtn)
|
||
else:
|
||
dir = os.listdir(path)
|
||
dir.sort()
|
||
markup = [Button.inline(file, data=str(file))
|
||
for file in dir]
|
||
markup = split_list(markup, 3)
|
||
if len(markup) > 30:
|
||
markup = split_list(markup, 30)
|
||
newmarkup = markup[page]
|
||
newmarkup.append(mybtn)
|
||
else:
|
||
newmarkup = markup
|
||
if path == _JdDir:
|
||
newmarkup.append([Button.inline('取消', data='cancel')])
|
||
else:
|
||
newmarkup.append(
|
||
[Button.inline('上级', data='updir'), Button.inline('取消', data='cancel')])
|
||
msg = await jdbot.edit_message(msg, '请做出您的选择：', buttons=newmarkup)
|
||
convdata = await conv.wait_event(press_event(SENDER))
|
||
res = bytes.decode(convdata.data)
|
||
if res == 'cancel':
|
||
msg = await jdbot.edit_message(msg, '对话已取消')
|
||
conv.cancel()
|
||
return None, None, None, None
|
||
elif res == 'next':
|
||
page = page + 1
|
||
if page > len(markup) - 1:
|
||
page = 0
|
||
return path, msg, page, markup
|
||
elif res == 'up':
|
||
page = page - 1
|
||
if page < 0:
|
||
page = len(markup) - 1
|
||
return path, msg, page, markup
|
||
elif res == 'updir':
|
||
path = '/'.join(path.split('/')[:-1])
|
||
logger.info(path)
|
||
if path == '':
|
||
path = _JdDir
|
||
return path, msg, page, None
|
||
elif os.path.isfile(path+'/'+res):
|
||
msg = await jdbot.edit_message(msg, '文件发送中，请注意查收')
|
||
await conv.send_file(path+'/'+res)
|
||
msg = await jdbot.edit_message(msg, res+'发送成功，请查收')
|
||
conv.cancel()
|
||
return None, None, None, None
|
||
else:
|
||
return path+'/'+res, msg, page, None
|
||
except asyncio.exceptions.TimeoutError:
|
||
msg = await jdbot.edit_message(msg, '选择已超时，本次对话已停止')
|
||
return None, None, None, None
|
||
except Exception as e:
|
||
msg = await jdbot.edit_message(msg, 'something wrong,I\'m sorry\n'+str(e))
|
||
logger.error('something wrong,I\'m sorry\n'+str(e))
|
||
return None, None, None, None
|
||
|
||
|
||
async def nodebtn(conv, SENDER, path, msg, page, filelist):
|
||
'''定义scripts脚本按钮'''
|
||
mybtn = [Button.inline('上一页', data='up'), Button.inline(
|
||
'下一页', data='next'), Button.inline('上级', data='updir'), Button.inline('取消', data='cancel')]
|
||
try:
|
||
if filelist:
|
||
markup = filelist
|
||
newmarkup = markup[page]
|
||
if mybtn not in newmarkup:
|
||
newmarkup.append(mybtn)
|
||
else:
|
||
if path == _JdDir and V4:
|
||
dir = ['scripts', _OwnDir.split('/')[-1]]
|
||
elif path == _JdDir and QL:
|
||
dir = ['scripts']
|
||
else:
|
||
dir = os.listdir(path)
|
||
dir = await getname(path, dir)
|
||
dir.sort()
|
||
markup = [Button.inline(file.split('--->')[0], data=str(file.split('--->')[-1]))
|
||
for file in dir if os.path.isdir(path+'/'+file) or file.endswith('.js')]
|
||
markup = split_list(markup, 3)
|
||
if len(markup) > 30:
|
||
markup = split_list(markup, 30)
|
||
newmarkup = markup[page]
|
||
newmarkup.append(mybtn)
|
||
else:
|
||
newmarkup = markup
|
||
if path == _JdDir:
|
||
newmarkup.append([Button.inline('取消', data='cancel')])
|
||
else:
|
||
newmarkup.append(
|
||
[Button.inline('上级', data='updir'), Button.inline('取消', data='cancel')])
|
||
msg = await jdbot.edit_message(msg, '请做出您的选择：', buttons=newmarkup)
|
||
convdata = await conv.wait_event(press_event(SENDER))
|
||
res = bytes.decode(convdata.data)
|
||
if res == 'cancel':
|
||
msg = await jdbot.edit_message(msg, '对话已取消')
|
||
conv.cancel()
|
||
return None, None, None, None
|
||
elif res == 'next':
|
||
page = page + 1
|
||
if page > len(markup) - 1:
|
||
page = 0
|
||
return path, msg, page, markup
|
||
elif res == 'up':
|
||
page = page - 1
|
||
if page < 0:
|
||
page = len(markup) - 1
|
||
return path, msg, page, markup
|
||
elif res == 'updir':
|
||
path = '/'.join(path.split('/')[:-1])
|
||
if path == '':
|
||
path = _JdDir
|
||
return path, msg, page, None
|
||
elif os.path.isfile(path+'/'+res):
|
||
conv.cancel()
|
||
logger.info(path+'/'+res+'脚本即将在后台运行')
|
||
msg = await jdbot.edit_message(msg, res + '在后台运行成功')
|
||
cmdtext = '{} {}/{} now'.format(jdcmd, path, res)
|
||
return None, None, None, 'CMD-->'+cmdtext
|
||
else:
|
||
return path+'/'+res, msg, page, None
|
||
except asyncio.exceptions.TimeoutError:
|
||
msg = await jdbot.edit_message(msg, '选择已超时，对话已停止')
|
||
return None, None, None, None
|
||
except Exception as e:
|
||
msg = await jdbot.edit_message(msg, 'something wrong,I\'m sorry\n'+str(e))
|
||
logger.error('something wrong,I\'m sorry\n'+str(e))
|
||
return None, None, None, None
|