mirror of
https://github.com/lan-tianxiang/JS_TOOL.git
synced 2025-07-14 00:42:35 +08:00
470 lines
20 KiB
Python
Executable File
470 lines
20 KiB
Python
Executable File
#!/usr/bin/env python3
|
||
# _*_ coding:utf-8 _*_
|
||
|
||
# author: https://github.com/SuMaiKaDe
|
||
|
||
from telethon import TelegramClient, events, Button
|
||
import requests
|
||
import re
|
||
import json
|
||
import time
|
||
import os
|
||
import qrcode
|
||
import logging
|
||
from asyncio import exceptions
|
||
logging.basicConfig(
|
||
format='%(asctime)s-%(name)s-%(levelname)s=> [%(funcName)s] %(message)s ', level=logging.INFO)
|
||
logger = logging.getLogger(__name__)
|
||
_JdDir = '/root/jd'
|
||
_ConfigDir = _JdDir + '/config'
|
||
_ScriptsDir = _JdDir + '/scripts'
|
||
_LogDir = _JdDir + '/log'
|
||
_ThirdpardDir = _JdDir +'/thirdpard'
|
||
# 频道id/用户id
|
||
with open('/root/jd/config/bot.json') as f:
|
||
bot = json.load(f)
|
||
chat_id = bot['user_id']
|
||
# 机器人 TOKEN
|
||
TOKEN = bot['bot_token']
|
||
# 发消息的TG代理
|
||
# my.telegram.org申请到的api_id,api_hash
|
||
api_id = bot['api_id']
|
||
api_hash = bot['api_hash']
|
||
proxystart = bot['proxy']
|
||
proxy = (bot['proxy_type'], bot['proxy_add'], bot['proxy_port'])
|
||
# 开启tg对话
|
||
if proxystart:
|
||
client = TelegramClient('bot', api_id, api_hash,proxy=proxy).start(bot_token=TOKEN)
|
||
else:
|
||
client = TelegramClient('bot', api_id, api_hash).start(bot_token=TOKEN)
|
||
cookiemsg =''
|
||
img_file = '/jd/config/qr.jpg'
|
||
StartCMD = bot['StartCMD']
|
||
def press_event(user_id):
|
||
return events.CallbackQuery(func=lambda e: e.sender_id == user_id)
|
||
|
||
# 扫码获取cookie 直接采用LOF大佬代码
|
||
# getSToken请求获取，s_token用于发送post请求是的必须参数
|
||
s_token = ""
|
||
# getSToken请求获取，guid,lsid,lstoken用于组装cookies
|
||
guid, lsid, lstoken = "", "", ""
|
||
# 由上面参数组装生成，getOKLToken函数发送请求需要使用
|
||
cookies = ""
|
||
# getOKLToken请求获取，token用户生成二维码使用、okl_token用户检查扫码登录结果使用
|
||
token, okl_token = "", ""
|
||
# 最终获取到的可用的cookie
|
||
jd_cookie = ""
|
||
|
||
|
||
def getSToken():
|
||
time_stamp = int(time.time() * 1000)
|
||
get_url = 'https://plogin.m.jd.com/cgi-bin/mm/new_login_entrance?lang=chs&appid=300&returnurl=https://wq.jd.com/passport/LoginRedirect?state=%s&returnurl=https://home.m.jd.com/myJd/newhome.action?sceneval=2&ufc=&/myJd/home.action&source=wq_passport' % time_stamp
|
||
get_header = {
|
||
'Connection': 'Keep-Alive',
|
||
'Content-Type': 'application/x-www-form-urlencoded',
|
||
'Accept': 'application/json, text/plain, */*',
|
||
'Accept-Language': 'zh-cn',
|
||
'Referer': 'https://plogin.m.jd.com/login/login?appid=300&returnurl=https://wq.jd.com/passport/LoginRedirect?state=%s&returnurl=https://home.m.jd.com/myJd/newhome.action?sceneval=2&ufc=&/myJd/home.action&source=wq_passport' % time_stamp,
|
||
'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/86.0.4240.111 Safari/537.36',
|
||
'Host': 'plogin.m.jd.com'
|
||
}
|
||
try:
|
||
resp = requests.get(url=get_url, headers=get_header)
|
||
parseGetRespCookie(resp.headers, resp.json())
|
||
logger.info(resp.headers)
|
||
logger.info(resp.json())
|
||
except Exception as error:
|
||
logger.exception("Get网络请求异常", error)
|
||
|
||
|
||
def parseGetRespCookie(headers, get_resp):
|
||
global s_token
|
||
global cookies
|
||
s_token = get_resp.get('s_token')
|
||
set_cookies = headers.get('set-cookie')
|
||
logger.info(set_cookies)
|
||
|
||
guid = re.findall(r"guid=(.+?);", set_cookies)[0]
|
||
lsid = re.findall(r"lsid=(.+?);", set_cookies)[0]
|
||
lstoken = re.findall(r"lstoken=(.+?);", set_cookies)[0]
|
||
|
||
cookies = f"guid={guid}; lang=chs; lsid={lsid}; lstoken={lstoken}; "
|
||
logger.info(cookies)
|
||
|
||
|
||
def getOKLToken():
|
||
post_time_stamp = int(time.time() * 1000)
|
||
post_url = 'https://plogin.m.jd.com/cgi-bin/m/tmauthreflogurl?s_token=%s&v=%s&remember=true' % (
|
||
s_token, post_time_stamp)
|
||
post_data = {
|
||
'lang': 'chs',
|
||
'appid': 300,
|
||
'returnurl': 'https://wqlogin2.jd.com/passport/LoginRedirect?state=%s&returnurl=//home.m.jd.com/myJd/newhome.action?sceneval=2&ufc=&/myJd/home.action' % post_time_stamp,
|
||
'source': 'wq_passport'
|
||
}
|
||
post_header = {
|
||
'Connection': 'Keep-Alive',
|
||
'Content-Type': 'application/x-www-form-urlencoded; Charset=UTF-8',
|
||
'Accept': 'application/json, text/plain, */*',
|
||
'Cookie': cookies,
|
||
'Referer': 'https://plogin.m.jd.com/login/login?appid=300&returnurl=https://wqlogin2.jd.com/passport/LoginRedirect?state=%s&returnurl=//home.m.jd.com/myJd/newhome.action?sceneval=2&ufc=&/myJd/home.action&source=wq_passport' % post_time_stamp,
|
||
'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/86.0.4240.111 Safari/537.36',
|
||
'Host': 'plogin.m.jd.com',
|
||
}
|
||
try:
|
||
global okl_token
|
||
resp = requests.post(
|
||
url=post_url, headers=post_header, data=post_data, timeout=20)
|
||
parsePostRespCookie(resp.headers, resp.json())
|
||
logger.info(resp.headers)
|
||
except Exception as error:
|
||
logger.exception("Post网络请求错误", error)
|
||
|
||
|
||
def parsePostRespCookie(headers, data):
|
||
global token
|
||
global okl_token
|
||
|
||
token = data.get('token')
|
||
okl_token = re.findall(r"okl_token=(.+?);", headers.get('set-cookie'))[0]
|
||
|
||
logger.info("token:" + token)
|
||
logger.info("okl_token:" + okl_token)
|
||
|
||
|
||
def chekLogin():
|
||
|
||
global login
|
||
|
||
|
||
|
||
|
||
def parseJDCookies(headers):
|
||
global jd_cookie
|
||
logger.info("扫码登录成功，下面为获取到的用户Cookie。")
|
||
set_cookie = headers.get('Set-Cookie')
|
||
pt_key = re.findall(r"pt_key=(.+?);", set_cookie)[0]
|
||
pt_pin = re.findall(r"pt_pin=(.+?);", set_cookie)[0]
|
||
logger.info(pt_key)
|
||
logger.info(pt_pin)
|
||
jd_cookie = f'pt_key={pt_key};pt_pin={pt_pin};'
|
||
|
||
|
||
def creatqr(text):
|
||
'''实例化QRCode生成qr对象'''
|
||
qr = qrcode.QRCode(
|
||
version=1,
|
||
error_correction=qrcode.constants.ERROR_CORRECT_H,
|
||
box_size=10,
|
||
border=4
|
||
)
|
||
qr.clear()
|
||
# 传入数据
|
||
qr.add_data(text)
|
||
qr.make(fit=True)
|
||
# 生成二维码
|
||
img = qr.make_image()
|
||
# 保存二维码
|
||
img.save(img_file)
|
||
|
||
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 logbtn(conv, SENDER, path: str, content: str, msg):
|
||
'''定义log日志按钮'''
|
||
try:
|
||
dir = os.listdir(path)
|
||
dir.sort()
|
||
markup = [Button.inline(file, data=str(path+'/'+file))
|
||
for file in dir]
|
||
markup.append(Button.inline('取消', data='cancle'))
|
||
markup = split_list(markup, 3)
|
||
msg = await client.edit_message(msg, '请做出你的选择：', buttons=markup)
|
||
convdata = await conv.wait_event(press_event(SENDER))
|
||
res = bytes.decode(convdata.data)
|
||
if res == 'cancle':
|
||
msg = await client.edit_message(msg, '对话已取消')
|
||
conv.cancel()
|
||
return None, None
|
||
elif os.path.isfile(res):
|
||
msg = await client.edit_message(msg, content + '中，请注意查收')
|
||
await conv.send_file(res)
|
||
msg = await client.edit_message(msg, res+ content + '成功，请查收')
|
||
conv.cancel()
|
||
return None, None
|
||
else:
|
||
return res, msg
|
||
except exceptions.TimeoutError:
|
||
msg = await client.edit_message(msg, '选择已超时，对话已停止')
|
||
return None, None
|
||
except Exception as e:
|
||
msg = await client.edit_message(msg, 'something wrong,I\'m sorry\n'+str(e))
|
||
logger.error('something wrong,I\'m sorry\n'+str(e))
|
||
return None, None
|
||
|
||
|
||
async def nodebtn(conv, SENDER, path: str, msg):
|
||
'''定义scripts脚本按钮'''
|
||
try:
|
||
if path == '/jd':
|
||
dir = ['scripts', 'thirdpard']
|
||
else:
|
||
dir = os.listdir(path)
|
||
dir.sort()
|
||
markup = [Button.inline(file, data=str(path+'/'+file))
|
||
for file in dir if os.path.isdir(path+'/'+file) or re.search(r'.js$', file)]
|
||
markup.append(Button.inline('取消', data='cancel'))
|
||
markup = split_list(markup, 3)
|
||
msg = await client.edit_message(msg, '请做出你的选择：', buttons=markup)
|
||
convdata = await conv.wait_event(press_event(SENDER))
|
||
res = bytes.decode(convdata.data)
|
||
if res == 'cancel':
|
||
msg = await client.edit_message(msg, '对话已取消')
|
||
conv.cancel()
|
||
return None, None
|
||
elif os.path.isfile(res):
|
||
msg = await client.edit_message(msg, '脚本即将在后台运行')
|
||
logger.info(res+'脚本即将在后台运行')
|
||
os.popen('/jd/jd.sh {} now >/jd/log/bot.log &'.format(res))
|
||
msg = await client.edit_message(msg, res + '在后台运行成功，请自行在程序结束后查看日志')
|
||
conv.cancel()
|
||
return None, None
|
||
else:
|
||
return res, msg
|
||
except exceptions.TimeoutError:
|
||
msg = await client.edit_message(msg, '选择已超时，对话已停止')
|
||
return None, None
|
||
except Exception as e:
|
||
msg = await client.edit_message(msg, 'something wrong,I\'m sorry\n'+str(e))
|
||
logger.error('something wrong,I\'m sorry\n'+str(e))
|
||
return None, None
|
||
|
||
|
||
@client.on(events.NewMessage(from_users=chat_id, pattern=r'^/log'))
|
||
async def mylog(event):
|
||
'''定义日志文件操作'''
|
||
SENDER = event.sender_id
|
||
path = _LogDir
|
||
async with client.conversation(SENDER, timeout=60) as conv:
|
||
msg = await conv.send_message('正在查询，请稍后')
|
||
while path:
|
||
path, msg = await logbtn(conv, SENDER, path, '查询日志', msg)
|
||
|
||
|
||
@client.on(events.NewMessage(from_users=chat_id, pattern=r'^/snode'))
|
||
async def mysnode(event):
|
||
'''定义supernode文件命令'''
|
||
SENDER = event.sender_id
|
||
path = _JdDir
|
||
async with client.conversation(SENDER, timeout=60) as conv:
|
||
msg = await conv.send_message('正在查询，请稍后')
|
||
while path:
|
||
path, msg = await nodebtn(conv, SENDER, path, msg)
|
||
|
||
|
||
@client.on(events.NewMessage(from_users=chat_id, pattern=r'^/getfile'))
|
||
async def mygetfile(event):
|
||
'''定义获取文件命令'''
|
||
SENDER = event.sender_id
|
||
path = _JdDir
|
||
async with client.conversation(SENDER, timeout=60) as conv:
|
||
msg = await conv.send_message('正在查询，请稍后')
|
||
while path:
|
||
path, msg = await logbtn(conv, SENDER, path, '文件发送', msg)
|
||
|
||
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')
|
||
|
||
|
||
@client.on(events.NewMessage(from_users=chat_id))
|
||
async def myfile(event):
|
||
'''定义文件操作'''
|
||
try:
|
||
SENDER = event.sender_id
|
||
if event.message.file:
|
||
markup = []
|
||
filename = event.message.file.name
|
||
async with client.conversation(SENDER, timeout=30) as conv:
|
||
msg = await conv.send_message('请选择您要放入的文件夹或操作：\n')
|
||
markup.append(Button.inline('放入config', data=_ConfigDir))
|
||
markup.append(Button.inline('放入scripts', data=_ScriptsDir))
|
||
markup.append(Button.inline('放入thirdpard', data=_ThirdpardDir))
|
||
markup.append(Button.inline('放入thirdpard并运行', data='node'))
|
||
msg = await client.edit_message(msg, '请做出你的选择：', buttons=markup)
|
||
convdata = await conv.wait_event(press_event(SENDER))
|
||
res = bytes.decode(convdata.data)
|
||
if res == 'node':
|
||
await backfile(_ThirdpardDir+'/'+filename)
|
||
await client.download_media(event.message, _ThirdpardDir)
|
||
os.popen('jd {}/{} now >/jd/log/bot.log &'.format(_ThirdpardDir,filename))
|
||
await client.edit_message(msg,'脚本已保存到thirdpard文件夹，并成功在后台运行，请稍后自行查看日志')
|
||
conv.cancel()
|
||
else:
|
||
await backfile(res+'/'+filename)
|
||
await client.download_media(event.message, res)
|
||
await client.edit_message(msg,filename+'已保存到'+res+'文件夹')
|
||
if filename == 'crontab.list':
|
||
os.popen('crontab '+res+'/'+filename)
|
||
await client.edit_message(msg, '定时文件已保存，并更新')
|
||
conv.cancel()
|
||
except Exception as e:
|
||
await client.send_message(chat_id, 'something wrong,I\'m sorry\n'+str(e))
|
||
logger.error('something wrong,I\'m sorry\n'+str(e))
|
||
|
||
@client.on(events.NewMessage(from_users=chat_id, pattern='/node'))
|
||
async def mynode(event):
|
||
'''接收/node命令后执行程序'''
|
||
nodereg = re.compile(r'^/node [\S]+')
|
||
text = re.findall(nodereg, event.raw_text)
|
||
if len(text) == 0:
|
||
res = '''请正确使用/node命令，如
|
||
/node /abc/123.js 运行abc/123.js脚本
|
||
/node /thirdpard/abc.js 运行thirdpard/abc.js脚本
|
||
'''
|
||
await client.send_message(chat_id, res)
|
||
else:
|
||
await cmd('jd '+text[0].replace('/node ', '')+' now')
|
||
|
||
|
||
@client.on(events.NewMessage(from_users=chat_id, pattern='/cmd'))
|
||
async def mycmd(event):
|
||
'''接收/cmd命令后执行程序'''
|
||
if StartCMD:
|
||
cmdreg = re.compile(r'^/cmd [\s\S]+')
|
||
text = re.findall(cmdreg, event.raw_text)
|
||
if len(text) == 0:
|
||
msg = '''请正确使用/cmd命令，如
|
||
/cmd jd clean # 删除旧日志
|
||
/cmd jd update # 更新所有脚本
|
||
/cmd jd myhelp # 导出所有互助码
|
||
不建议直接使用cmd命令执行脚本，请使用/node或/snode
|
||
'''
|
||
await client.send_message(chat_id, msg)
|
||
else:
|
||
print(text)
|
||
await cmd(text[0].replace('/cmd ', ''))
|
||
else:
|
||
await client.send_message(chat_id, '未开启CMD命令，如需使用请修改配置文件')
|
||
|
||
|
||
async def cmd(cmdtext):
|
||
'''定义执行cmd命令'''
|
||
try:
|
||
await client.send_message(chat_id, '开始执行程序，如程序复杂，建议稍等')
|
||
res = os.popen(cmdtext).read()
|
||
if len(res) == 0:
|
||
await client.send_message(chat_id, '已执行，但返回值为空')
|
||
elif len(res) <= 4000:
|
||
await client.send_message(chat_id, res)
|
||
else:
|
||
with open(_LogDir+'/botres.log','w+') as f:
|
||
f.write(res)
|
||
await client.send_message(chat_id, '执行结果较长，请查看日志',file=_LogDir+'/botres.log')
|
||
except Exception as e:
|
||
await client.send_message(chat_id, 'something wrong,I\'m sorry\n'+str(e))
|
||
logger.error('something wrong,I\'m sorry'+str(e))
|
||
|
||
|
||
@client.on(events.NewMessage(from_users=chat_id, pattern=r'^/getcookie'))
|
||
async def mycookie(event):
|
||
'''接收/getcookie后执行程序'''
|
||
login = True
|
||
msg = await client.send_message(chat_id,'正在获取二维码，请稍后')
|
||
global cookiemsg
|
||
try:
|
||
SENDER = event.sender_id
|
||
async with client.conversation(SENDER, timeout=30) as conv:
|
||
getSToken()
|
||
getOKLToken()
|
||
url = 'https://plogin.m.jd.com/cgi-bin/m/tmauth?appid=300&client_type=m&token='+token
|
||
creatqr(url)
|
||
markup = [Button.inline("已扫码", data='confirm'),Button.inline("取消", data='cancel')]
|
||
await client.delete_messages(chat_id,msg)
|
||
cookiemsg = await client.send_message(chat_id, '30s内点击取消将取消本次操作\n如不取消，扫码结果将于30s后显示\n扫码后不想等待点击已扫码', file=img_file,buttons=markup)
|
||
convdata = await conv.wait_event(press_event(SENDER))
|
||
res = bytes.decode(convdata.data)
|
||
if res == 'cancel':
|
||
login = False
|
||
await client.delete_messages(chat_id,cookiemsg)
|
||
msg = await conv.send_message('对话已取消')
|
||
conv.cancel()
|
||
else:
|
||
raise exceptions.TimeoutError()
|
||
except exceptions.TimeoutError:
|
||
expired_time = time.time() + 60 * 2
|
||
while login:
|
||
check_time_stamp = int(time.time() * 1000)
|
||
check_url = 'https://plogin.m.jd.com/cgi-bin/m/tmauthchecktoken?&token=%s&ou_state=0&okl_token=%s' % (
|
||
token, okl_token)
|
||
check_data = {
|
||
'lang': 'chs',
|
||
'appid': 300,
|
||
'returnurl': 'https://wqlogin2.jd.com/passport/LoginRedirect?state=%s&returnurl=//home.m.jd.com/myJd/newhome.action?sceneval=2&ufc=&/myJd/home.action' % check_time_stamp,
|
||
'source': 'wq_passport'
|
||
|
||
}
|
||
check_header = {
|
||
'Referer': f'https://plogin.m.jd.com/login/login?appid=300&returnurl=https://wqlogin2.jd.com/passport/LoginRedirect?state=%s&returnurl=//home.m.jd.com/myJd/newhome.action?sceneval=2&ufc=&/myJd/home.action&source=wq_passport' % check_time_stamp,
|
||
'Cookie': cookies,
|
||
'Connection': 'Keep-Alive',
|
||
'Content-Type': 'application/x-www-form-urlencoded; Charset=UTF-8',
|
||
'Accept': 'application/json, text/plain, */*',
|
||
'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/86.0.4240.111 Safari/537.36',
|
||
|
||
}
|
||
resp = requests.post(
|
||
url=check_url, headers=check_header, data=check_data, timeout=30)
|
||
data = resp.json()
|
||
if data.get("errcode") == 0:
|
||
parseJDCookies(resp.headers)
|
||
await client.delete_messages(chat_id,cookiemsg)
|
||
await client.send_message(chat_id, '以下为获取到的cookie')
|
||
await client.send_message(chat_id, jd_cookie)
|
||
return
|
||
if data.get("errcode") == 21:
|
||
await client.delete_messages(chat_id,cookiemsg)
|
||
await client.send_message(chat_id, '发生了某些错误\n'+data.get("errcode"))
|
||
return
|
||
if time.time() > expired_time:
|
||
await client.delete_messages(chat_id,cookiemsg)
|
||
await client.send_message(chat_id, '超过3分钟未扫码，二维码已过期')
|
||
return
|
||
except Exception as e:
|
||
await client.send_message(chat_id, 'something wrong,I\'m sorry\n'+str(e))
|
||
logger.error('something wrong,I\'m sorry\n'+str(e))
|
||
|
||
|
||
@client.on(events.NewMessage(from_users=chat_id, pattern='/help'))
|
||
@client.on(events.NewMessage(from_users=chat_id, pattern='/start'))
|
||
async def mystart(event):
|
||
'''接收/help /start命令后执行程序'''
|
||
msg = '''使用方法如下：
|
||
/start 开始使用本程序
|
||
/node 执行js脚本文件，直接输入/node jd_bean_change 如执行其他自己js，需输入绝对路径。即可进行执行。该命令会等待脚本执行完，期间不能使用机器人，建议使用snode命令。
|
||
/cmd 执行cmd命令,例如/cmd python3 /python/bot.py 则将执行python目录下的bot.py 不建议使用机器人使用并发，可能产生不明原因的崩溃
|
||
/snode 命令可以选择脚本执行，只能选择/scripts 和/thirdpard目录下的脚本，选择完后直接后台运行，不影响机器人响应其他命令
|
||
/log 选择查看执行日志
|
||
/getfile 获取jd目录下文件
|
||
/getcookie 扫码获取cookie 增加30s内取消按钮，30s后不能进行其他交互直到2分钟或获取到cookie
|
||
此外直接发送文件，会让你选择保存到哪个文件夹，如果选择运行，将保存至thirdpard目录下，并立即运行脚本，crontab.list文件会自动更新时间'''
|
||
await client.send_message(chat_id, msg)
|
||
|
||
with client:
|
||
client.loop.run_forever()
|