拼多多集成
This commit is contained in:
809
WebSocket/BackendClient.py
Normal file
809
WebSocket/BackendClient.py
Normal file
@@ -0,0 +1,809 @@
|
||||
# WebSocket/BackendClient.py
|
||||
import json
|
||||
import threading
|
||||
|
||||
import websockets
|
||||
|
||||
import uuid
|
||||
import asyncio
|
||||
from typing import List, Dict, Any, Optional, Callable
|
||||
from config import get_gui_ws_url
|
||||
|
||||
|
||||
class BackendClient:
|
||||
"""后端WebSocket客户端"""
|
||||
|
||||
def __init__(self, url: str, token: str = None):
|
||||
|
||||
self.token = token
|
||||
self.uuid = str(uuid.uuid4())
|
||||
self.url = url
|
||||
|
||||
# 消息处理回调函数
|
||||
self.store_list_callback: Optional[Callable] = None
|
||||
self.customers_callback: Optional[Callable] = None
|
||||
self.message_callback: Optional[Callable] = None
|
||||
self.transfer_callback: Optional[Callable] = None
|
||||
self.close_callback: Optional[Callable] = None
|
||||
self.error_callback: Optional[Callable] = None
|
||||
self.login_callback: Optional[Callable] = None # 新增：平台登录（下发cookies）回调
|
||||
self.success_callback: Optional[Callable] = None # 新增：后端连接成功回调
|
||||
|
||||
self.is_connected = False
|
||||
|
||||
def connect(self):
|
||||
"""连接到WebSocket服务器"""
|
||||
if self.is_connected:
|
||||
return
|
||||
|
||||
self.thread = threading.Thread(target=self._run_loop, daemon=True)
|
||||
self.thread.start()
|
||||
|
||||
def disconnect(self):
|
||||
"""断开WebSocket连接"""
|
||||
if self.loop and self.loop.is_running():
|
||||
asyncio.run_coroutine_threadsafe(self._close(), self.loop)
|
||||
|
||||
|
||||
if self.thread and self.thread.is_alive():
|
||||
self.thread.join(timeout=3)
|
||||
|
||||
self.is_connected = False
|
||||
|
||||
async def _close(self):
|
||||
"""异步关闭连接"""
|
||||
if self.websocket:
|
||||
await self.websocket.close()
|
||||
self.is_connected = False
|
||||
|
||||
def _run_loop(self):
|
||||
"""在新线程中运行事件循环"""
|
||||
self.loop = asyncio.new_event_loop()
|
||||
asyncio.set_event_loop(self.loop)
|
||||
|
||||
try:
|
||||
self.loop.run_until_complete(self._connect_and_listen())
|
||||
except Exception as e:
|
||||
print(f"WebSocket异常: {e}")
|
||||
finally:
|
||||
self.loop.close()
|
||||
|
||||
async def _connect_and_listen(self):
|
||||
"""连接并监听消息"""
|
||||
try:
|
||||
self.websocket = await websockets.connect(self.url)
|
||||
self.is_connected = True
|
||||
self.on_connected()
|
||||
|
||||
async for message in self.websocket:
|
||||
try:
|
||||
# 打印原始文本帧与长度
|
||||
try:
|
||||
raw_len = len(message.encode('utf-8')) if isinstance(message, str) else len(message)
|
||||
print(f"后端发送消息体内容：{message}")
|
||||
except Exception:
|
||||
pass
|
||||
data = json.loads(message)
|
||||
self.on_message_received(data)
|
||||
except json.JSONDecodeError:
|
||||
print(f"JSON解析错误: {message}")
|
||||
|
||||
except Exception as e:
|
||||
self.is_connected = False
|
||||
self.on_error(str(e))
|
||||
|
||||
@classmethod
|
||||
def from_exe_token(cls, exe_token: str):
|
||||
"""使用 exe_token 构造单连接客户端（ws/gui/<token>/）"""
|
||||
url = get_gui_ws_url(exe_token)
|
||||
return cls(url=url, token=exe_token)
|
||||
|
||||
def set_callbacks(self,
|
||||
store_list: Callable = None,
|
||||
customers: Callable = None,
|
||||
message: Callable = None,
|
||||
transfer: Callable = None,
|
||||
close: Callable = None,
|
||||
error: Callable = None,
|
||||
login: Callable = None,
|
||||
success: Callable = None):
|
||||
"""设置各种消息类型的回调函数"""
|
||||
if store_list:
|
||||
self.store_list_callback = store_list
|
||||
if customers:
|
||||
self.customers_callback = customers
|
||||
if message:
|
||||
self.message_callback = message
|
||||
if transfer:
|
||||
self.transfer_callback = transfer
|
||||
if close:
|
||||
self.close_callback = close
|
||||
if error:
|
||||
self.error_callback = error
|
||||
if login:
|
||||
self.login_callback = login
|
||||
if success:
|
||||
self.success_callback = success
|
||||
|
||||
def on_connected(self):
|
||||
"""连接成功时的处理"""
|
||||
print("后端WebSocket连接成功")
|
||||
# 不再主动请求 get_store，避免与后端不兼容导致协程未完成
|
||||
|
||||
def on_message_received(self, message: Dict[str, Any]):
|
||||
"""处理接收到的消息 - 根据WebSocket文档v2更新"""
|
||||
# 统一打印后端下发的完整消息结构体
|
||||
try:
|
||||
import json as _json
|
||||
print("=== Backend -> GUI Message ===")
|
||||
print(_json.dumps(message, ensure_ascii=False, indent=2))
|
||||
print("=== End Message ===")
|
||||
except Exception:
|
||||
pass
|
||||
msg_type = message.get('type')
|
||||
|
||||
try:
|
||||
if msg_type == 'get_store':
|
||||
self._handle_store_list(message)
|
||||
elif msg_type == 'get_customers':
|
||||
self._handle_customers_list(message)
|
||||
elif msg_type == 'success':
|
||||
print("后端连接服务成功")
|
||||
# 可在此触发上层UI通知
|
||||
if self.success_callback:
|
||||
try:
|
||||
self.success_callback()
|
||||
except Exception:
|
||||
pass
|
||||
elif msg_type == 'message':
|
||||
self._handle_message(message)
|
||||
elif msg_type == 'transfer':
|
||||
self._handle_transfer(message)
|
||||
elif msg_type == 'close':
|
||||
self._handle_close(message)
|
||||
elif msg_type == 'pong':
|
||||
self._handle_pong(message)
|
||||
elif msg_type == 'connect_success': # 兼容旧版
|
||||
self._handle_connect_success(message)
|
||||
elif msg_type == 'login': # 新版：后台下发平台cookies
|
||||
self._handle_login(message)
|
||||
elif msg_type == 'error':
|
||||
self._handle_error_message(message)
|
||||
elif msg_type == 'staff_list':
|
||||
self._handle_staff_list(message)
|
||||
else:
|
||||
print(f"未知消息类型: {msg_type}")
|
||||
|
||||
except Exception as e:
|
||||
error_msg = f"处理消息异常: {e}"
|
||||
print(error_msg)
|
||||
if self.error_callback:
|
||||
self.error_callback(error_msg, message)
|
||||
|
||||
def on_error(self, error: str):
|
||||
"""错误处理"""
|
||||
print(f"后端连接错误: {error}")
|
||||
if self.error_callback:
|
||||
self.error_callback(error, None)
|
||||
|
||||
# ==================== 发送消息方法 ====================
|
||||
|
||||
def send_message(self, message: Dict[str, Any]):
|
||||
"""
|
||||
发送消息到后端
|
||||
|
||||
Args:
|
||||
message: 要发送的消息字典
|
||||
"""
|
||||
if not self.is_connected or not self.loop:
|
||||
raise Exception("WebSocket未连接")
|
||||
|
||||
future = asyncio.run_coroutine_threadsafe(
|
||||
self._send_to_backend(message), self.loop
|
||||
)
|
||||
return future.result(timeout=8)
|
||||
|
||||
async def _send_to_backend(self, message: Dict[str, Any]):
|
||||
"""异步发送消息到后端"""
|
||||
if not self.websocket:
|
||||
raise Exception("WebSocket连接不存在")
|
||||
|
||||
import json
|
||||
message_str = json.dumps(message, ensure_ascii=False)
|
||||
await self.websocket.send(message_str)
|
||||
print(f"发送消息到后端: {message}")
|
||||
|
||||
def send_ping(self, custom_uuid: str = None, custom_token: str = None):
|
||||
"""
|
||||
发送心跳包
|
||||
如果接收到关闭的消息后，心跳包要带上token
|
||||
"""
|
||||
ping_message = {
|
||||
'type': 'ping',
|
||||
'uuid': custom_uuid or self.uuid
|
||||
}
|
||||
|
||||
token = custom_token or self.token
|
||||
if token:
|
||||
ping_message['token'] = token
|
||||
|
||||
return self.send_message(ping_message)
|
||||
|
||||
def get_store(self):
|
||||
"""获取店铺信息"""
|
||||
message = {
|
||||
'type': 'get_store',
|
||||
'token': self.token or ''
|
||||
}
|
||||
return self.send_message(message)
|
||||
|
||||
def send_text_message(self, content: str, sender_id: str, store_id: str, pin_image: str = None):
|
||||
"""发送文本消息 - 根据WebSocket文档v2更新"""
|
||||
message = {
|
||||
'type': 'message',
|
||||
'content': content,
|
||||
'msg_type': 'text',
|
||||
'sender': {'id': sender_id},
|
||||
'store_id': store_id
|
||||
}
|
||||
if pin_image:
|
||||
message['pin_image'] = pin_image
|
||||
return self.send_message(message)
|
||||
|
||||
def send_image_message(self, image_url: str, sender_id: str, store_id: str, pin_image: str = None):
|
||||
"""发送图片消息 - 根据WebSocket文档v2更新"""
|
||||
message = {
|
||||
'type': 'message',
|
||||
'content': image_url,
|
||||
'msg_type': 'image',
|
||||
'sender': {'id': sender_id},
|
||||
'store_id': store_id
|
||||
}
|
||||
if pin_image:
|
||||
message['pin_image'] = pin_image
|
||||
return self.send_message(message)
|
||||
|
||||
def send_video_message(self, video_url: str, sender_id: str, store_id: str, pin_image: str = None):
|
||||
"""发送视频消息 - 根据WebSocket文档v2更新"""
|
||||
message = {
|
||||
'type': 'message',
|
||||
'content': video_url,
|
||||
'msg_type': 'video',
|
||||
'sender': {'id': sender_id},
|
||||
'store_id': store_id
|
||||
}
|
||||
if pin_image:
|
||||
message['pin_image'] = pin_image
|
||||
return self.send_message(message)
|
||||
|
||||
def send_order_card(self, product_id: str, order_number: str, sender_id: str, store_id: str, pin_image: str = None):
|
||||
"""发送订单卡片 - 根据WebSocket文档v2更新"""
|
||||
message = {
|
||||
'type': 'message',
|
||||
'content': f'商品id：{product_id} 订单号：{order_number}',
|
||||
'msg_type': 'order_card',
|
||||
'sender': {'id': sender_id},
|
||||
'store_id': store_id
|
||||
}
|
||||
if pin_image:
|
||||
message['pin_image'] = pin_image
|
||||
return self.send_message(message)
|
||||
|
||||
def send_product_card(self, product_url: str, sender_id: str, store_id: str, pin_image: str = None):
|
||||
"""发送商品卡片 - 根据WebSocket文档v2更新"""
|
||||
message = {
|
||||
'type': 'message',
|
||||
'content': product_url,
|
||||
'msg_type': 'product_card',
|
||||
'sender': {'id': sender_id},
|
||||
'store_id': store_id
|
||||
}
|
||||
if pin_image:
|
||||
message['pin_image'] = pin_image
|
||||
return self.send_message(message)
|
||||
|
||||
def send_staff_list(self, staff_list: List[Dict], store_id: str):
|
||||
"""发送客服列表 - 根据WebSocket文档v2更新"""
|
||||
message = {
|
||||
'type': 'staff_list',
|
||||
'content': '客服列表更新',
|
||||
'data': {'staff_list': staff_list},
|
||||
'store_id': store_id
|
||||
}
|
||||
return self.send_message(message)
|
||||
|
||||
# 保持向后兼容的旧方法（标记为已废弃）
|
||||
def send_file_message(self, content: str, uid: str, pin: str, store_id: str):
|
||||
"""发送文件消息 - 已废弃，请使用send_video_message或send_image_message"""
|
||||
print("警告: send_file_message已废弃，请根据文件类型使用send_video_message或send_image_message")
|
||||
# 尝试自动检测文件类型
|
||||
content_lower = content.lower()
|
||||
if any(ext in content_lower for ext in ['.mp4', '.avi', '.mov', '.wmv', '.flv']):
|
||||
return self.send_video_message(content, uid, store_id)
|
||||
elif any(ext in content_lower for ext in ['.jpg', '.jpeg', '.png', '.gif', '.webp']):
|
||||
return self.send_image_message(content, uid, store_id)
|
||||
else:
|
||||
# 默认作为文本消息发送
|
||||
return self.send_text_message(content, uid, store_id)
|
||||
|
||||
def send_transfer(self, customer: str, pin: str, store_id: str):
|
||||
"""发送转接消息"""
|
||||
message = {
|
||||
'type': 'transfer',
|
||||
'customer': customer, # 客服名称
|
||||
'pin': pin, # 顾客名称
|
||||
'store_id': store_id
|
||||
}
|
||||
return self.send_message(message)
|
||||
|
||||
def close_store(self, store_id: str):
|
||||
"""关闭店铺"""
|
||||
message = {
|
||||
'type': 'close',
|
||||
'store_id': store_id
|
||||
}
|
||||
return self.send_message(message)
|
||||
|
||||
# ==================== 消息处理方法 ====================
|
||||
|
||||
def _handle_store_list(self, message: Dict[str, Any]):
|
||||
"""处理店铺列表"""
|
||||
store_list = message.get('store_list', [])
|
||||
print(f"获取到{len(store_list)}个店铺:")
|
||||
|
||||
for store in store_list:
|
||||
merchant_name = store.get('merchant_name', '')
|
||||
store_id = store.get('store_id', '')
|
||||
store_platform = store.get('store_platform', '')
|
||||
print(f" - {merchant_name} (ID: {store_id}, 平台: {store_platform})")
|
||||
|
||||
if self.store_list_callback:
|
||||
self.store_list_callback(store_list)
|
||||
|
||||
def _handle_customers_list(self, message: Dict[str, Any]):
|
||||
"""处理客服列表"""
|
||||
customers = message.get('customers', [])
|
||||
store_id = message.get('store_id', '')
|
||||
|
||||
print(f"店铺{store_id}的客服列表，共{len(customers)}个客服:")
|
||||
for customer in customers:
|
||||
pin = customer.get('pin', '')
|
||||
nickname = customer.get('nickname', '')
|
||||
print(f" - {nickname} ({pin})")
|
||||
|
||||
if self.customers_callback:
|
||||
self.customers_callback(customers, store_id)
|
||||
|
||||
def _handle_message(self, message: Dict[str, Any]):
|
||||
"""处理消息"""
|
||||
store_id = message.get('store_id', '')
|
||||
data = message.get('data')
|
||||
content = message.get('content', '')
|
||||
print(f"[{store_id}] [{message.get('msg_type', 'unknown')}] : {content}")
|
||||
|
||||
# 尝试将后端AI/客服回复转发到对应平台
|
||||
try:
|
||||
receiver = message.get('receiver') or (data.get('receiver') if isinstance(data, dict) else None) or {}
|
||||
recv_pin = receiver.get('id')
|
||||
if recv_pin and store_id:
|
||||
# 根据store_id动态确定平台类型
|
||||
platform_type = self._get_platform_by_store_id(store_id)
|
||||
|
||||
if platform_type == "京东":
|
||||
self._forward_to_jd(store_id, recv_pin, content)
|
||||
elif platform_type == "抖音":
|
||||
self._forward_to_douyin(store_id, recv_pin, content)
|
||||
elif platform_type == "千牛":
|
||||
self._forward_to_qianniu(store_id, recv_pin, content)
|
||||
elif platform_type == "拼多多":
|
||||
self._forward_to_pdd(store_id, recv_pin, content)
|
||||
else:
|
||||
print(f"[Forward] 未知平台类型或未找到店铺: {platform_type}, store_id={store_id}")
|
||||
except Exception as e:
|
||||
print(f"转发到平台失败: {e}")
|
||||
|
||||
if self.message_callback:
|
||||
self.message_callback(data if data else message, store_id)
|
||||
|
||||
def _get_platform_by_store_id(self, store_id: str) -> str:
|
||||
"""根据店铺ID获取平台类型"""
|
||||
try:
|
||||
# 从WebSocket管理器获取平台信息
|
||||
from WebSocket.backend_singleton import get_websocket_manager
|
||||
manager = get_websocket_manager()
|
||||
if manager and hasattr(manager, 'platform_listeners'):
|
||||
for key, listener_info in manager.platform_listeners.items():
|
||||
if listener_info.get('store_id') == store_id:
|
||||
return listener_info.get('platform', '')
|
||||
return ""
|
||||
except Exception as e:
|
||||
print(f"获取平台类型失败: {e}")
|
||||
return ""
|
||||
|
||||
def _forward_to_jd(self, store_id: str, recv_pin: str, content: str):
|
||||
"""转发消息到京东平台"""
|
||||
try:
|
||||
from Utils.JD.JdUtils import WebsocketManager as JDWSManager
|
||||
jd_mgr = JDWSManager()
|
||||
shop_key = f"京东:{store_id}"
|
||||
entry = jd_mgr.get_connection(shop_key)
|
||||
if not entry:
|
||||
print(f"[JD Forward] 未找到连接: {shop_key}")
|
||||
return
|
||||
|
||||
platform_info = (entry or {}).get('platform') or {}
|
||||
ws = platform_info.get('ws')
|
||||
aid = platform_info.get('aid')
|
||||
pin_zj = platform_info.get('pin_zj')
|
||||
vender_id = platform_info.get('vender_id')
|
||||
loop = platform_info.get('loop')
|
||||
|
||||
print(f"[JD Forward] shop_key={shop_key} has_ws={bool(ws)} aid={aid} pin_zj={pin_zj} vender_id={vender_id} has_loop={bool(loop)} recv_pin={recv_pin}")
|
||||
|
||||
if ws and aid and pin_zj and vender_id and loop and content:
|
||||
async def _send():
|
||||
import hashlib as _hashlib
|
||||
import time as _time
|
||||
import json as _json
|
||||
msg = {
|
||||
"ver": "4.3",
|
||||
"type": "chat_message",
|
||||
"from": {"pin": pin_zj, "app": "im.waiter", "clientType": "comet"},
|
||||
"to": {"app": "im.customer", "pin": recv_pin},
|
||||
"id": _hashlib.md5(str(int(_time.time() * 1000)).encode()).hexdigest(),
|
||||
"lang": "zh_CN",
|
||||
"aid": aid,
|
||||
"timestamp": int(_time.time() * 1000),
|
||||
"readFlag": 0,
|
||||
"body": {
|
||||
"content": content,
|
||||
"translated": False,
|
||||
"param": {"cusVenderId": vender_id},
|
||||
"type": "text"
|
||||
}
|
||||
}
|
||||
await ws.send(_json.dumps(msg))
|
||||
|
||||
import asyncio as _asyncio
|
||||
_future = _asyncio.run_coroutine_threadsafe(_send(), loop)
|
||||
try:
|
||||
_future.result(timeout=2)
|
||||
print(f"[JD Forward] 已转发到平台: pin={recv_pin}, content_len={len(content)}")
|
||||
except Exception as fe:
|
||||
print(f"[JD Forward] 转发提交失败: {fe}")
|
||||
else:
|
||||
print("[JD Forward] 条件不足，未转发：",
|
||||
{
|
||||
'has_ws': bool(ws), 'has_aid': bool(aid), 'has_pin_zj': bool(pin_zj),
|
||||
'has_vender_id': bool(vender_id), 'has_loop': bool(loop), 'has_content': bool(content)
|
||||
})
|
||||
except Exception as e:
|
||||
print(f"[JD Forward] 转发失败: {e}")
|
||||
|
||||
def _forward_to_douyin(self, store_id: str, recv_pin: str, content: str):
|
||||
"""转发消息到抖音平台"""
|
||||
try:
|
||||
from Utils.Dy.DyUtils import DouYinWebsocketManager
|
||||
dy_mgr = DouYinWebsocketManager()
|
||||
shop_key = f"抖音:{store_id}"
|
||||
entry = dy_mgr.get_connection(shop_key)
|
||||
|
||||
if not entry:
|
||||
print(f"[DY Forward] 未找到连接: {shop_key}")
|
||||
return
|
||||
|
||||
platform_info = entry.get('platform', {})
|
||||
douyin_bot = platform_info.get('douyin_bot')
|
||||
message_handler = platform_info.get('message_handler')
|
||||
|
||||
print(f"[DY Forward] shop_key={shop_key} has_bot={bool(douyin_bot)} has_handler={bool(message_handler)} recv_pin={recv_pin}")
|
||||
|
||||
if douyin_bot and message_handler and content:
|
||||
# 在消息处理器的事件循环中发送消息
|
||||
def send_in_loop():
|
||||
try:
|
||||
# 获取消息处理器的事件循环
|
||||
loop = message_handler._loop
|
||||
if loop and not loop.is_closed():
|
||||
# 在事件循环中执行发送
|
||||
future = asyncio.run_coroutine_threadsafe(
|
||||
message_handler.send_message_external(recv_pin, content),
|
||||
loop
|
||||
)
|
||||
# 等待结果
|
||||
try:
|
||||
result = future.result(timeout=5)
|
||||
if result:
|
||||
print(f"[DY Forward] 已转发到平台: pin={recv_pin}, content_len={len(content)}")
|
||||
else:
|
||||
print(f"[DY Forward] 转发失败: 消息处理器返回False")
|
||||
except Exception as fe:
|
||||
print(f"[DY Forward] 转发执行失败: {fe}")
|
||||
else:
|
||||
print(f"[DY Forward] 事件循环不可用")
|
||||
except Exception as e:
|
||||
print(f"[DY Forward] 发送过程异常: {e}")
|
||||
|
||||
# 在新线程中执行发送操作
|
||||
import threading
|
||||
send_thread = threading.Thread(target=send_in_loop, daemon=True)
|
||||
send_thread.start()
|
||||
|
||||
else:
|
||||
print("[DY Forward] 条件不足，未转发：",
|
||||
{
|
||||
'has_bot': bool(douyin_bot),
|
||||
'has_handler': bool(message_handler),
|
||||
'has_content': bool(content)
|
||||
})
|
||||
except Exception as e:
|
||||
print(f"[DY Forward] 转发失败: {e}")
|
||||
|
||||
def _forward_to_qianniu(self, store_id: str, recv_pin: str, content: str):
|
||||
"""转发消息到千牛平台"""
|
||||
try:
|
||||
# TODO: 实现千牛平台的消息转发逻辑
|
||||
print(f"[QN Forward] 千牛平台消息转发功能待实现: store_id={store_id}, recv_pin={recv_pin}, content={content}")
|
||||
except Exception as e:
|
||||
print(f"[QN Forward] 转发失败: {e}")
|
||||
|
||||
def _forward_to_pdd(self, store_id: str, recv_pin: str, content: str):
|
||||
"""转发消息到拼多多平台"""
|
||||
try:
|
||||
from Utils.Pdd.PddUtils import WebsocketManager as PDDWSManager
|
||||
pdd_mgr = PDDWSManager()
|
||||
shop_key = f"拼多多:{store_id}"
|
||||
entry = pdd_mgr.get_connection(shop_key)
|
||||
|
||||
if not entry:
|
||||
print(f"[PDD Forward] 未找到连接: {shop_key}")
|
||||
return
|
||||
|
||||
platform_info = entry.get('platform', {})
|
||||
pdd_instance = platform_info.get('pdd_instance')
|
||||
loop = platform_info.get('loop')
|
||||
|
||||
print(f"[PDD Forward] shop_key={shop_key} has_pdd_instance={bool(pdd_instance)} has_loop={bool(loop)} recv_pin={recv_pin}")
|
||||
|
||||
if pdd_instance and loop and content:
|
||||
# 在拼多多实例的事件循环中发送消息
|
||||
def send_in_loop():
|
||||
try:
|
||||
# 在事件循环中执行发送
|
||||
future = asyncio.run_coroutine_threadsafe(
|
||||
pdd_instance.send_message_external(recv_pin, content),
|
||||
loop
|
||||
)
|
||||
# 等待结果
|
||||
try:
|
||||
result = future.result(timeout=10) # 拼多多可能需要更长时间
|
||||
if result:
|
||||
print(f"[PDD Forward] 已转发到平台: uid={recv_pin}, content_len={len(content)}")
|
||||
else:
|
||||
print(f"[PDD Forward] 转发失败: 拼多多实例返回False")
|
||||
except Exception as fe:
|
||||
print(f"[PDD Forward] 转发执行失败: {fe}")
|
||||
except Exception as e:
|
||||
print(f"[PDD Forward] 发送过程异常: {e}")
|
||||
|
||||
# 在新线程中执行发送操作
|
||||
import threading
|
||||
send_thread = threading.Thread(target=send_in_loop, daemon=True)
|
||||
send_thread.start()
|
||||
|
||||
else:
|
||||
print("[PDD Forward] 条件不足，未转发：",
|
||||
{
|
||||
'has_pdd_instance': bool(pdd_instance),
|
||||
'has_loop': bool(loop),
|
||||
'has_content': bool(content)
|
||||
})
|
||||
except Exception as e:
|
||||
print(f"[PDD Forward] 转发失败: {e}")
|
||||
|
||||
def _transfer_to_pdd(self, customer_service_id: str, user_id: str, store_id: str):
|
||||
"""执行拼多多平台转接操作"""
|
||||
try:
|
||||
from Utils.Pdd.PddUtils import WebsocketManager as PDDWSManager
|
||||
pdd_mgr = PDDWSManager()
|
||||
shop_key = f"拼多多:{store_id}"
|
||||
entry = pdd_mgr.get_connection(shop_key)
|
||||
|
||||
if not entry:
|
||||
print(f"[PDD Transfer] 未找到拼多多连接: {shop_key}")
|
||||
return
|
||||
|
||||
platform_info = entry.get('platform', {})
|
||||
pdd_instance = platform_info.get('pdd_instance')
|
||||
loop = platform_info.get('loop')
|
||||
|
||||
print(f"[PDD Transfer] 找到拼多多连接，准备执行转接: user_id={user_id}, cs_id={customer_service_id}")
|
||||
|
||||
if pdd_instance and loop:
|
||||
# 设置目标客服ID并执行转接
|
||||
def transfer_in_loop():
|
||||
try:
|
||||
# 设置目标客服ID
|
||||
pdd_instance.csid = customer_service_id
|
||||
|
||||
# 在事件循环中执行转接
|
||||
future = asyncio.run_coroutine_threadsafe(
|
||||
pdd_instance.handle_transfer_message({
|
||||
"content": customer_service_id,
|
||||
"receiver": {"id": user_id}
|
||||
}),
|
||||
loop
|
||||
)
|
||||
|
||||
# 等待转接结果
|
||||
try:
|
||||
result = future.result(timeout=15) # 转接可能需要更长时间
|
||||
if result:
|
||||
print(f"[PDD Transfer] ✅ 转接成功: user_id={user_id} -> cs_id={customer_service_id}")
|
||||
else:
|
||||
print(f"[PDD Transfer] ❌ 转接失败: user_id={user_id}")
|
||||
except Exception as fe:
|
||||
print(f"[PDD Transfer] 转接执行失败: {fe}")
|
||||
except Exception as e:
|
||||
print(f"[PDD Transfer] 转接过程异常: {e}")
|
||||
|
||||
# 在新线程中执行转接操作
|
||||
import threading
|
||||
transfer_thread = threading.Thread(target=transfer_in_loop, daemon=True)
|
||||
transfer_thread.start()
|
||||
|
||||
else:
|
||||
print(f"[PDD Transfer] 条件不足: has_pdd_instance={bool(pdd_instance)}, has_loop={bool(loop)}")
|
||||
except Exception as e:
|
||||
print(f"[PDD Transfer] 拼多多转接失败: {e}")
|
||||
|
||||
def _handle_transfer(self, message: Dict[str, Any]):
|
||||
"""处理转接消息"""
|
||||
# 新版转接消息格式: {"type": "transfer", "content": "客服ID", "receiver": {"id": "用户ID"}, "store_id": "店铺ID"}
|
||||
customer_service_id = message.get('content', '') # 目标客服ID
|
||||
receiver_info = message.get('receiver', {})
|
||||
user_id = receiver_info.get('id', '') # 用户ID
|
||||
store_id = message.get('store_id', '')
|
||||
|
||||
print(f"转接消息: 顾客{user_id}已转接给客服{customer_service_id} (店铺: {store_id})")
|
||||
|
||||
# 根据店铺ID确定平台类型并执行转接
|
||||
try:
|
||||
platform_type = self._get_platform_by_store_id(store_id)
|
||||
|
||||
if platform_type == "京东":
|
||||
# 京东转接逻辑 - 待实现
|
||||
print(f"[JD Transfer] 京东平台转接功能待实现")
|
||||
elif platform_type == "抖音":
|
||||
# 抖音转接逻辑 - 待实现
|
||||
print(f"[DY Transfer] 抖音平台转接功能待实现")
|
||||
elif platform_type == "千牛":
|
||||
# 千牛转接逻辑 - 待实现
|
||||
print(f"[QN Transfer] 千牛平台转接功能待实现")
|
||||
elif platform_type == "拼多多":
|
||||
self._transfer_to_pdd(customer_service_id, user_id, store_id)
|
||||
else:
|
||||
print(f"[Transfer] 未知平台类型或未找到店铺: {platform_type}, store_id={store_id}")
|
||||
except Exception as e:
|
||||
print(f"执行转接操作失败: {e}")
|
||||
|
||||
# 保持旧版回调兼容性
|
||||
if self.transfer_callback:
|
||||
self.transfer_callback(customer_service_id, user_id, store_id)
|
||||
|
||||
def _handle_close(self, message: Dict[str, Any]):
|
||||
"""处理店铺关闭"""
|
||||
store_id = message.get('store_id', '')
|
||||
print(f"店铺{store_id}已关闭")
|
||||
|
||||
if self.close_callback:
|
||||
self.close_callback(store_id)
|
||||
|
||||
def _handle_pong(self, message: Dict[str, Any]):
|
||||
"""处理心跳响应"""
|
||||
uuid_received = message.get('uuid', '')
|
||||
print(f"收到心跳响应: {uuid_received}")
|
||||
|
||||
def _handle_connect_success(self, message: Dict[str, Any]):
|
||||
"""处理连接成功消息（旧版兼容）"""
|
||||
print("后端连接成功(connect_success)")
|
||||
if self.token:
|
||||
self.get_store()
|
||||
|
||||
def _handle_login(self, message: Dict[str, Any]):
|
||||
"""处理平台登录消息（新版：type=login, cookies, store_id, platform_name）"""
|
||||
cookies = message.get('cookies', '')
|
||||
store_id = message.get('store_id', '')
|
||||
platform_name = message.get('platform_name', '')
|
||||
print(f"收到登录指令: 平台={platform_name}, 店铺={store_id}, cookies_len={len(cookies) if cookies else 0}")
|
||||
if self.login_callback:
|
||||
self.login_callback(platform_name, store_id, cookies)
|
||||
|
||||
def _handle_error_message(self, message: Dict[str, Any]):
|
||||
"""处理错误消息"""
|
||||
error_msg = message.get('error', '未知错误')
|
||||
print(f"后端连接错误: {error_msg}")
|
||||
if self.error_callback:
|
||||
self.error_callback(error_msg, message)
|
||||
|
||||
def _handle_staff_list(self, message: Dict[str, Any]):
|
||||
"""处理客服列表更新消息"""
|
||||
staff_list = message.get('data', {}).get('staff_list', [])
|
||||
store_id = message.get('store_id', '')
|
||||
print(f"店铺{store_id}的客服列表已更新，共{len(staff_list)}个客服:")
|
||||
for staff in staff_list:
|
||||
pin = staff.get('pin', '')
|
||||
nickname = staff.get('nickname', '')
|
||||
print(f" - {nickname} ({pin})")
|
||||
|
||||
if self.customers_callback: # 假设客服列表更新也触发客服列表回调
|
||||
self.customers_callback(staff_list, store_id)
|
||||
|
||||
# ==================== 辅助方法 ====================
|
||||
|
||||
def set_token(self, token: str):
|
||||
"""设置或更新令牌"""
|
||||
self.token = token
|
||||
|
||||
def get_connection_info(self) -> Dict[str, Any]:
|
||||
"""获取连接信息"""
|
||||
return {
|
||||
'url': self.url,
|
||||
'token': self.token,
|
||||
'uuid': self.uuid,
|
||||
'is_connected': self.is_connected
|
||||
}
|
||||
|
||||
|
||||
# 使用示例
|
||||
if __name__ == '__main__':
|
||||
pass
|
||||
# import time
|
||||
#
|
||||
# def on_store_list(stores):
|
||||
# print(f"回调: 收到{len(stores)}个店铺")
|
||||
#
|
||||
# def on_message(data, store_id):
|
||||
# print(f"回调: 店铺{store_id}收到消息: {data.get('content')}")
|
||||
#
|
||||
# def on_error(error, message):
|
||||
# print(f"回调: 发生错误: {error}")
|
||||
#
|
||||
# def on_login(platform_name, store_id, cookies):
|
||||
# print(f"回调: 登录指令 平台={platform_name}, 店铺={store_id}, cookies_len={len(cookies) if cookies else 0}")
|
||||
# # 此处触发对应平台的WS连接/更新cookies逻辑，并将该平台WS与store_id绑定
|
||||
#
|
||||
# def on_success():
|
||||
# print("回调: 后端连接成功")
|
||||
#
|
||||
# # 创建客户端（新版：单连接）
|
||||
# client = BackendClient.from_exe_token("your_exe_token_here")
|
||||
#
|
||||
# # 设置回调
|
||||
# client.set_callbacks(
|
||||
# store_list=on_store_list,
|
||||
# message=on_message,
|
||||
# error=on_error,
|
||||
# login=on_login,
|
||||
# success=on_success
|
||||
# )
|
||||
#
|
||||
# try:
|
||||
# # 连接
|
||||
# client.connect()
|
||||
#
|
||||
# # 等待连接
|
||||
# time.sleep(2)
|
||||
#
|
||||
# # 发送心跳
|
||||
# client.send_ping()
|
||||
#
|
||||
# # 保持运行
|
||||
# while True:
|
||||
# time.sleep(30)
|
||||
# client.send_ping()
|
||||
#
|
||||
# except KeyboardInterrupt:
|
||||
# print("用户中断")
|
||||
# finally:
|
||||
# client.disconnect()
|
||||
Reference in New Issue
Block a user