mirror of
https://github.com/whyour/qinglong.git
synced 2025-07-30 08:26:08 +08:00
177 lines
7.9 KiB
Python
177 lines
7.9 KiB
Python
from telethon import events, Button
|
||
import requests
|
||
import re
|
||
import time
|
||
import qrcode
|
||
from asyncio import exceptions
|
||
from .. import jdbot, chat_id, img_file
|
||
from .utils import press_event
|
||
|
||
cookiemsg = ''
|
||
# 扫码获取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'
|
||
}
|
||
resp = requests.get(url=get_url, headers=get_header)
|
||
parseGetRespCookie(resp.headers, resp.json())
|
||
|
||
|
||
def parseGetRespCookie(headers, get_resp):
|
||
global s_token
|
||
global cookies
|
||
s_token = get_resp.get('s_token')
|
||
set_cookies = headers.get('set-cookie')
|
||
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}; "
|
||
|
||
|
||
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())
|
||
except Exception as error:
|
||
print("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]
|
||
|
||
|
||
def parseJDCookies(headers):
|
||
global jd_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]
|
||
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)
|
||
|
||
|
||
@jdbot.on(events.NewMessage(from_users=chat_id, pattern=r'^/getcookie'))
|
||
async def mycookie(event):
|
||
'''接收/getcookie后执行程序'''
|
||
login = True
|
||
msg = await jdbot.send_message(chat_id, '正在获取二维码，请稍后')
|
||
global cookiemsg
|
||
try:
|
||
SENDER = event.sender_id
|
||
async with jdbot.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 jdbot.delete_messages(chat_id, msg)
|
||
cookiemsg = await jdbot.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 jdbot.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 jdbot.delete_messages(chat_id, cookiemsg)
|
||
await jdbot.send_message(chat_id, '以下为获取到的cookie')
|
||
await jdbot.send_message(chat_id, jd_cookie)
|
||
return
|
||
if data.get("errcode") == 21:
|
||
await jdbot.delete_messages(chat_id, cookiemsg)
|
||
await jdbot.send_message(chat_id, '发生了某些错误\n'+data.get("errcode"))
|
||
return
|
||
if time.time() > expired_time:
|
||
await jdbot.delete_messages(chat_id, cookiemsg)
|
||
await jdbot.send_message(chat_id, '超过3分钟未扫码，二维码已过期')
|
||
return
|
||
except Exception as e:
|
||
await jdbot.send_message(chat_id, 'something wrong,I\'m sorry\n'+str(e))
|