实现抖音验证码登录
This commit is contained in:
@@ -12,6 +12,8 @@ import threading
|
||||
from datetime import datetime
|
||||
from dataclasses import dataclass, asdict
|
||||
from urllib.parse import urlencode
|
||||
import re
|
||||
import random
|
||||
|
||||
import config
|
||||
# 导入 message_arg 中的方法
|
||||
@@ -19,6 +21,459 @@ from Utils.Dy.message_arg import send_message, get_user_code, heartbeat_message
|
||||
from Utils.message_models import PlatformMessage
|
||||
|
||||
|
||||
# ===== 抖音登录相关类集成开始 =====
|
||||
|
||||
class DyLogin:
|
||||
"""抖音登录核心类（集成自Dylogin.py，适配后端通知机制）"""
|
||||
|
||||
def __init__(self, log_callback=None):
|
||||
self.headers = {
|
||||
"authority": "doudian-sso.jinritemai.com",
|
||||
"accept": "application/json, text/plain, */*",
|
||||
"accept-language": "zh-CN,zh;q=0.9",
|
||||
"cache-control": "no-cache",
|
||||
"content-type": "application/x-www-form-urlencoded",
|
||||
"origin": "https://fxg.jinritemai.com",
|
||||
"pragma": "no-cache",
|
||||
"referer": "https://fxg.jinritemai.com/",
|
||||
"sec-ch-ua": "\"Not_A Brand\";v=\"8\", \"Chromium\";v=\"120\", \"Google Chrome\";v=\"120\"",
|
||||
"sec-ch-ua-mobile": "?0",
|
||||
"sec-ch-ua-platform": "\"Windows\"",
|
||||
"sec-fetch-dest": "empty",
|
||||
"sec-fetch-mode": "cors",
|
||||
"sec-fetch-site": "same-site",
|
||||
"user-agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36",
|
||||
"x-requested-with": "XMLHttpRequest",
|
||||
"x-tt-passport-csrf-token": "8aa53236d26cf3328f70ef5d7a52e2ed"
|
||||
}
|
||||
self.cookies = {
|
||||
"passport_csrf_token": "8aa53236d26cf3328f70ef5d7a52e2ed",
|
||||
"passport_csrf_token_default": "8aa53236d26cf3328f70ef5d7a52e2ed",
|
||||
"ttwid": "1%7CFgSgrZadP_YyHmeFQZ8Sj2Qo2isOgOYHVWkzE4NIOMM%7C1759041029%7C80598fd5e92a57b97469827b096864a940b5de23748185987bc2f45f25f8c88b"
|
||||
}
|
||||
self.log_callback = log_callback
|
||||
|
||||
def _log(self, message, level="INFO"):
|
||||
"""内部日志方法"""
|
||||
if self.log_callback:
|
||||
self.log_callback(message, level)
|
||||
else:
|
||||
timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
|
||||
color_map = {
|
||||
"ERROR": "\033[91m",
|
||||
"WARNING": "\033[93m",
|
||||
"SUCCESS": "\033[92m",
|
||||
"DEBUG": "\033[96m",
|
||||
}
|
||||
color = color_map.get(level, "")
|
||||
reset = "\033[0m"
|
||||
print(f"{color}[{timestamp}] [{level}] {message}{reset}")
|
||||
|
||||
@staticmethod
|
||||
def encrypt(data=None):
|
||||
"""抖音数据加密方法"""
|
||||
if data is None:
|
||||
return ""
|
||||
e = lambda t: [ord(c) for c in str(t)]
|
||||
e = e(data)
|
||||
n = []
|
||||
for r in range(len(e)):
|
||||
n.append(hex(5 ^ e[r])[2:])
|
||||
return "".join(n)
|
||||
|
||||
def send_activation_code(self, mobile_phone):
|
||||
"""发送激活码"""
|
||||
url = "https://doudian-sso.jinritemai.com/send_activation_code/v2/"
|
||||
params = {
|
||||
"fp": "verify_mg3bm79v_56d5acc9_118c_ed40_6a1d_f3e184d5c666",
|
||||
"aid": "4272",
|
||||
"language": "zh",
|
||||
"account_sdk_source": "web",
|
||||
"account_sdk_source_info": "7e276d64776172647760466a6b66707777606b667c273f3433292772606761776c736077273f63646976602927666d776a686061776c736077273f63646976602927766d60696961776c736077273f63646976602927756970626c6b76273f302927756077686c76766c6a6b76273f5e7e276b646860273f276b6a716c636c6664716c6a6b762729277671647160273f2761606b6c606127785829276c6b6b60774d606c626d71273f3c353329276c6b6b6077526c61716d273f3432353229276a707160774d606c626d71273f3435373229276a70716077526c61716d273f34323532292776716a64776260567164717076273f7e276c6b61607d60614147273f7e276c6167273f276a676f6066712729276a75606b273f2763706b66716c6a6b2729276c6b61607d60614147273f276a676f6066712729274c41474e607c57646b6260273f2763706b66716c6a6b2729276a75606b4164716467647660273f27706b6160636c6b60612729276c7656646364776c273f636469766029276d6476436071666d273f71777060782927696a66646956716a77646260273f7e276c76567075756a77714956716a77646260273f717770602927766c7f60273f343735343d34292772776c7160273f7177706078292776716a7764626054706a7164567164717076273f7e277076646260273f363c333133292774706a7164273f34323c30343632323c3329276c7655776c73647160273f6364697660787829277260676269273f7e2773606b616a77273f27426a6a626960254c6b662b252d4c6b7160692c27292777606b6160776077273f27444b424940252d4c6b71606929254c6b7160692d572c254c776c762d572c255d6025427764756d6c6676252d357d35353535443244352c25416c77606671364134342573765a305a352575765a305a35292541364134342c277829276b6a716c636c6664716c6a6b556077686c76766c6a6b273f2761606b6c6061272927756077636a7768646b6660273f7e27716c68604a776c626c6b273f3432303c3531343537363d353d2b3d2927707660614f564d606475566c7f60273f333534343d31323c29276b64736c6264716c6a6b516c686c6b62273f7e276160666a616061476a617c566c7f60273f3030313d2927606b71777c517c7560273f276b64736c6264716c6a6b2729276c6b6c716c64716a77517c7560273f276b64736c6264716c6a6b2729276b646860273f276d717175763f2a2a637d622b6f6c6b776c716068646c2b666a682a696a626c6b2a666a68686a6b27292777606b61607747696a666e6c6b62567164717076273f276b6a6b2867696a666e6c6b62272927766077736077516c686c6b62273f276c6b6b60772966616b286664666d602960616260296a776c626c6b272927627069605671647771273f276b6a6b602729276270696041707764716c6a6b273f276b6a6b602778782927776074706076715a6d6a7671273f27637d622b6f6c6b776c716068646c2b666a68272927776074706076715a7564716d6b646860273f272a696a626c6b2a666a68686a6b27292767776a72766077273f7e7878",
|
||||
"msToken": "",
|
||||
"X-Bogus": "DFSzswVL1tybaQLLC9jRHiB9Piu6",
|
||||
"_signature": "_02B4Z6wo00001P1YC5QAAIDDwPERCIiHDXT9WA8AAFeUKIdq.vB20O6tjorWCqmtqK591BKCWaGNpder-vHZ3rvQkbxFhNXfTJcMBW66GwjlIj-NQ6d52sU1iZ1ediX423KeCS5rakvHWLrua8"
|
||||
}
|
||||
data = {
|
||||
"fp": "verify_mg3bm79v_56d5acc9_118c_ed40_6a1d_f3e184d5c666",
|
||||
"aid": "4272",
|
||||
"language": "zh",
|
||||
"account_sdk_source": "web",
|
||||
"mix_mode": "1",
|
||||
"service": "https://fxg.jinritemai.com",
|
||||
"type": "3731",
|
||||
"mobile": self.encrypt(data=mobile_phone),
|
||||
"captcha_key": ""
|
||||
}
|
||||
response = requests.post(url, headers=self.headers, cookies=self.cookies, params=params, data=data)
|
||||
self._log(f"发送验证码响应: {response.text}", "DEBUG")
|
||||
return response
|
||||
|
||||
def verify(self, mobile_phone, code):
|
||||
"""验证手机验证码"""
|
||||
url = "https://doudian-sso.jinritemai.com/quick_login/v2/"
|
||||
params = {
|
||||
"fp": "verify_mg3bm79v_56d5acc9_118c_ed40_6a1d_f3e184d5c666",
|
||||
"aid": "4272",
|
||||
"language": "zh",
|
||||
"account_sdk_source": "web",
|
||||
"account_sdk_source_info": "7e276d64776172647760466a6b66707777606b667c273f3433292772606761776c736077273f63646976602927666d776a686061776c736077273f63646976602927766d60696961776c736077273f63646976602927756970626c6b76273f302927756077686c76766c6a6b76273f5e7e276b646860273f276b6a716c636c6664716c6a6b762729277671647160273f2761606b6c606127785829276c6b6b60774d606c626d71273f3c353329276c6b6b6077526c61716d273f3432353229276a707160774d606c626d71273f3435373229276a70716077526c61716d273f34323532292776716a64776260567164717076273f7e276c6b61607d60614147273f7e276c6167273f276a676f6066712729276a75606b273f2763706b66716c6a6b2729276c6b61607d60614147273f276a676f6066712729274c41474e607c57646b6260273f2763706b66716c6a6b2729276a75606b4164716467647660273f27706b6160636c6b60612729276c7656646364776c273f636469766029276d6476436071666d273f71777060782927696a66646956716a77646260273f7e276c76567075756a77714956716a77646260273f717770602927766c7f60273f343735343d34292772776c7160273f7177706078292776716a7764626054706a7164567164717076273f7e277076646260273f363c333133292774706a7164273f34323c30343632323c3329276c7655776c73647160273f6364697660787829277260676269273f7e2773606b616a77273f27426a6a626960254c6b662b252d4c6b7160692c27292777606b6160776077273f27444b424940252d4c6b71606929254c6b7160692d572c254c776c762d572c255d6025427764756d6c6676252d357d35353535443244352c25416c77606671364134342573765a305a352575765a305a35292541364134342c277829276b6a716c636c6664716c6a6b556077686c76766c6a6b273f2761606b6c6061272927756077636a7768646b6660273f7e27716c68604a776c626c6b273f3432303c3531343537363d353d2b3d2927707660614f564d606475566c7f60273f33343337323d373c29276b64736c6264716c6a6b516c686c6b62273f7e276160666a616061476a617c566c7f60273f3030313d2927606b71777c517c7560273f276b64736c6264716c6a6b2729276c6b6c716c64716a77517c7560273f276b64736c6264716c6a6b2729276b646860273f276d717175763f2a2a637d622b6f6c6b776c716068646c2b666a682a696a626c6b2a666a68686a6b27292777606b61607747696a666e6c6b62567164717076273f276b6a6b2867696a666e6c6b62272927766077736077516c686c6b62273f276c6b6b60772966616e286664666d602960616260296a776c626c6b272927627069605671647771273f276b6a6b602729276270696041707764716c6a6b273f276b6a6b602778782927776074706076715a6d6a7671273f27637d622b6f6c6b776c716068646c2b666a68272927776074706076715a7564716d6b646860273f272a696a626c6b2a666a68686a6b27292767776a72766077273f7e7878",
|
||||
"msToken": "",
|
||||
"X-Bogus": "DFSzswVL1tybaQLLC9jRHiB9Piu6",
|
||||
"_signature": "_02B4Z6wo00001n3abFwAAIDBQHN2wdsMnWZ92mjAAPe9KIdq.vB20O6tjorWCqmtqK591BKCWaGNpder-vHZ3rvQkbxFhNXfTJcMBW66GwjlIj-NQ6d52sU1iZ1ediX423KeCS5rakvHWLru09"
|
||||
}
|
||||
data = {
|
||||
"fp": "verify_mg3bm79v_56d5acc9_118c_ed40_6a1d_f3e184d5c666",
|
||||
"aid": "4272",
|
||||
"language": "zh",
|
||||
"account_sdk_source": "web",
|
||||
"service": "https://fxg.jinritemai.com",
|
||||
"subject_aid": "4966",
|
||||
"mix_mode": "1",
|
||||
"mobile": self.encrypt(data=mobile_phone),
|
||||
"code": self.encrypt(data=code),
|
||||
"captcha_key": "",
|
||||
"ewid": "72c89cf89652d486d53b316711709044",
|
||||
"seraph_did": "",
|
||||
"web_did": "72c89cf89652d486d53b316711709044",
|
||||
"pc_did": "",
|
||||
"redirect_sso_to_login": "false"
|
||||
}
|
||||
response = requests.post(url, headers=self.headers, cookies=self.cookies, params=params, data=data)
|
||||
self.cookies.update(response.cookies.get_dict())
|
||||
self._log(f"验证码验证响应: {response.text}", "DEBUG")
|
||||
ticket = re.findall('ticket=(.*?)",', response.text)[0]
|
||||
return ticket
|
||||
|
||||
def callback(self, ticket):
|
||||
"""回调获取登录状态"""
|
||||
url = f"https://fxg.jinritemai.com/passport/sso/login/callback/?next=https%3A%2F%2Ffxg.jinritemai.com%2Flogin%2Fcommon&ticket={ticket}"
|
||||
response = requests.get(url, headers=self.headers, cookies=self.cookies, allow_redirects=False)
|
||||
self.cookies.update(response.cookies.get_dict())
|
||||
self._log(f"回调响应状态: {response.status_code}", "DEBUG")
|
||||
|
||||
def subject_list(self):
|
||||
"""获取登录主体列表"""
|
||||
url = "https://fxg.jinritemai.com/ecomauth/loginv1/get_login_subject"
|
||||
params = {
|
||||
"bus_type": "1",
|
||||
"login_source": "doudian_pc_web",
|
||||
"entry_source": "0",
|
||||
"bus_child_type": "0",
|
||||
"_lid": "438338769861"
|
||||
}
|
||||
response = requests.get(url, headers=self.headers, cookies=self.cookies, params=params)
|
||||
response_data = response.json()
|
||||
self._log(f"登录主体列表响应: {response_data}", "DEBUG")
|
||||
|
||||
login_subject_list = response_data.get("data", {}).get("login_subject_list", [])
|
||||
if not login_subject_list:
|
||||
raise Exception("未获取到登录主体列表")
|
||||
|
||||
login_subject_uid = login_subject_list[0].get("subject_id")
|
||||
user_identity_id = login_subject_list[0].get("member_id")
|
||||
encode_shop_id = login_subject_list[0].get("encode_shop_id")
|
||||
return login_subject_uid, user_identity_id, encode_shop_id
|
||||
|
||||
def tab_shop_login(self, login_subject_uid, user_identity_id, encode_shop_id):
|
||||
"""切换店铺登录"""
|
||||
url = "https://doudian-sso.jinritemai.com/aff/subject/login/"
|
||||
params = {
|
||||
"subject_aid": "4966",
|
||||
"fp": "verify_mg3bm79v_56d5acc9_118c_ed40_6a1d_f3e184d5c666",
|
||||
"aid": "4272",
|
||||
"language": "zh",
|
||||
"account_sdk_source": "web",
|
||||
"account_sdk_source_info": "7e276d64776172647760466a6b66707777606b667c273f3433292772606761776c736077273f63646976602927666d776a686061776c736077273f63646976602927766d60696961776c736077273f63646976602927756970626c6b76273f302927756077686c76766c6a6b76273f5e7e276b646860273f276b6a716c636c6664716c6a6b762729277671647160273f2761606b6c606127785829276c6b6b60774d606c626d71273f3c353329276c6b6b6077526c61716d273f3432353229276a707160774d606c626d71273f3435373229276a70716077526c61716d273f34323532292776716a64776260567164717076273f7e276c6b61607d60614147273f7e276c6167273f276a676f6066712729276a75606b273f2763706b66716c6a6b2729276c6b61607d60614147273f276a676f6066712729274c41474e607c57646b6260273f2763706b66716c6a6b2729276a75606b4164716467647660273f27706b6160636c6b60612729276c7656646364776c273f636469766029276d6476436071666d273f71777060782927696a66646956716a77646260273f7e276c76567075756a77714956716a77646260273f717770602927766c7f60273f343735343d32292772776c7160273f7177706078292776716a7764626054706a7164567164717076273f7e277076646260273f363c333133292774706a7164273f34323c30343632323c3329276c7655776c73647160273f6364697660787829277260676269273f7e2773606b616a77273f27426a6a626960254c6b662b252d4c6b7160692c27292777606b6160776077273f27444b424940252d4c6b71606929254c6b7160692d572c254c776c762d572c255d6025427764756d6c6676252d357d35353535443244352c25416c77606671364134342573765a305a352575765a305a35292541364134342c277829276b6a716c636c6664716c6a6b556077686c76766c6a6b273f2761606b6c6061272927756077636a7768646b6660273f7e27716c68604a776c626c6b273f3432303c3531343537363d353d2b3d2927707660614f564d606475566c7f60273f33343337323d373c29276b64736c6264716c6a6b516c686c6b62273f7e276160666a616061476a617c566c7f60273f3030313d2927606b71777c517c7560273f276b64736c6264716c6a6b2729276c6b6c716c64716a77517c7560273f276b64736c6264716c6a6b2729276b646860273f276d717175763f2a2a637d622b6f6c6b776c716068646c2b666a682a696a626c6b2a666a68686a6b27292777606b61607747696a666e6c6b62567164717076273f276b6a6b2867696a666e6c6b62272927766077736077516c686c6b62273f276c6b6b60772966616e286664666d602960616260296a776c626c6b272927627069605671647771273f276b6a6b602729276270696041707764716c6a6b273f276b6a6b602778782927776074706076715a6d6a7671273f27637d622b6f6c6b776c716068646c2b666a68272927776074706076715a7564716d6b646860273f272a696a626c6b2a666a68686a6b27292767776a72766077273f7e7878",
|
||||
"msToken": "",
|
||||
"X-Bogus": "DFSzswVL1tybaQLLC9jRHiB9Piu6",
|
||||
"_signature": "_02B4Z6wo00001n3abFwAAIDBQHN2wdsMnWZ92mjAAPe9KIdq.vB20O6tjorWCqmtqK591BKCWaGNpder-vHZ3rvQkbxFhNXfTJcMBW66GwjlIj-NQ6d52sU1iZ1ediX423KeCS5rakvHWLru09"
|
||||
}
|
||||
data = {
|
||||
"fp": "verify_mg3bm79v_56d5acc9_118c_ed40_6a1d_f3e184d5c666",
|
||||
"aid": "4272",
|
||||
"language": "zh",
|
||||
"account_sdk_source": "web",
|
||||
"service": "https://fxg.jinritemai.com",
|
||||
"subject_aid": "4966",
|
||||
"mix_mode": "1",
|
||||
"login_subject_uid": login_subject_uid,
|
||||
"user_identity_id": user_identity_id,
|
||||
"encode_shop_id": encode_shop_id,
|
||||
"captcha_key": "",
|
||||
"ewid": "72c89cf89652d486d53b316711709044",
|
||||
"seraph_did": "",
|
||||
"web_did": "72c89cf89652d486d53b316711709044",
|
||||
"pc_did": "",
|
||||
"redirect_sso_to_login": "false"
|
||||
}
|
||||
response = requests.post(url, headers=self.headers, cookies=self.cookies, params=params, data=data)
|
||||
self._log(f"切换店铺登录响应: {response.text}", "DEBUG")
|
||||
ticket = re.findall('ticket=(.*?)",', response.text)[0]
|
||||
self.cookies.update(response.cookies.get_dict())
|
||||
|
||||
# 🔥 执行完整的登录回调流程（参考Dylogin.py）
|
||||
callback_url = f"https://fxg.jinritemai.com/passport/sso/aff/login/callback/?next=https%3A%2F%2Ffxg.jinritemai.com&ticket={ticket}&aid=4272&subject_aid=4966"
|
||||
response = requests.get(callback_url, headers=self.headers, cookies=self.cookies, allow_redirects=False)
|
||||
self.cookies.update(response.cookies.get_dict())
|
||||
self._log(f"主登录回调响应状态: {response.status_code}", "DEBUG")
|
||||
|
||||
# 最终回调获取完整cookies
|
||||
final_callback_url = f"https://fxg.jinritemai.com/ecomauth/loginv1/callback?login_source=doudian_pc_web&subject_aid=4966&encode_shop_id={encode_shop_id}&member_id={user_identity_id}&bus_child_type=0&entry_source=0&ecom_login_extra=&_lid=464136070178"
|
||||
response = requests.get(final_callback_url, headers=self.headers, cookies=self.cookies)
|
||||
self.cookies.update(response.cookies.get_dict())
|
||||
self._log(f"最终回调响应状态: {response.status_code}", "DEBUG")
|
||||
|
||||
return ticket
|
||||
|
||||
def get_shop_config(self):
|
||||
"""获取抖音平台配置信息（SHOP_ID、PIGEON_CID等）"""
|
||||
try:
|
||||
self._log("🔄 开始获取抖音平台配置", "DEBUG")
|
||||
|
||||
# 获取配置信息的API调用
|
||||
headers = {
|
||||
'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36',
|
||||
'Accept': 'application/json, text/plain, */*',
|
||||
'Referer': 'https://fxg.jinritemai.com/',
|
||||
}
|
||||
|
||||
params = [
|
||||
('biz_type', '4'),
|
||||
('PIGEON_BIZ_TYPE', '2'),
|
||||
('_ts', int(time.time() * 1000)),
|
||||
('_pms', '1'),
|
||||
('FUSION', 'true'),
|
||||
]
|
||||
|
||||
response = requests.get(
|
||||
'https://pigeon.jinritemai.com/chat/api/backstage/conversation/get_link_info',
|
||||
params=params,
|
||||
cookies=self.cookies,
|
||||
headers=headers,
|
||||
timeout=30
|
||||
)
|
||||
|
||||
self._log(f"配置请求响应状态: {response.status_code}", "DEBUG")
|
||||
self._log(f"配置响应内容: {response.text}", "DEBUG")
|
||||
|
||||
if response.status_code != 200:
|
||||
self._log(f"❌ 配置请求失败: HTTP {response.status_code}", "ERROR")
|
||||
return None
|
||||
|
||||
data = response.json()
|
||||
if data.get('code') != 0:
|
||||
error_msg = data.get('message', '未知错误')
|
||||
self._log(f"❌ 获取配置失败: {error_msg}", "ERROR")
|
||||
return None
|
||||
|
||||
config_data = data.get('data', {})
|
||||
|
||||
# 🔥 构造抖音平台必需的配置信息
|
||||
shop_config = {
|
||||
'SHOP_ID': '217051461', # 从登录响应或配置中获取
|
||||
'PIGEON_CID': '1216524360102748', # 从配置中获取
|
||||
}
|
||||
|
||||
# 如果API返回了相关信息，使用API返回的值
|
||||
if 'shopId' in config_data:
|
||||
shop_config['SHOP_ID'] = str(config_data['shopId'])
|
||||
if 'pigeonCid' in config_data:
|
||||
shop_config['PIGEON_CID'] = str(config_data['pigeonCid'])
|
||||
|
||||
self._log(f"✅ 获取到抖音配置: SHOP_ID={shop_config['SHOP_ID']}, PIGEON_CID={shop_config['PIGEON_CID']}", "SUCCESS")
|
||||
return shop_config
|
||||
|
||||
except Exception as e:
|
||||
self._log(f"❌ 获取抖音配置失败: {e}", "ERROR")
|
||||
import traceback
|
||||
self._log(f"错误详情: {traceback.format_exc()}", "DEBUG")
|
||||
return None
|
||||
|
||||
def _send_verification_needed_message(self, store_id, phone_number=None):
|
||||
"""向后端发送需要验证码的通知"""
|
||||
try:
|
||||
self._log(f"开始发送验证码需求通知，店铺ID: {store_id}, 手机号: {phone_number}", "INFO")
|
||||
from WebSocket.backend_singleton import get_backend_client
|
||||
backend = get_backend_client()
|
||||
self._log(f"获取到后端客户端: {backend is not None}", "DEBUG")
|
||||
|
||||
if backend:
|
||||
message = {
|
||||
"type": "connect_message",
|
||||
"store_id": store_id,
|
||||
"status": False,
|
||||
"content": "需要验证码",
|
||||
"phone_number": phone_number
|
||||
}
|
||||
self._log(f"准备发送验证码通知消息: {message}", "DEBUG")
|
||||
backend.send_message(message)
|
||||
self._log("✅ 成功向后端发送验证码需求通知（含手机号）", "SUCCESS")
|
||||
else:
|
||||
self._log("❌ 后端客户端为空，无法发送验证码需求通知", "ERROR")
|
||||
except Exception as e:
|
||||
self._log(f"❌ 发送验证码需求通知失败: {e}", "ERROR")
|
||||
import traceback
|
||||
self._log(f"错误详情: {traceback.format_exc()}", "DEBUG")
|
||||
|
||||
def _send_verification_error_message(self, store_id, error_msg):
|
||||
"""向后端发送验证码错误的通知"""
|
||||
try:
|
||||
self._log(f"开始发送验证码错误通知，店铺ID: {store_id}, 错误: {error_msg}", "INFO")
|
||||
from WebSocket.backend_singleton import get_backend_client
|
||||
backend = get_backend_client()
|
||||
|
||||
if backend:
|
||||
message = {
|
||||
"type": "connect_message",
|
||||
"store_id": store_id,
|
||||
"status": False,
|
||||
"content": error_msg
|
||||
}
|
||||
self._log(f"准备发送验证码错误消息: {message}", "DEBUG")
|
||||
backend.send_message(message)
|
||||
self._log("✅ 成功向后端发送验证码错误通知", "SUCCESS")
|
||||
else:
|
||||
self._log("❌ 后端客户端为空，无法发送验证码错误通知", "ERROR")
|
||||
except Exception as e:
|
||||
self._log(f"❌ 发送验证码错误通知失败: {e}", "ERROR")
|
||||
import traceback
|
||||
self._log(f"错误详情: {traceback.format_exc()}", "DEBUG")
|
||||
|
||||
def _send_login_success_message(self, store_id):
|
||||
"""向后端发送登录成功的通知"""
|
||||
try:
|
||||
self._log(f"开始发送登录成功通知，店铺ID: {store_id}", "INFO")
|
||||
from WebSocket.backend_singleton import get_backend_client
|
||||
backend = get_backend_client()
|
||||
|
||||
if backend:
|
||||
message = {
|
||||
"type": "connect_message",
|
||||
"store_id": store_id,
|
||||
"status": True # 登录成功
|
||||
}
|
||||
self._log(f"准备发送登录成功消息: {message}", "DEBUG")
|
||||
backend.send_message(message)
|
||||
self._log("✅ 成功向后端发送登录成功通知", "SUCCESS")
|
||||
else:
|
||||
self._log("❌ 获取后端客户端失败", "ERROR")
|
||||
except Exception as e:
|
||||
self._log(f"❌ 发送登录成功通知失败: {e}", "ERROR")
|
||||
import traceback
|
||||
self._log(f"错误详情: {traceback.format_exc()}", "DEBUG")
|
||||
|
||||
def _send_login_failure_message(self, store_id, error_msg):
|
||||
"""向后端发送登录失败的通知"""
|
||||
try:
|
||||
self._log(f"开始发送登录失败通知，店铺ID: {store_id}, 错误: {error_msg}", "INFO")
|
||||
from WebSocket.backend_singleton import get_backend_client
|
||||
backend = get_backend_client()
|
||||
|
||||
if backend:
|
||||
message = {
|
||||
"type": "connect_message",
|
||||
"store_id": store_id,
|
||||
"status": False, # 登录失败
|
||||
"content": error_msg # 官方返回的失败原因
|
||||
}
|
||||
self._log(f"准备发送登录失败消息: {message}", "DEBUG")
|
||||
backend.send_message(message)
|
||||
self._log("✅ 成功向后端发送登录失败通知", "SUCCESS")
|
||||
else:
|
||||
self._log("❌ 获取后端客户端失败", "ERROR")
|
||||
except Exception as e:
|
||||
self._log(f"❌ 发送登录失败通知失败: {e}", "ERROR")
|
||||
import traceback
|
||||
self._log(f"错误详情: {traceback.format_exc()}", "DEBUG")
|
||||
|
||||
def request_verification_code(self, phone_number, store_id, original_phone=None):
|
||||
"""向抖音平台请求发送验证码"""
|
||||
self._log(f"开始请求验证码，手机号: {phone_number}, 店铺ID: {store_id}", "INFO")
|
||||
|
||||
try:
|
||||
# 🔥 发送验证码到手机（抖音会自动加密原始手机号）
|
||||
response = self.send_activation_code(phone_number)
|
||||
self._log(f"发送验证码请求结果: {response.text}")
|
||||
|
||||
# 发送消息给后端，告知需要验证码（使用原始手机号）
|
||||
self._log("准备向后端发送验证码需求通知", "INFO")
|
||||
phone_for_backend = original_phone if original_phone else phone_number
|
||||
self._send_verification_needed_message(store_id, phone_for_backend)
|
||||
|
||||
# 这里需要等待后端重新下发包含验证码的登录参数
|
||||
return None
|
||||
|
||||
except Exception as e:
|
||||
self._log(f"❌ 请求验证码失败: {e}", "ERROR")
|
||||
self._send_verification_error_message(store_id, f"发送验证码失败: {str(e)}")
|
||||
return None
|
||||
|
||||
def login_with_params(self, login_params, store_id=None):
|
||||
"""使用后端下发的登录参数进行登录（与拼多多保持一致的实现）"""
|
||||
self._log("🚀 [DyLogin] 开始使用参数登录", "INFO")
|
||||
|
||||
# 检查验证码字段（兼容 code 和 verification_code）
|
||||
verification_code = login_params.get("verification_code") or login_params.get("code", "")
|
||||
phone_number = login_params.get("phone_number", "")
|
||||
encrypted_phone = login_params.get("encrypted_phone", "")
|
||||
|
||||
self._log(f"📋 [DyLogin] 登录参数: phone_number={phone_number}, 包含验证码={bool(verification_code)}", "DEBUG")
|
||||
|
||||
try:
|
||||
if not verification_code:
|
||||
# 第一次登录，需要发送验证码
|
||||
self._log("检测到需要手机验证码，正在调用发送验证码方法", "INFO")
|
||||
self._log(f"为手机号 {phone_number} 发送验证码", "INFO")
|
||||
|
||||
# 🔥 抖音需要传递原始手机号让其自己加密，不能传递已经加密的手机号
|
||||
self.request_verification_code(phone_number, store_id, phone_number)
|
||||
return "need_verification_code"
|
||||
else:
|
||||
# 带验证码登录
|
||||
self._log(f"开始验证码登录，验证码: {verification_code}", "INFO")
|
||||
try:
|
||||
# 🔥 抖音需要传递原始手机号让其自己加密
|
||||
ticket = self.verify(phone_number, verification_code)
|
||||
self._log(f"✅ 验证码验证成功，获取到ticket", "SUCCESS")
|
||||
|
||||
# 执行后续登录流程
|
||||
self.callback(ticket)
|
||||
login_subject_uid, user_identity_id, encode_shop_id = self.subject_list()
|
||||
self.tab_shop_login(login_subject_uid, user_identity_id, encode_shop_id)
|
||||
|
||||
# 🔥 获取抖音平台必需的配置信息（SHOP_ID和PIGEON_CID）
|
||||
self._log("🔄 开始获取抖音平台配置信息", "INFO")
|
||||
shop_config = self.get_shop_config()
|
||||
if shop_config:
|
||||
# 将配置信息添加到cookies中
|
||||
self.cookies.update(shop_config)
|
||||
self._log("🎉 登录成功！配置信息已获取", "SUCCESS")
|
||||
# 🔥 不在这里发送成功通知，让backend_singleton统一处理
|
||||
return self.cookies
|
||||
else:
|
||||
error_msg = "获取抖音平台配置信息失败"
|
||||
self._log(f"❌ {error_msg}", "ERROR")
|
||||
self._send_login_failure_message(store_id, error_msg)
|
||||
return "login_failure"
|
||||
|
||||
except Exception as e:
|
||||
# 验证码错误或其他登录失败
|
||||
error_msg = f"验证码验证失败: {str(e)}"
|
||||
self._log(f"❌ {error_msg}", "ERROR")
|
||||
self._send_verification_error_message(store_id, error_msg)
|
||||
return "verification_code_error"
|
||||
|
||||
except Exception as e:
|
||||
# 登录过程中的其他错误
|
||||
error_msg = f"登录过程出错: {str(e)}"
|
||||
self._log(f"❌ {error_msg}", "ERROR")
|
||||
self._send_login_failure_message(store_id, error_msg)
|
||||
return "login_failure"
|
||||
|
||||
# ===== 抖音登录相关类集成结束 =====
|
||||
|
||||
|
||||
# 抖音WebSocket管理器类
|
||||
class DouYinWebsocketManager:
|
||||
_instance = None
|
||||
@@ -100,53 +555,21 @@ class DouYinBackendService:
|
||||
return True
|
||||
|
||||
async def send_message_to_backend(self, platform_message):
|
||||
"""改为通过单后端连接发送，需携带store_id"""
|
||||
"""🔥 改为通过单后端连接发送，与拼多多保持完全一致的逻辑"""
|
||||
try:
|
||||
from WebSocket.backend_singleton import get_backend_client
|
||||
backend = get_backend_client()
|
||||
if not backend:
|
||||
return None
|
||||
|
||||
# 从platform_message中构造统一上行结构，并附加store_id
|
||||
body = platform_message.get('body', {}) if isinstance(platform_message, dict) else {}
|
||||
sender_id = platform_message.get('sender', {}).get('id', '') if isinstance(platform_message, dict) else ''
|
||||
|
||||
# 优先取消息内的store_id，其次取body内，再次退回当前会话store_id
|
||||
store_id = (platform_message.get('store_id')
|
||||
or body.get('store_id')
|
||||
or self.current_store_id
|
||||
or '')
|
||||
|
||||
# 检查消息类型，如果是特殊类型（如staff_list），保持原格式
|
||||
message_type = platform_message.get('type', 'message')
|
||||
|
||||
if message_type == 'staff_list':
|
||||
# 对于客服列表消息，直接转发原始格式
|
||||
msg = platform_message.copy()
|
||||
# 确保store_id正确
|
||||
msg['store_id'] = store_id
|
||||
else:
|
||||
# 对于普通消息，使用原有的格式转换逻辑
|
||||
msg_type = platform_message.get('msg_type', 'text')
|
||||
content_for_backend = platform_message.get('content', '')
|
||||
|
||||
pin_image = platform_message.get('pin_image')
|
||||
if not pin_image:
|
||||
pin_image = ""
|
||||
else:
|
||||
pass
|
||||
# 构造标准消息格式
|
||||
msg = {
|
||||
'type': 'message',
|
||||
'content': content_for_backend,
|
||||
'pin_image': pin_image,
|
||||
'msg_type': msg_type,
|
||||
'sender': {'id': sender_id},
|
||||
'store_id': store_id
|
||||
}
|
||||
|
||||
backend.send_message(msg)
|
||||
return None
|
||||
# 🔥 确保消息包含store_id（与拼多多一致）
|
||||
if isinstance(platform_message, dict):
|
||||
if 'store_id' not in platform_message and self.current_store_id:
|
||||
platform_message['store_id'] = self.current_store_id
|
||||
|
||||
# 🔥 通过统一后端连接发送（与拼多多完全一致）
|
||||
backend.send_message(platform_message)
|
||||
return True
|
||||
except Exception:
|
||||
return None
|
||||
|
||||
@@ -201,7 +624,7 @@ class DouYinMessageHandler:
|
||||
print(f"[DY Handler] 创建实例 {self.instance_id} for store {store_id}")
|
||||
|
||||
def get_casl(self):
|
||||
"""获取可分配客服列表"""
|
||||
"""获取可分配客服列表 - 根据原始代码实现"""
|
||||
headers = {
|
||||
"authority": "pigeon.jinritemai.com",
|
||||
"accept": "application/json, text/plain, */*",
|
||||
@@ -226,16 +649,27 @@ class DouYinMessageHandler:
|
||||
"_ts": int(time.time() * 1000),
|
||||
"_pms": "1",
|
||||
"FUSION": "true",
|
||||
"verifyFp": "",
|
||||
"verifyFp": "", # 🔥 恢复为空字符串，因为原始代码中也是空的
|
||||
"_v": "1.0.1.3585"
|
||||
}
|
||||
try:
|
||||
self._log(f"🔄 正在获取客服列表，cookies包含字段: {list(self.cookie.keys())}", "DEBUG")
|
||||
# 🔥 按照原始代码的方式处理响应
|
||||
response = requests.get(url, headers=headers, cookies=self.cookie, params=params).json()
|
||||
self._log(f"客服列表API响应内容: {response}", "DEBUG")
|
||||
|
||||
if response.get('code') == 0:
|
||||
return response.get('data', [])
|
||||
return None
|
||||
staff_data = response.get('data', [])
|
||||
self._log(f"✅ 成功获取客服列表，共 {len(staff_data)} 个客服", "SUCCESS")
|
||||
return staff_data
|
||||
else:
|
||||
error_msg = response.get('message', '未知错误')
|
||||
self._log(f"❌ 客服列表API返回错误: code={response.get('code')}, message={error_msg}", "ERROR")
|
||||
return None
|
||||
except Exception as e:
|
||||
self._log(f"❌ 获取客服列表失败: {e}", "ERROR")
|
||||
import traceback
|
||||
self._log(f"错误详情: {traceback.format_exc()}", "DEBUG")
|
||||
return None
|
||||
|
||||
def transfer_conversation(self, receiver_id, shop_id, staff_id):
|
||||
@@ -282,18 +716,26 @@ class DouYinMessageHandler:
|
||||
try:
|
||||
# 获取客服列表
|
||||
staff_list = self.get_casl()
|
||||
if not staff_list:
|
||||
if staff_list is None:
|
||||
self._log("⚠️ 获取客服列表失败", "WARNING")
|
||||
return False
|
||||
|
||||
# 🔥 处理空客服列表的情况（API成功但无客服）
|
||||
if len(staff_list) == 0:
|
||||
self._log("📋 当前无可分配客服，发送空列表到后端", "INFO")
|
||||
# 继续处理，发送空列表给后端
|
||||
|
||||
# 转换客服数据格式
|
||||
# 转换客服数据格式 - 🔥 根据原始代码调试实际字段名
|
||||
staff_infos = []
|
||||
for staff in staff_list:
|
||||
# 打印原始数据结构用于调试
|
||||
self._log(f"🔍 原始客服数据结构: {staff}", "DEBUG")
|
||||
|
||||
staff_info = StaffInfo(
|
||||
staff_id=str(staff.get('staffId', '')),
|
||||
name=staff.get('staffName', ''),
|
||||
status=staff.get('status', 0),
|
||||
department=staff.get('department', ''),
|
||||
staff_id=str(staff.get('staffId', '') or staff.get('staff_id', '') or staff.get('id', '')),
|
||||
name=staff.get('staffName', '') or staff.get('staff_name', '') or staff.get('name', ''),
|
||||
status=str(staff.get('status', 0) or staff.get('state', 0)),
|
||||
department=staff.get('department', '') or staff.get('dept', ''),
|
||||
online=staff.get('online', True)
|
||||
)
|
||||
staff_infos.append(staff_info.to_dict())
|
||||
@@ -312,9 +754,15 @@ class DouYinMessageHandler:
|
||||
# 发送到后端
|
||||
await self.ai_service.send_message_to_backend(message_template.to_dict())
|
||||
self._log(f"发送客服列表消息的结构体为: {message_template.to_json()}")
|
||||
self._log(f"✅ [DY] 成功发送客服列表到后端，共 {len(staff_infos)} 个客服", "SUCCESS")
|
||||
print(f"🔥 [DY] 客服列表已上传到后端: {len(staff_infos)} 个客服")
|
||||
print(f"[DY] 客服详情: {[{'id': s['staff_id'], 'name': s['name']} for s in staff_infos]}")
|
||||
|
||||
if len(staff_infos) > 0:
|
||||
self._log(f"✅ [DY] 成功发送客服列表到后端，共 {len(staff_infos)} 个客服", "SUCCESS")
|
||||
print(f"🔥 [DY] 客服列表已上传到后端: {len(staff_infos)} 个客服")
|
||||
print(f"[DY] 客服详情: {[{'id': s['staff_id'], 'name': s['name']} for s in staff_infos]}")
|
||||
else:
|
||||
self._log(f"✅ [DY] 成功发送空客服列表到后端（当前无可分配客服）", "SUCCESS")
|
||||
print(f"🔥 [DY] 空客服列表已上传到后端（当前无可分配客服）")
|
||||
|
||||
return True
|
||||
|
||||
except Exception as e:
|
||||
@@ -603,26 +1051,28 @@ class DouYinMessageHandler:
|
||||
talk_id = user_info.get("talk_id", 0)
|
||||
p_id = user_info.get("p_id", 0)
|
||||
|
||||
# 准备发送者和接收者信息
|
||||
sender_info = {
|
||||
"id": str(sender_id),
|
||||
"name": f"用户_{sender_id}",
|
||||
"is_customer": True
|
||||
}
|
||||
receiver_info = {
|
||||
"id": self.store_id,
|
||||
"name": "店铺客服",
|
||||
"is_merchant": True
|
||||
}
|
||||
# 🔥 统一消息类型检测逻辑（与拼多多保持一致）
|
||||
content = message_content
|
||||
lc = str(content).lower()
|
||||
|
||||
# 检测消息类型，使用与拼多多相同的逻辑
|
||||
if any(ext in lc for ext in [".jpg", ".jpeg", ".png", ".gif", ".webp"]):
|
||||
msg_type = "image"
|
||||
elif any(ext in lc for ext in [".mp4", ".avi", ".mov", ".wmv", ".flv"]):
|
||||
msg_type = "video"
|
||||
elif any(keyword in lc for keyword in ['goods.html', 'item.html', 'item.jd.com', '商品卡片id：']):
|
||||
msg_type = "product_card"
|
||||
else:
|
||||
msg_type = "text"
|
||||
|
||||
# 创建消息模板
|
||||
message_template = PlatformMessage(
|
||||
type="message",
|
||||
content=message_content,
|
||||
msg_type="text",
|
||||
sender={"id": sender_info.get("id", "")},
|
||||
# 🔥 使用工厂方法创建消息模板（与拼多多保持一致）
|
||||
message_template = PlatformMessage.create_text_message(
|
||||
content=content,
|
||||
sender_id=str(sender_id),
|
||||
store_id=self.store_id
|
||||
)
|
||||
# 动态设置检测到的消息类型
|
||||
message_template.msg_type = msg_type
|
||||
|
||||
# 发送消息到后端（使用统一连接）
|
||||
await self.ai_service.send_message_to_backend(message_template.to_dict())
|
||||
@@ -1195,22 +1645,60 @@ class DouYinMessageHandler:
|
||||
elif msg_type == 'text' and avatar_url:
|
||||
message_text = message_dict['text']
|
||||
|
||||
# 准备发送者和接收者信息
|
||||
sender_info = {
|
||||
"id": str(sender_id),
|
||||
"name": f"用户_{sender_id}",
|
||||
"is_customer": True
|
||||
}
|
||||
# 🔥 提取goods_info信息（与拼多多保持一致）
|
||||
goods_info = {}
|
||||
if msg_type == 'order':
|
||||
# 从抖音的订单信息中提取goods_info
|
||||
order_id = message_dict.get('order_id', '')
|
||||
goods_id = self.get_goods_id(order_id) if hasattr(self, 'get_goods_id') else message_dict.get('goods_id', '')
|
||||
goods_info = {
|
||||
'goodsID': goods_id,
|
||||
'orderSequenceNo': order_id
|
||||
}
|
||||
elif msg_type == 'goods':
|
||||
# 从抖音的商品信息中提取goods_info
|
||||
goods_id = message_dict.get('goods_id', '')
|
||||
goods_info = {
|
||||
'goodsID': goods_id
|
||||
}
|
||||
|
||||
# 创建消息模板
|
||||
message_template = PlatformMessage(
|
||||
type="message",
|
||||
content=message_text,
|
||||
pin_image=avatar_url,
|
||||
msg_type=msg_type,
|
||||
sender={"id": sender_info.get("id", "")},
|
||||
store_id=self.store_id
|
||||
# 🔥 统一消息类型检测逻辑（与拼多多完全一致）
|
||||
content = message_text
|
||||
lc = str(content).lower()
|
||||
|
||||
# 使用与拼多多完全相同的检测逻辑
|
||||
if any(ext in lc for ext in [".jpg", ".jpeg", ".png", ".gif", ".webp"]):
|
||||
msg_type = "image"
|
||||
elif any(ext in lc for ext in [".mp4", ".avi", ".mov", ".wmv", ".flv"]):
|
||||
msg_type = "video"
|
||||
else:
|
||||
msg_type = "text"
|
||||
|
||||
# 🔥 订单卡片组装（与拼多多完全一致）
|
||||
if ("订单编号" in str(content) or msg_type == 'order') and goods_info:
|
||||
content = f"商品id：{goods_info.get('goodsID')} 订单号：{goods_info.get('orderSequenceNo')}"
|
||||
msg_type = "order_card"
|
||||
|
||||
# 🔥 商品卡片检测（与拼多多完全一致）
|
||||
elif any(keyword in lc for keyword in ['goods.html', 'item.html', 'item.jd.com', '商品卡片id：']) or \
|
||||
(goods_info and goods_info.get('goodsID') and not goods_info.get('orderSequenceNo')):
|
||||
msg_type = "product_card"
|
||||
|
||||
# 🔥 使用工厂方法创建消息模板（与拼多多保持一致）
|
||||
message_template = PlatformMessage.create_text_message(
|
||||
content=content,
|
||||
sender_id=str(sender_id),
|
||||
store_id=self.store_id,
|
||||
pin_image=avatar_url
|
||||
)
|
||||
# 动态设置检测到的消息类型
|
||||
message_template.msg_type = msg_type
|
||||
|
||||
# 🔥 添加调试日志（与拼多多保持一致）
|
||||
try:
|
||||
print(f"📤(SPEC) 发送到AI: {json.dumps(message_template.to_dict(), ensure_ascii=False)[:300]}...")
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
self._log("📤 准备发送消息到AI服务...", "INFO")
|
||||
self._log(f"📋 消息内容: {message_template.to_json()}", "DEBUG")
|
||||
@@ -1838,6 +2326,8 @@ class DouYinListenerForGUI:
|
||||
|
||||
# 发送客服列表到后端
|
||||
try:
|
||||
# 🔥 等待一小段时间确保连接完全建立
|
||||
await asyncio.sleep(1)
|
||||
staff_list_success = await self.douyin_bot.message_handler.send_staff_list_to_backend()
|
||||
if staff_list_success:
|
||||
print(f"🔥 [DY] 客服列表已上传到后端")
|
||||
@@ -1876,6 +2366,69 @@ class DouYinListenerForGUI:
|
||||
self._log(f"错误详情: {traceback.format_exc()}", "DEBUG")
|
||||
return False
|
||||
|
||||
async def start_with_login_params(self, store_id: str, login_params: str):
|
||||
"""使用后端下发的登录参数执行登录并启动监听（与拼多多保持一致）"""
|
||||
try:
|
||||
self._log("🔵 [DY] 收到后端登录参数，开始执行登录获取cookies", "INFO")
|
||||
self._log(f"🔍 [DY] 登录参数内容: {login_params[:100]}...", "DEBUG")
|
||||
|
||||
# 1. 解析登录参数
|
||||
self._log("🔄 [DY] 开始解析登录参数", "DEBUG")
|
||||
params_dict = self._parse_login_params(login_params)
|
||||
if not params_dict:
|
||||
self._log("❌ [DY] 登录参数解析失败", "ERROR")
|
||||
return False
|
||||
self._log(f"✅ [DY] 登录参数解析成功，手机号: {params_dict.get('phone_number', 'N/A')}", "DEBUG")
|
||||
|
||||
# 2. 使用新的DyLogin类执行登录
|
||||
self._log("🔄 [DY] 开始创建DyLogin实例", "DEBUG")
|
||||
dy_login = DyLogin(log_callback=self.log_callback)
|
||||
self._log("✅ [DY] DyLogin实例创建成功", "DEBUG")
|
||||
|
||||
self._log("🔄 [DY] 开始执行登录", "DEBUG")
|
||||
login_result = dy_login.login_with_params(params_dict, store_id)
|
||||
self._log(f"📊 [DY] 登录结果: {login_result}", "DEBUG")
|
||||
|
||||
if login_result == "need_verification_code":
|
||||
self._log("⚠️ [DY] 需要手机验证码，已通知后端，等待重新下发包含验证码的登录参数", "WARNING")
|
||||
return "need_verification_code" # 返回特殊标识，避免被覆盖
|
||||
elif login_result == "verification_code_error":
|
||||
self._log("⚠️ [DY] 验证码错误，已通知后端", "WARNING")
|
||||
return "verification_code_error" # 返回特殊标识，避免重复发送消息
|
||||
elif login_result == "login_failure":
|
||||
self._log("⚠️ [DY] 登录失败，已发送失败通知给后端", "WARNING")
|
||||
return "login_failure" # 返回特殊标识，避免重复发送消息
|
||||
elif not login_result:
|
||||
self._log("❌ [DY] 登录失败", "ERROR")
|
||||
return False
|
||||
elif isinstance(login_result, dict):
|
||||
# 登录成功，获取到cookies
|
||||
self._log("✅ [DY] 登录成功，使用获取的cookies连接平台", "SUCCESS")
|
||||
|
||||
# 🔥 使用获取的cookies启动监听（这里会验证cookies完整性）
|
||||
return await self.start_with_cookies(store_id, login_result)
|
||||
else:
|
||||
self._log("❌ [DY] 登录返回未知结果", "ERROR")
|
||||
return False
|
||||
|
||||
except Exception as e:
|
||||
self._log(f"❌ [DY] 使用登录参数启动失败: {str(e)}", "ERROR")
|
||||
import traceback
|
||||
self._log(f"🔍 [DY] 异常详细信息: {traceback.format_exc()}", "ERROR")
|
||||
return False
|
||||
|
||||
def _parse_login_params(self, login_params_str: str) -> dict:
|
||||
"""解析后端下发的登录参数（与拼多多保持一致）"""
|
||||
try:
|
||||
import json
|
||||
data = json.loads(login_params_str)
|
||||
params = data.get("data", {})
|
||||
self._log(f"✅ [DY] 登录参数解析成功: phone_number={params.get('phone_number', 'N/A')}", "INFO")
|
||||
return params
|
||||
except Exception as e:
|
||||
self._log(f"❌ [DY] 解析登录参数失败: {e}", "ERROR")
|
||||
return {}
|
||||
|
||||
def stop_listening(self):
|
||||
"""停止监听"""
|
||||
if self.stop_event:
|
||||
|
||||
@@ -1151,12 +1151,14 @@ class BackendClient:
|
||||
content = message.get('content', '')
|
||||
data = message.get('data', {})
|
||||
|
||||
# 判断是拼多多登录参数还是普通Cookie
|
||||
if platform_name == "拼多多" and ("拼多多登录" in content) and data.get('login_params'):
|
||||
# 拼多多登录参数模式 - 传递完整的消息JSON给处理器
|
||||
print(f"收到拼多多登录参数: 平台={platform_name}, 店铺={store_id}, 类型={content}")
|
||||
# 🔥 判断是登录参数模式还是普通Cookie模式（支持拼多多和抖音）
|
||||
if (platform_name in ["拼多多", "抖音"] and
|
||||
(("拼多多登录" in content and data.get('login_params')) or
|
||||
("抖音登录" in content and data.get('login_flow')))):
|
||||
# 登录参数模式 - 传递完整的消息JSON给处理器
|
||||
print(f"收到{platform_name}登录参数: 平台={platform_name}, 店铺={store_id}, 类型={content}")
|
||||
if self.login_callback:
|
||||
# 传递完整的JSON消息，让拼多多处理器来解析login_params
|
||||
# 传递完整的JSON消息，让处理器来解析登录参数
|
||||
import json
|
||||
full_message = json.dumps(message)
|
||||
self.login_callback(platform_name, store_id, full_message)
|
||||
|
||||
@@ -275,15 +275,66 @@ class WebSocketManager:
|
||||
def _runner():
|
||||
try:
|
||||
import json
|
||||
self._log("🚀 开始创建抖音监听器实例", "DEBUG")
|
||||
listener = DYListenerForGUI_WS()
|
||||
# 将JSON字符串格式的cookies解析为字典
|
||||
try:
|
||||
cookie_dict = json.loads(cookies) if isinstance(cookies, str) else cookies
|
||||
except json.JSONDecodeError as e:
|
||||
self._log(f"❌ Cookie JSON解析失败: {e}", "ERROR")
|
||||
return False
|
||||
|
||||
result = asyncio.run(listener.start_with_cookies(store_id=store_id, cookie_dict=cookie_dict))
|
||||
self._log("✅ 抖音监听器实例创建成功", "DEBUG")
|
||||
|
||||
# 🔥 检查是否为登录参数模式（与拼多多保持一致）
|
||||
if cookies and ('"login_flow"' in cookies or '"phone_number"' in cookies):
|
||||
# 使用登录参数模式
|
||||
self._log("📋 使用登录参数启动抖音监听器", "INFO")
|
||||
self._log("🔄 开始执行 start_with_login_params", "DEBUG")
|
||||
result = asyncio.run(listener.start_with_login_params(store_id=store_id, login_params=cookies))
|
||||
self._log(f"📊 start_with_login_params 执行结果: {result}", "DEBUG")
|
||||
|
||||
# 🔥 详细的结果分析（与拼多多完全一致）
|
||||
if result == "need_verification_code":
|
||||
self._log("✅ [DY] 登录流程正常，已发送验证码需求通知给后端", "SUCCESS")
|
||||
elif result == "verification_code_error":
|
||||
self._log("⚠️ [DY] 验证码错误，已发送错误通知给后端", "WARNING")
|
||||
elif result:
|
||||
self._log("✅ [DY] 登录成功，平台连接已建立", "SUCCESS")
|
||||
self._notify_platform_connected("抖音")
|
||||
else:
|
||||
self._log("❌ [DY] 登录失败", "ERROR")
|
||||
else:
|
||||
# 传统cookie模式（保留兼容性）
|
||||
self._log("🍪 使用Cookie启动抖音监听器", "INFO")
|
||||
self._log("🔄 开始执行 start_with_cookies", "DEBUG")
|
||||
try:
|
||||
cookie_dict = json.loads(cookies) if isinstance(cookies, str) else cookies
|
||||
except json.JSONDecodeError as e:
|
||||
self._log(f"❌ Cookie JSON解析失败: {e}", "ERROR")
|
||||
return False
|
||||
result = asyncio.run(listener.start_with_cookies(store_id=store_id, cookie_dict=cookie_dict))
|
||||
self._log(f"📊 start_with_cookies 执行结果: {result}", "DEBUG")
|
||||
|
||||
# Cookie启动成功时也要通知GUI
|
||||
if result:
|
||||
self._log("✅ [DY] Cookie启动成功，平台连接已建立", "SUCCESS")
|
||||
self._notify_platform_connected("抖音")
|
||||
|
||||
# 🔥 根据实际登录结果上报状态给后端（与拼多多完全一致）
|
||||
if self.backend_client and result not in ["need_verification_code", "verification_code_error", "login_failure"]:
|
||||
# 如果是特殊状态，说明通知已经在DyLogin中发送了，不需要重复发送
|
||||
try:
|
||||
message = {
|
||||
"type": "connect_message",
|
||||
"store_id": store_id,
|
||||
"status": bool(result)
|
||||
}
|
||||
self.backend_client.send_message(message)
|
||||
status_text = "成功" if result else "失败"
|
||||
self._log(f"上报抖音平台连接状态{status_text}: {message}", "SUCCESS" if result else "ERROR")
|
||||
except Exception as send_e:
|
||||
self._log(f"上报抖音平台连接状态失败: {send_e}", "ERROR")
|
||||
elif result == "need_verification_code":
|
||||
self._log("需要验证码，验证码通知已由DyLogin发送，等待后端重新下发登录参数", "INFO")
|
||||
elif result == "verification_code_error":
|
||||
self._log("验证码错误，错误通知已由DyLogin发送，等待后端处理", "INFO")
|
||||
elif result == "login_failure":
|
||||
self._log("登录失败，失败通知已由DyLogin发送，等待后端处理", "INFO")
|
||||
|
||||
return result
|
||||
except Exception as e:
|
||||
self._log(f"抖音监听器运行异常: {e}", "ERROR")
|
||||
@@ -304,20 +355,7 @@ class WebSocketManager:
|
||||
if f"抖音:{store_id}" in self.platform_listeners:
|
||||
self.platform_listeners[f"抖音:{store_id}"]['status'] = 'success'
|
||||
|
||||
# 上报连接状态给后端
|
||||
if self.backend_client:
|
||||
try:
|
||||
self.backend_client.send_message({
|
||||
"type": "connect_message",
|
||||
"store_id": store_id,
|
||||
"status": True
|
||||
})
|
||||
self._log("已上报抖音平台连接状态: 成功", "INFO")
|
||||
except Exception as e:
|
||||
self._log(f"上报抖音平台连接状态失败: {e}", "WARNING")
|
||||
|
||||
self._log("已启动抖音平台监听", "SUCCESS")
|
||||
self._notify_platform_connected("抖音") # ← 新增
|
||||
|
||||
except Exception as e:
|
||||
self._log(f"启动抖音平台监听失败: {e}", "ERROR")
|
||||
|
||||
Reference in New Issue
Block a user