1045 lines
50 KiB
Python
1045 lines
50 KiB
Python
"""
|
||
订单状态处理器
|
||
专门处理订单状态更新逻辑，用于更新订单管理中的状态
|
||
"""
|
||
|
||
import re
|
||
import json
|
||
import time
|
||
import uuid
|
||
import threading
|
||
from loguru import logger
|
||
from typing import Optional, Dict, Any
|
||
|
||
# ==================== 订单状态处理器配置 ====================
|
||
# 订单状态处理器配置
|
||
ORDER_STATUS_HANDLER_CONFIG = {
|
||
'use_pending_queue': True, # 是否使用待处理队列
|
||
'strict_validation': True, # 是否启用严格的状态转换验证
|
||
'log_level': 'info', # 日志级别 (debug/info/warning/error)
|
||
'max_pending_age_hours': 24, # 待处理更新的最大保留时间（小时）
|
||
'enable_status_logging': True, # 是否启用详细的状态变更日志
|
||
}
|
||
|
||
|
||
class OrderStatusHandler:
|
||
"""订单状态处理器"""
|
||
|
||
# 状态转换规则常量
|
||
# 规则说明：
|
||
# 1. 已付款的订单和已完成的订单不能回退到处理中
|
||
# 2. 已付款的订单和已完成的订单可以设置为已关闭（因为会出现退款）
|
||
# 3. 退款中的订单或者退货中的订单设置为退款中
|
||
# 4. 退款中的订单可以设置为已完成（因为买家可能取消退款）
|
||
# 5. 只有退款完成才设置为已关闭
|
||
VALID_TRANSITIONS = {
|
||
'processing': ['pending_ship', 'shipped', 'completed', 'cancelled'],
|
||
'pending_ship': ['shipped', 'completed', 'cancelled', 'refunding'], # 已付款，可以退款
|
||
'shipped': ['completed', 'cancelled', 'refunding'], # 已发货，可以退款
|
||
'completed': ['cancelled', 'refunding'], # 已完成，可以退款
|
||
'refunding': ['completed', 'cancelled', 'refund_cancelled'], # 退款中，可以完成（取消退款）、关闭（退款完成）或撤销
|
||
'refund_cancelled': [], # 退款撤销（临时状态，会立即回退到上一次状态）
|
||
'cancelled': [] # 已关闭，不能转换到其他状态
|
||
}
|
||
|
||
def __init__(self):
|
||
"""初始化订单状态处理器"""
|
||
# 加载配置
|
||
self.config = ORDER_STATUS_HANDLER_CONFIG
|
||
|
||
self.status_mapping = {
|
||
'processing': '处理中', # 初始状态/基本信息阶段
|
||
'pending_ship': '待发货', # 已付款，等待发货
|
||
'shipped': '已发货', # 发货确认后
|
||
'completed': '已完成', # 交易完成
|
||
'refunding': '退款中', # 退款中/退货中
|
||
'refund_cancelled': '退款撤销', # 退款撤销（临时状态，会回退）
|
||
'cancelled': '已关闭', # 交易关闭
|
||
}
|
||
|
||
# 待处理的订单状态更新队列 {order_id: [update_info, ...]}
|
||
self.pending_updates = {}
|
||
# 待处理的系统消息队列（用于延迟处理）{cookie_id: [message_info, ...]}
|
||
self._pending_system_messages = {}
|
||
# 待处理的红色提醒消息队列（用于延迟处理）{cookie_id: [message_info, ...]}
|
||
self._pending_red_reminder_messages = {}
|
||
|
||
# 订单状态历史记录 {order_id: [status_history, ...]}
|
||
# 用于退款撤销时回退到上一次状态
|
||
self._order_status_history = {}
|
||
|
||
# 线程锁，保护并发访问
|
||
self._lock = threading.RLock()
|
||
|
||
# 设置日志级别
|
||
log_level = self.config.get('log_level', 'info')
|
||
logger.info(f"订单状态处理器初始化完成，配置: {self.config}")
|
||
|
||
def extract_order_id(self, message: dict) -> Optional[str]:
|
||
"""从消息中提取订单ID"""
|
||
try:
|
||
order_id = None
|
||
|
||
# 先查看消息的完整结构
|
||
logger.info(f"🔍 完整消息结构: {message}")
|
||
|
||
# 检查message['1']的结构，处理可能是列表、字典或字符串的情况
|
||
message_1 = message.get('1', {})
|
||
content_json_str = ''
|
||
|
||
if isinstance(message_1, dict):
|
||
logger.info(f"🔍 message['1'] 是字典，keys: {list(message_1.keys())}")
|
||
|
||
# 检查message['1']['6']的结构
|
||
message_1_6 = message_1.get('6', {})
|
||
if isinstance(message_1_6, dict):
|
||
logger.info(f"🔍 message['1']['6'] 是字典，keys: {list(message_1_6.keys())}")
|
||
# 方法1: 从button的targetUrl中提取orderId
|
||
content_json_str = message_1_6.get('3', {}).get('5', '') if isinstance(message_1_6.get('3', {}), dict) else ''
|
||
else:
|
||
logger.info(f"🔍 message['1']['6'] 不是字典: {type(message_1_6)}")
|
||
|
||
elif isinstance(message_1, list):
|
||
logger.info(f"🔍 message['1'] 是列表，长度: {len(message_1)}")
|
||
# 如果message['1']是列表，跳过这种提取方式
|
||
|
||
elif isinstance(message_1, str):
|
||
logger.info(f"🔍 message['1'] 是字符串，长度: {len(message_1)}")
|
||
# 如果message['1']是字符串，跳过这种提取方式
|
||
|
||
else:
|
||
logger.info(f"🔍 message['1'] 未知类型: {type(message_1)}")
|
||
# 其他类型，跳过这种提取方式
|
||
|
||
if content_json_str:
|
||
try:
|
||
content_data = json.loads(content_json_str)
|
||
|
||
# 方法1a: 从button的targetUrl中提取orderId
|
||
target_url = content_data.get('dxCard', {}).get('item', {}).get('main', {}).get('exContent', {}).get('button', {}).get('targetUrl', '')
|
||
if target_url:
|
||
# 从URL中提取orderId参数
|
||
order_match = re.search(r'orderId=(\d+)', target_url)
|
||
if order_match:
|
||
order_id = order_match.group(1)
|
||
logger.info(f'✅ 从button提取到订单ID: {order_id}')
|
||
|
||
# 方法1b: 从main的targetUrl中提取order_detail的id
|
||
if not order_id:
|
||
main_target_url = content_data.get('dxCard', {}).get('item', {}).get('main', {}).get('targetUrl', '')
|
||
if main_target_url:
|
||
order_match = re.search(r'order_detail\?id=(\d+)', main_target_url)
|
||
if order_match:
|
||
order_id = order_match.group(1)
|
||
logger.info(f'✅ 从main targetUrl提取到订单ID: {order_id}')
|
||
|
||
except Exception as parse_e:
|
||
logger.error(f"解析内容JSON失败: {parse_e}")
|
||
|
||
# 方法2: 从dynamicOperation中的order_detail URL提取orderId
|
||
if not order_id and content_json_str:
|
||
try:
|
||
content_data = json.loads(content_json_str)
|
||
dynamic_target_url = content_data.get('dynamicOperation', {}).get('changeContent', {}).get('dxCard', {}).get('item', {}).get('main', {}).get('exContent', {}).get('button', {}).get('targetUrl', '')
|
||
if dynamic_target_url:
|
||
# 从order_detail URL中提取id参数
|
||
order_match = re.search(r'order_detail\?id=(\d+)', dynamic_target_url)
|
||
if order_match:
|
||
order_id = order_match.group(1)
|
||
logger.info(f'✅ 从order_detail提取到订单ID: {order_id}')
|
||
except Exception as parse_e:
|
||
logger.error(f"解析dynamicOperation JSON失败: {parse_e}")
|
||
|
||
# 方法3: 如果前面的方法都失败，尝试在整个消息中搜索订单ID模式
|
||
if not order_id:
|
||
try:
|
||
# 将整个消息转换为字符串进行搜索
|
||
message_str = str(message)
|
||
|
||
# 搜索各种可能的订单ID模式
|
||
patterns = [
|
||
r'orderId[=:](\d{10,})', # orderId=123456789 或 orderId:123456789
|
||
r'order_detail\?id=(\d{10,})', # order_detail?id=123456789
|
||
r'"id"\s*:\s*"?(\d{10,})"?', # "id":"123456789" 或 "id":123456789
|
||
r'bizOrderId[=:](\d{10,})', # bizOrderId=123456789
|
||
]
|
||
|
||
for pattern in patterns:
|
||
matches = re.findall(pattern, message_str)
|
||
if matches:
|
||
# 取第一个匹配的订单ID
|
||
order_id = matches[0]
|
||
logger.info(f'✅ 从消息字符串中提取到订单ID: {order_id} (模式: {pattern})')
|
||
break
|
||
|
||
except Exception as search_e:
|
||
logger.error(f"在消息字符串中搜索订单ID失败: {search_e}")
|
||
|
||
if order_id:
|
||
logger.info(f'🎯 最终提取到订单ID: {order_id}')
|
||
else:
|
||
logger.error(f'❌ 未能从消息中提取到订单ID')
|
||
|
||
return order_id
|
||
|
||
except Exception as e:
|
||
logger.error(f"提取订单ID失败: {str(e)}")
|
||
return None
|
||
|
||
def update_order_status(self, order_id: str, new_status: str, cookie_id: str, context: str = "") -> bool:
|
||
"""更新订单状态到数据库
|
||
|
||
Args:
|
||
order_id: 订单ID
|
||
new_status: 新状态 (processing/pending_ship/shipped/completed/cancelled)
|
||
cookie_id: Cookie ID
|
||
context: 上下文信息，用于日志记录
|
||
|
||
Returns:
|
||
bool: 更新是否成功
|
||
"""
|
||
logger.info(f"🔄 订单状态处理器.update_order_status开始: order_id={order_id}, new_status={new_status}, cookie_id={cookie_id}, context={context}")
|
||
with self._lock:
|
||
try:
|
||
from db_manager import db_manager
|
||
|
||
# 验证状态值是否有效
|
||
if new_status not in self.status_mapping:
|
||
logger.error(f"❌ 无效的订单状态: {new_status}，有效状态: {list(self.status_mapping.keys())}")
|
||
return False
|
||
|
||
logger.info(f"✅ 订单状态验证通过: {new_status}")
|
||
|
||
# 检查订单是否存在于数据库中（带重试机制）
|
||
current_order = None
|
||
max_retries = 3
|
||
for attempt in range(max_retries):
|
||
try:
|
||
logger.info(f"🔍 尝试获取订单信息 (尝试 {attempt + 1}/{max_retries}): {order_id}")
|
||
current_order = db_manager.get_order_by_id(order_id)
|
||
logger.info(f"✅ 订单信息获取成功: {order_id}")
|
||
break
|
||
except Exception as db_e:
|
||
if attempt == max_retries - 1:
|
||
logger.error(f"❌ 获取订单信息失败 (尝试 {attempt + 1}/{max_retries}): {str(db_e)}")
|
||
return False
|
||
else:
|
||
logger.error(f"⚠️ 获取订单信息失败，重试中 (尝试 {attempt + 1}/{max_retries}): {str(db_e)}")
|
||
time.sleep(0.1 * (attempt + 1)) # 递增延迟
|
||
|
||
if not current_order:
|
||
# 订单不存在，根据配置决定是否添加到待处理队列
|
||
logger.info(f"⚠️ 订单 {order_id} 不存在于数据库中")
|
||
if self.config.get('use_pending_queue', True):
|
||
logger.info(f"📝 订单 {order_id} 不存在于数据库中，添加到待处理队列等待主程序拉取订单详情")
|
||
self._add_to_pending_updates(order_id, new_status, cookie_id, context)
|
||
else:
|
||
logger.error(f"❌ 订单 {order_id} 不存在于数据库中且未启用待处理队列，跳过状态更新")
|
||
return False
|
||
|
||
current_status = current_order.get('order_status', 'processing')
|
||
logger.info(f"📊 当前订单状态: {current_status}, 目标状态: {new_status}")
|
||
|
||
# 检查是否是相同的状态更新（避免重复处理）
|
||
if current_status == new_status:
|
||
status_text = self.status_mapping.get(new_status, new_status)
|
||
logger.info(f"⏭️ 订单 {order_id} 状态无变化，跳过重复更新: {status_text}")
|
||
return True # 返回True表示"成功"，避免重复日志
|
||
|
||
# 检查状态转换是否合理（根据配置决定是否启用严格验证）
|
||
if self.config.get('strict_validation', True) and not self._is_valid_status_transition(current_status, new_status):
|
||
logger.error(f"❌ 订单 {order_id} 状态转换不合理: {current_status} -> {new_status} (严格验证已启用)")
|
||
logger.error(f"当前状态 '{current_status}' 允许转换到: {self._get_allowed_transitions(current_status)}")
|
||
return False
|
||
|
||
logger.info(f"✅ 状态转换验证通过: {current_status} -> {new_status}")
|
||
|
||
# 处理退款撤销的特殊逻辑
|
||
if new_status == 'refund_cancelled':
|
||
# 从历史记录中获取上一次状态
|
||
previous_status = self._get_previous_status(order_id)
|
||
if previous_status:
|
||
logger.info(f"🔄 退款撤销，回退到上一次状态: {previous_status}")
|
||
new_status = previous_status
|
||
else:
|
||
logger.warning(f"⚠️ 退款撤销但无法获取上一次状态，保持当前状态: {current_status}")
|
||
new_status = current_status
|
||
|
||
# 更新订单状态（带重试机制）
|
||
success = False
|
||
for attempt in range(max_retries):
|
||
try:
|
||
logger.info(f"💾 尝试更新订单状态 (尝试 {attempt + 1}/{max_retries}): {order_id}")
|
||
success = db_manager.insert_or_update_order(
|
||
order_id=order_id,
|
||
order_status=new_status,
|
||
cookie_id=cookie_id
|
||
)
|
||
logger.info(f"✅ 订单状态更新成功: {order_id}")
|
||
break
|
||
except Exception as db_e:
|
||
if attempt == max_retries - 1:
|
||
logger.error(f"❌ 更新订单状态失败 (尝试 {attempt + 1}/{max_retries}): {str(db_e)}")
|
||
return False
|
||
else:
|
||
logger.error(f"⚠️ 更新订单状态失败，重试中 (尝试 {attempt + 1}/{max_retries}): {str(db_e)}")
|
||
time.sleep(0.1 * (attempt + 1)) # 递增延迟
|
||
|
||
if success:
|
||
# 记录状态历史（用于退款撤销时回退）
|
||
self._record_status_history(order_id, current_status, new_status, context)
|
||
|
||
status_text = self.status_mapping.get(new_status, new_status)
|
||
if self.config.get('enable_status_logging', True):
|
||
logger.info(f"✅ 订单状态更新成功: {order_id} -> {status_text} ({context})")
|
||
else:
|
||
logger.error(f"❌ 订单状态更新失败: {order_id} -> {new_status} ({context})")
|
||
|
||
return success
|
||
|
||
except Exception as e:
|
||
logger.error(f"更新订单状态时出错: {str(e)}")
|
||
import traceback
|
||
logger.error(f"详细错误信息: {traceback.format_exc()}")
|
||
return False
|
||
|
||
def _is_valid_status_transition(self, current_status: str, new_status: str) -> bool:
|
||
"""检查状态转换是否合理
|
||
|
||
Args:
|
||
current_status: 当前状态
|
||
new_status: 新状态
|
||
|
||
Returns:
|
||
bool: 转换是否合理
|
||
"""
|
||
# 如果当前状态不在规则中，允许转换（兼容性）
|
||
if current_status not in self.VALID_TRANSITIONS:
|
||
return True
|
||
|
||
# 特殊规则：已付款的订单和已完成的订单不能回退到处理中
|
||
if new_status == 'processing' and current_status in ['pending_ship', 'shipped', 'completed', 'refunding', 'refund_cancelled']:
|
||
logger.warning(f"❌ 状态转换被拒绝：{current_status} -> {new_status} (已付款/已完成的订单不能回退到处理中)")
|
||
return False
|
||
|
||
# 检查新状态是否在允许的转换列表中
|
||
allowed_statuses = self.VALID_TRANSITIONS.get(current_status, [])
|
||
return new_status in allowed_statuses
|
||
|
||
def _get_allowed_transitions(self, current_status: str) -> list:
|
||
"""获取当前状态允许转换到的状态列表
|
||
|
||
Args:
|
||
current_status: 当前状态
|
||
|
||
Returns:
|
||
list: 允许转换到的状态列表
|
||
"""
|
||
if current_status not in self.VALID_TRANSITIONS:
|
||
return ['所有状态'] # 兼容性
|
||
|
||
return self.VALID_TRANSITIONS.get(current_status, [])
|
||
|
||
def _check_refund_message(self, message: dict, send_message: str) -> Optional[str]:
|
||
"""检查退款申请消息，需要同时识别标题和按钮文本
|
||
|
||
Args:
|
||
message: 原始消息数据
|
||
send_message: 消息内容
|
||
|
||
Returns:
|
||
str: 对应的状态，如果不是退款消息则返回None
|
||
"""
|
||
try:
|
||
# 检查消息结构，寻找退款相关的信息
|
||
message_1 = message.get('1', {})
|
||
if not isinstance(message_1, dict):
|
||
return None
|
||
|
||
# 检查消息卡片内容
|
||
message_1_6 = message_1.get('6', {})
|
||
if not isinstance(message_1_6, dict):
|
||
return None
|
||
|
||
# 解析JSON内容
|
||
content_json_str = message_1_6.get('3', {}).get('5', '') if isinstance(message_1_6.get('3', {}), dict) else ''
|
||
if not content_json_str:
|
||
return None
|
||
|
||
try:
|
||
content_data = json.loads(content_json_str)
|
||
|
||
# 检查dynamicOperation中的内容
|
||
dynamic_content = content_data.get('dynamicOperation', {}).get('changeContent', {})
|
||
if not dynamic_content:
|
||
return None
|
||
|
||
dx_card = dynamic_content.get('dxCard', {}).get('item', {}).get('main', {})
|
||
if not dx_card:
|
||
return None
|
||
|
||
ex_content = dx_card.get('exContent', {})
|
||
if not ex_content:
|
||
return None
|
||
|
||
# 获取标题和按钮文本
|
||
title = ex_content.get('title', '')
|
||
button_text = ex_content.get('button', {}).get('text', '')
|
||
|
||
logger.info(f"🔍 检查退款消息 - 标题: '{title}', 按钮: '{button_text}'")
|
||
|
||
# 检查是否是退款申请且已同意
|
||
if title == '我发起了退款申请' and button_text == '已同意':
|
||
logger.info(f"✅ 识别到退款申请已同意消息")
|
||
return 'refunding'
|
||
|
||
# 检查是否是退款撤销（买家主动撤销）
|
||
if title == '我发起了退款申请' and button_text == '已撤销':
|
||
logger.info(f"✅ 识别到退款撤销消息")
|
||
return 'refund_cancelled'
|
||
|
||
# 退款申请被拒绝不需要改变状态，因为没同意
|
||
# if title == '我发起了退款申请' and button_text == '已拒绝':
|
||
# logger.info(f"ℹ️ 识别到退款申请被拒绝消息，不改变订单状态")
|
||
# return None
|
||
|
||
except Exception as parse_e:
|
||
logger.debug(f"解析退款消息JSON失败: {parse_e}")
|
||
return None
|
||
|
||
return None
|
||
|
||
except Exception as e:
|
||
logger.debug(f"检查退款消息失败: {e}")
|
||
return None
|
||
|
||
def _record_status_history(self, order_id: str, from_status: str, to_status: str, context: str):
|
||
"""记录订单状态历史
|
||
|
||
Args:
|
||
order_id: 订单ID
|
||
from_status: 原状态
|
||
to_status: 新状态
|
||
context: 上下文信息
|
||
"""
|
||
with self._lock:
|
||
if order_id not in self._order_status_history:
|
||
self._order_status_history[order_id] = []
|
||
|
||
# 只记录非临时状态的历史（排除 refund_cancelled）
|
||
if to_status != 'refund_cancelled':
|
||
history_entry = {
|
||
'from_status': from_status,
|
||
'to_status': to_status,
|
||
'context': context,
|
||
'timestamp': time.time()
|
||
}
|
||
self._order_status_history[order_id].append(history_entry)
|
||
|
||
# 限制历史记录数量，只保留最近10条
|
||
if len(self._order_status_history[order_id]) > 10:
|
||
self._order_status_history[order_id] = self._order_status_history[order_id][-10:]
|
||
|
||
logger.debug(f"📝 记录订单状态历史: {order_id} {from_status} -> {to_status}")
|
||
|
||
def _get_previous_status(self, order_id: str) -> Optional[str]:
|
||
"""获取订单的上一次状态（用于退款撤销时回退）
|
||
|
||
Args:
|
||
order_id: 订单ID
|
||
|
||
Returns:
|
||
str: 上一次状态，如果没有历史记录则返回None
|
||
"""
|
||
with self._lock:
|
||
if order_id not in self._order_status_history or not self._order_status_history[order_id]:
|
||
return None
|
||
|
||
# 获取最后一次状态变化的目标状态
|
||
last_entry = self._order_status_history[order_id][-1]
|
||
return last_entry['to_status']
|
||
|
||
def _add_to_pending_updates(self, order_id: str, new_status: str, cookie_id: str, context: str):
|
||
"""添加到待处理更新队列
|
||
|
||
Args:
|
||
order_id: 订单ID
|
||
new_status: 新状态
|
||
cookie_id: Cookie ID
|
||
context: 上下文信息
|
||
"""
|
||
with self._lock:
|
||
if order_id not in self.pending_updates:
|
||
self.pending_updates[order_id] = []
|
||
|
||
update_info = {
|
||
'new_status': new_status,
|
||
'cookie_id': cookie_id,
|
||
'context': context,
|
||
'timestamp': time.time()
|
||
}
|
||
|
||
self.pending_updates[order_id].append(update_info)
|
||
logger.info(f"订单 {order_id} 状态更新已添加到待处理队列: {new_status} ({context})")
|
||
|
||
def process_pending_updates(self, order_id: str) -> bool:
|
||
"""处理指定订单的待处理更新
|
||
|
||
Args:
|
||
order_id: 订单ID
|
||
|
||
Returns:
|
||
bool: 是否有更新被处理
|
||
"""
|
||
with self._lock:
|
||
if order_id not in self.pending_updates:
|
||
return False
|
||
|
||
updates = self.pending_updates.pop(order_id)
|
||
processed_count = 0
|
||
|
||
for update_info in updates:
|
||
try:
|
||
success = self.update_order_status(
|
||
order_id=order_id,
|
||
new_status=update_info['new_status'],
|
||
cookie_id=update_info['cookie_id'],
|
||
context=f"待处理队列: {update_info['context']}"
|
||
)
|
||
|
||
if success:
|
||
processed_count += 1
|
||
logger.info(f"处理待处理更新成功: 订单 {order_id} -> {update_info['new_status']}")
|
||
else:
|
||
logger.error(f"处理待处理更新失败: 订单 {order_id} -> {update_info['new_status']}")
|
||
|
||
except Exception as e:
|
||
logger.error(f"处理待处理更新时出错: {str(e)}")
|
||
|
||
if processed_count > 0:
|
||
logger.info(f"订单 {order_id} 共处理了 {processed_count} 个待处理状态更新")
|
||
|
||
return processed_count > 0
|
||
|
||
def process_all_pending_updates(self) -> int:
|
||
"""处理所有待处理的更新
|
||
|
||
Returns:
|
||
int: 处理的订单数量
|
||
"""
|
||
with self._lock:
|
||
if not self.pending_updates:
|
||
return 0
|
||
|
||
order_ids = list(self.pending_updates.keys())
|
||
processed_orders = 0
|
||
|
||
for order_id in order_ids:
|
||
if self.process_pending_updates(order_id):
|
||
processed_orders += 1
|
||
|
||
return processed_orders
|
||
|
||
def get_pending_updates_count(self) -> int:
|
||
"""获取待处理更新的数量
|
||
|
||
Returns:
|
||
int: 待处理更新的数量
|
||
"""
|
||
with self._lock:
|
||
return len(self.pending_updates)
|
||
|
||
def clear_old_pending_updates(self, max_age_hours: int = None):
|
||
"""清理过期的待处理更新
|
||
|
||
Args:
|
||
max_age_hours: 最大保留时间（小时），如果为None则使用配置中的默认值
|
||
"""
|
||
# 检查是否启用待处理队列
|
||
if not self.config.get('use_pending_queue', True):
|
||
logger.error("未启用待处理队列，跳过清理操作")
|
||
return
|
||
|
||
if max_age_hours is None:
|
||
max_age_hours = self.config.get('max_pending_age_hours', 24)
|
||
|
||
current_time = time.time()
|
||
max_age_seconds = max_age_hours * 3600
|
||
|
||
with self._lock:
|
||
# 清理 pending_updates
|
||
expired_orders = []
|
||
for order_id, updates in self.pending_updates.items():
|
||
# 过滤掉过期的更新
|
||
valid_updates = [
|
||
update for update in updates
|
||
if current_time - update['timestamp'] < max_age_seconds
|
||
]
|
||
|
||
if not valid_updates:
|
||
expired_orders.append(order_id)
|
||
else:
|
||
self.pending_updates[order_id] = valid_updates
|
||
|
||
# 移除完全过期的订单
|
||
for order_id in expired_orders:
|
||
del self.pending_updates[order_id]
|
||
logger.info(f"清理过期的待处理更新: 订单 {order_id}")
|
||
|
||
if expired_orders:
|
||
logger.info(f"共清理了 {len(expired_orders)} 个过期的待处理订单更新")
|
||
|
||
# 清理 _pending_system_messages
|
||
expired_cookies_system = []
|
||
for cookie_id, messages in self._pending_system_messages.items():
|
||
valid_messages = [
|
||
msg for msg in messages
|
||
if current_time - msg.get('timestamp', 0) < max_age_seconds
|
||
]
|
||
|
||
if not valid_messages:
|
||
expired_cookies_system.append(cookie_id)
|
||
else:
|
||
self._pending_system_messages[cookie_id] = valid_messages
|
||
|
||
for cookie_id in expired_cookies_system:
|
||
del self._pending_system_messages[cookie_id]
|
||
logger.info(f"清理过期的待处理系统消息: 账号 {cookie_id}")
|
||
|
||
# 清理 _pending_red_reminder_messages
|
||
expired_cookies_red = []
|
||
for cookie_id, messages in self._pending_red_reminder_messages.items():
|
||
valid_messages = [
|
||
msg for msg in messages
|
||
if current_time - msg.get('timestamp', 0) < max_age_seconds
|
||
]
|
||
|
||
if not valid_messages:
|
||
expired_cookies_red.append(cookie_id)
|
||
else:
|
||
self._pending_red_reminder_messages[cookie_id] = valid_messages
|
||
|
||
for cookie_id in expired_cookies_red:
|
||
del self._pending_red_reminder_messages[cookie_id]
|
||
logger.info(f"清理过期的待处理红色提醒消息: 账号 {cookie_id}")
|
||
|
||
total_cleared = len(expired_orders) + len(expired_cookies_system) + len(expired_cookies_red)
|
||
if total_cleared > 0:
|
||
logger.info(f"内存清理完成，共清理了 {total_cleared} 个过期项目")
|
||
|
||
def handle_system_message(self, message: dict, send_message: str, cookie_id: str, msg_time: str) -> bool:
|
||
"""处理系统消息并更新订单状态
|
||
|
||
Args:
|
||
message: 原始消息数据
|
||
send_message: 消息内容
|
||
cookie_id: Cookie ID
|
||
msg_time: 消息时间
|
||
|
||
Returns:
|
||
bool: 是否处理了订单状态更新
|
||
"""
|
||
try:
|
||
# 定义消息类型与状态的映射
|
||
message_status_mapping = {
|
||
'[买家确认收货，交易成功]': 'completed',
|
||
'[你已确认收货，交易成功]': 'completed', # 已完成
|
||
'[你已发货]': 'shipped', # 已发货
|
||
'你已发货': 'shipped', # 已发货（无方括号）
|
||
'[你已发货，请等待买家确认收货]': 'shipped', # 已发货（完整格式）
|
||
'[我已付款，等待你发货]': 'pending_ship', # 已付款，等待发货
|
||
'[我已拍下，待付款]': 'processing', # 已拍下，待付款
|
||
'[买家已付款]': 'pending_ship', # 买家已付款
|
||
'[付款完成]': 'pending_ship', # 付款完成
|
||
'[已付款，待发货]': 'pending_ship', # 已付款，待发货
|
||
'[退款成功，钱款已原路退返]': 'cancelled', # 退款成功，设置为已关闭
|
||
'[你关闭了订单，钱款已原路退返]': 'cancelled', # 卖家关闭订单，设置为已关闭
|
||
}
|
||
|
||
# 特殊处理：检查退款申请消息（需要同时识别标题和按钮文本）
|
||
refund_status = self._check_refund_message(message, send_message)
|
||
if refund_status:
|
||
new_status = refund_status
|
||
elif send_message in message_status_mapping:
|
||
new_status = message_status_mapping[send_message]
|
||
else:
|
||
return False
|
||
|
||
# 提取订单ID
|
||
order_id = self.extract_order_id(message)
|
||
if not order_id:
|
||
# 如果无法提取订单ID，根据配置决定是否添加到待处理队列
|
||
if self.config.get('use_pending_queue', True):
|
||
logger.info(f'[{msg_time}] 【{cookie_id}】{send_message}，暂时无法提取订单ID，添加到待处理队列')
|
||
else:
|
||
logger.error(f'[{msg_time}] 【{cookie_id}】{send_message}，无法提取订单ID且未启用待处理队列，跳过处理')
|
||
return False
|
||
|
||
# 创建一个临时的订单ID占位符，用于标识这个待处理的状态更新
|
||
temp_order_id = f"temp_{int(time.time() * 1000)}_{uuid.uuid4().hex[:8]}"
|
||
|
||
# 获取对应的状态
|
||
new_status = message_status_mapping[send_message]
|
||
|
||
# 添加到待处理队列，使用特殊标记
|
||
self._add_to_pending_updates(
|
||
order_id=temp_order_id,
|
||
new_status=new_status,
|
||
cookie_id=cookie_id,
|
||
context=f"{send_message} - {msg_time} - 等待订单ID提取"
|
||
)
|
||
|
||
# 添加到待处理的系统消息队列
|
||
if cookie_id not in self._pending_system_messages:
|
||
self._pending_system_messages[cookie_id] = []
|
||
|
||
self._pending_system_messages[cookie_id].append({
|
||
'message': message,
|
||
'send_message': send_message,
|
||
'cookie_id': cookie_id,
|
||
'msg_time': msg_time,
|
||
'new_status': new_status,
|
||
'temp_order_id': temp_order_id,
|
||
'message_hash': hash(str(sorted(message.items()))) if isinstance(message, dict) else hash(str(message)), # 添加消息哈希用于匹配
|
||
'timestamp': time.time() # 添加时间戳用于清理
|
||
})
|
||
|
||
return True
|
||
|
||
# 获取对应的状态（new_status已经在上面通过_check_refund_message或message_status_mapping确定了）
|
||
|
||
# 更新订单状态
|
||
success = self.update_order_status(
|
||
order_id=order_id,
|
||
new_status=new_status,
|
||
cookie_id=cookie_id,
|
||
context=f"{send_message} - {msg_time}"
|
||
)
|
||
|
||
if success:
|
||
status_text = self.status_mapping.get(new_status, new_status)
|
||
logger.info(f'[{msg_time}] 【{cookie_id}】{send_message}，订单 {order_id} 状态已更新为{status_text}')
|
||
else:
|
||
logger.error(f'[{msg_time}] 【{cookie_id}】{send_message}，但订单 {order_id} 状态更新失败')
|
||
|
||
return True
|
||
|
||
except Exception as e:
|
||
logger.error(f'[{msg_time}] 【{cookie_id}】处理系统消息订单状态更新时出错: {str(e)}')
|
||
return False
|
||
|
||
def handle_red_reminder_message(self, message: dict, red_reminder: str, user_id: str, cookie_id: str, msg_time: str) -> bool:
|
||
"""处理红色提醒消息并更新订单状态
|
||
|
||
Args:
|
||
message: 原始消息数据
|
||
red_reminder: 红色提醒内容
|
||
user_id: 用户ID
|
||
cookie_id: Cookie ID
|
||
msg_time: 消息时间
|
||
|
||
Returns:
|
||
bool: 是否处理了订单状态更新
|
||
"""
|
||
try:
|
||
# 只处理交易关闭的情况
|
||
if red_reminder != '交易关闭':
|
||
return False
|
||
|
||
# 提取订单ID
|
||
order_id = self.extract_order_id(message)
|
||
if not order_id:
|
||
# 如果无法提取订单ID，根据配置决定是否添加到待处理队列
|
||
if self.config.get('use_pending_queue', True):
|
||
logger.info(f'[{msg_time}] 【{cookie_id}】交易关闭，暂时无法提取订单ID，添加到待处理队列')
|
||
else:
|
||
logger.error(f'[{msg_time}] 【{cookie_id}】交易关闭，无法提取订单ID且未启用待处理队列，跳过处理')
|
||
return False
|
||
|
||
# 创建一个临时的订单ID占位符，用于标识这个待处理的状态更新
|
||
temp_order_id = f"temp_{int(time.time() * 1000)}_{uuid.uuid4().hex[:8]}"
|
||
|
||
# 添加到待处理队列，使用特殊标记
|
||
self._add_to_pending_updates(
|
||
order_id=temp_order_id,
|
||
new_status='cancelled',
|
||
cookie_id=cookie_id,
|
||
context=f"交易关闭 - 用户{user_id} - {msg_time} - 等待订单ID提取"
|
||
)
|
||
|
||
# 添加到待处理的红色提醒消息队列
|
||
if cookie_id not in self._pending_red_reminder_messages:
|
||
self._pending_red_reminder_messages[cookie_id] = []
|
||
|
||
self._pending_red_reminder_messages[cookie_id].append({
|
||
'message': message,
|
||
'red_reminder': red_reminder,
|
||
'user_id': user_id,
|
||
'cookie_id': cookie_id,
|
||
'msg_time': msg_time,
|
||
'new_status': 'cancelled',
|
||
'temp_order_id': temp_order_id,
|
||
'message_hash': hash(str(sorted(message.items()))) if isinstance(message, dict) else hash(str(message)), # 添加消息哈希用于匹配
|
||
'timestamp': time.time() # 添加时间戳用于清理
|
||
})
|
||
|
||
return True
|
||
|
||
# 更新订单状态为已关闭
|
||
success = self.update_order_status(
|
||
order_id=order_id,
|
||
new_status='cancelled',
|
||
cookie_id=cookie_id,
|
||
context=f"交易关闭 - 用户{user_id} - {msg_time}"
|
||
)
|
||
|
||
if success:
|
||
logger.info(f'[{msg_time}] 【{cookie_id}】交易关闭，订单 {order_id} 状态已更新为已关闭')
|
||
else:
|
||
logger.error(f'[{msg_time}] 【{cookie_id}】交易关闭，但订单 {order_id} 状态更新失败')
|
||
|
||
return True
|
||
|
||
except Exception as e:
|
||
logger.error(f'[{msg_time}] 【{cookie_id}】处理交易关闭订单状态更新时出错: {str(e)}')
|
||
return False
|
||
|
||
def handle_auto_delivery_order_status(self, order_id: str, cookie_id: str, context: str = "自动发货") -> bool:
|
||
"""处理自动发货时的订单状态更新
|
||
|
||
Args:
|
||
order_id: 订单ID
|
||
cookie_id: Cookie ID
|
||
context: 上下文信息
|
||
|
||
Returns:
|
||
bool: 更新是否成功
|
||
"""
|
||
return self.update_order_status(
|
||
order_id=order_id,
|
||
new_status='shipped', # 已发货
|
||
cookie_id=cookie_id,
|
||
context=context
|
||
)
|
||
|
||
def handle_order_basic_info_status(self, order_id: str, cookie_id: str, context: str = "基本信息保存") -> bool:
|
||
"""处理订单基本信息保存时的状态设置
|
||
|
||
Args:
|
||
order_id: 订单ID
|
||
cookie_id: Cookie ID
|
||
context: 上下文信息
|
||
|
||
Returns:
|
||
bool: 更新是否成功
|
||
"""
|
||
return self.update_order_status(
|
||
order_id=order_id,
|
||
new_status='processing', # 处理中
|
||
cookie_id=cookie_id,
|
||
context=context
|
||
)
|
||
|
||
def handle_order_detail_fetched_status(self, order_id: str, cookie_id: str, context: str = "详情已获取") -> bool:
|
||
"""处理订单详情拉取后的状态设置
|
||
|
||
Args:
|
||
order_id: 订单ID
|
||
cookie_id: Cookie ID
|
||
context: 上下文信息
|
||
|
||
Returns:
|
||
bool: 更新是否成功
|
||
"""
|
||
logger.info(f"🔄 订单状态处理器.handle_order_detail_fetched_status开始: order_id={order_id}, cookie_id={cookie_id}, context={context}")
|
||
|
||
# 订单详情获取成功后，不需要改变状态，只是处理待处理队列
|
||
logger.info(f"✅ 订单详情已获取，处理待处理队列: order_id={order_id}")
|
||
return True
|
||
|
||
def on_order_details_fetched(self, order_id: str):
|
||
"""当主程序拉取到订单详情后调用此方法处理待处理的更新
|
||
|
||
Args:
|
||
order_id: 订单ID
|
||
"""
|
||
logger.info(f"🔄 订单状态处理器.on_order_details_fetched开始: order_id={order_id}")
|
||
|
||
# 检查是否启用待处理队列
|
||
if not self.config.get('use_pending_queue', True):
|
||
logger.info(f"⏭️ 订单 {order_id} 详情已拉取，但未启用待处理队列，跳过处理")
|
||
return
|
||
|
||
logger.info(f"✅ 待处理队列已启用，检查订单 {order_id} 的待处理更新")
|
||
|
||
with self._lock:
|
||
if order_id in self.pending_updates:
|
||
logger.info(f"📝 检测到订单 {order_id} 详情已拉取，开始处理待处理的状态更新")
|
||
# 注意：process_pending_updates 内部也有锁，这里需要先释放锁避免死锁
|
||
updates = self.pending_updates.pop(order_id)
|
||
logger.info(f"📊 订单 {order_id} 有 {len(updates)} 个待处理更新")
|
||
else:
|
||
logger.info(f"ℹ️ 订单 {order_id} 没有待处理的更新")
|
||
return
|
||
|
||
# 在锁外处理更新，避免死锁
|
||
if 'updates' in locals():
|
||
logger.info(f"🔄 开始处理订单 {order_id} 的 {len(updates)} 个待处理更新")
|
||
self._process_updates_outside_lock(order_id, updates)
|
||
logger.info(f"✅ 订单 {order_id} 的待处理更新处理完成")
|
||
|
||
def _process_updates_outside_lock(self, order_id: str, updates: list):
|
||
"""在锁外处理更新，避免死锁
|
||
|
||
Args:
|
||
order_id: 订单ID
|
||
updates: 更新列表
|
||
"""
|
||
processed_count = 0
|
||
|
||
for update_info in updates:
|
||
try:
|
||
success = self.update_order_status(
|
||
order_id=order_id,
|
||
new_status=update_info['new_status'],
|
||
cookie_id=update_info['cookie_id'],
|
||
context=f"待处理队列: {update_info['context']}"
|
||
)
|
||
|
||
if success:
|
||
processed_count += 1
|
||
logger.info(f"处理待处理更新成功: 订单 {order_id} -> {update_info['new_status']}")
|
||
else:
|
||
logger.error(f"处理待处理更新失败: 订单 {order_id} -> {update_info['new_status']}")
|
||
|
||
except Exception as e:
|
||
logger.error(f"处理待处理更新时出错: {str(e)}")
|
||
|
||
if processed_count > 0:
|
||
logger.info(f"订单 {order_id} 共处理了 {processed_count} 个待处理状态更新")
|
||
|
||
def on_order_id_extracted(self, order_id: str, cookie_id: str, message: dict = None):
|
||
"""当主程序成功提取到订单ID后调用此方法处理待处理的系统消息
|
||
|
||
Args:
|
||
order_id: 订单ID
|
||
cookie_id: Cookie ID
|
||
message: 原始消息（可选，用于匹配）
|
||
"""
|
||
logger.info(f"🔄 订单状态处理器.on_order_id_extracted开始: order_id={order_id}, cookie_id={cookie_id}")
|
||
|
||
with self._lock:
|
||
# 检查是否启用待处理队列
|
||
if not self.config.get('use_pending_queue', True):
|
||
logger.info(f"⏭️ 订单 {order_id} ID已提取，但未启用待处理队列，跳过处理")
|
||
return
|
||
|
||
logger.info(f"✅ 待处理队列已启用，检查账号 {cookie_id} 的待处理系统消息")
|
||
|
||
# 处理待处理的系统消息队列
|
||
if cookie_id in self._pending_system_messages and self._pending_system_messages[cookie_id]:
|
||
logger.info(f"📝 账号 {cookie_id} 有 {len(self._pending_system_messages[cookie_id])} 个待处理的系统消息")
|
||
pending_msg = None
|
||
|
||
# 如果提供了消息，尝试匹配
|
||
if message:
|
||
logger.info(f"🔍 尝试通过消息哈希匹配待处理的系统消息")
|
||
message_hash = hash(str(sorted(message.items()))) if isinstance(message, dict) else hash(str(message))
|
||
# 从后往前遍历，避免pop时索引变化问题
|
||
for i in range(len(self._pending_system_messages[cookie_id]) - 1, -1, -1):
|
||
msg = self._pending_system_messages[cookie_id][i]
|
||
if msg.get('message_hash') == message_hash:
|
||
pending_msg = self._pending_system_messages[cookie_id].pop(i)
|
||
logger.info(f"✅ 通过消息哈希匹配到待处理的系统消息: {pending_msg['send_message']}")
|
||
break
|
||
|
||
# 如果没有匹配到，使用FIFO原则
|
||
if not pending_msg and self._pending_system_messages[cookie_id]:
|
||
pending_msg = self._pending_system_messages[cookie_id].pop(0)
|
||
logger.info(f"✅ 使用FIFO原则处理待处理的系统消息: {pending_msg['send_message']}")
|
||
|
||
if pending_msg:
|
||
logger.info(f"🔄 开始处理待处理的系统消息: {pending_msg['send_message']}")
|
||
|
||
# 更新订单状态
|
||
success = self.update_order_status(
|
||
order_id=order_id,
|
||
new_status=pending_msg['new_status'],
|
||
cookie_id=cookie_id,
|
||
context=f"{pending_msg['send_message']} - {pending_msg['msg_time']} - 延迟处理"
|
||
)
|
||
|
||
if success:
|
||
status_text = self.status_mapping.get(pending_msg['new_status'], pending_msg['new_status'])
|
||
logger.info(f'✅ [{pending_msg["msg_time"]}] 【{cookie_id}】{pending_msg["send_message"]}，订单 {order_id} 状态已更新为{status_text} (延迟处理)')
|
||
else:
|
||
logger.error(f'❌ [{pending_msg["msg_time"]}] 【{cookie_id}】{pending_msg["send_message"]}，但订单 {order_id} 状态更新失败 (延迟处理)')
|
||
|
||
# 清理临时订单ID的待处理更新
|
||
temp_order_id = pending_msg['temp_order_id']
|
||
if temp_order_id in self.pending_updates:
|
||
del self.pending_updates[temp_order_id]
|
||
logger.info(f"🗑️ 清理临时订单ID {temp_order_id} 的待处理更新")
|
||
|
||
# 如果队列为空，删除该账号的队列
|
||
if not self._pending_system_messages[cookie_id]:
|
||
del self._pending_system_messages[cookie_id]
|
||
logger.info(f"🗑️ 账号 {cookie_id} 的待处理系统消息队列已清空")
|
||
else:
|
||
logger.info(f"ℹ️ 订单 {order_id} ID已提取，但没有找到对应的待处理系统消息")
|
||
else:
|
||
logger.info(f"ℹ️ 账号 {cookie_id} 没有待处理的系统消息")
|
||
|
||
# 处理待处理的红色提醒消息队列
|
||
if cookie_id in self._pending_red_reminder_messages and self._pending_red_reminder_messages[cookie_id]:
|
||
pending_msg = None
|
||
|
||
# 如果提供了消息，尝试匹配
|
||
if message:
|
||
message_hash = hash(str(sorted(message.items()))) if isinstance(message, dict) else hash(str(message))
|
||
# 从后往前遍历，避免pop时索引变化问题
|
||
for i in range(len(self._pending_red_reminder_messages[cookie_id]) - 1, -1, -1):
|
||
msg = self._pending_red_reminder_messages[cookie_id][i]
|
||
if msg.get('message_hash') == message_hash:
|
||
pending_msg = self._pending_red_reminder_messages[cookie_id].pop(i)
|
||
logger.info(f"通过消息哈希匹配到待处理的红色提醒消息: {pending_msg['red_reminder']}")
|
||
break
|
||
|
||
# 如果没有匹配到，使用FIFO原则
|
||
if not pending_msg and self._pending_red_reminder_messages[cookie_id]:
|
||
pending_msg = self._pending_red_reminder_messages[cookie_id].pop(0)
|
||
logger.info(f"使用FIFO原则处理待处理的红色提醒消息: {pending_msg['red_reminder']}")
|
||
|
||
if pending_msg:
|
||
logger.info(f"检测到订单 {order_id} ID已提取，开始处理待处理的红色提醒消息: {pending_msg['red_reminder']}")
|
||
|
||
# 更新订单状态
|
||
success = self.update_order_status(
|
||
order_id=order_id,
|
||
new_status=pending_msg['new_status'],
|
||
cookie_id=cookie_id,
|
||
context=f"{pending_msg['red_reminder']} - 用户{pending_msg['user_id']} - {pending_msg['msg_time']} - 延迟处理"
|
||
)
|
||
|
||
if success:
|
||
status_text = self.status_mapping.get(pending_msg['new_status'], pending_msg['new_status'])
|
||
logger.info(f'[{pending_msg["msg_time"]}] 【{cookie_id}】{pending_msg["red_reminder"]}，订单 {order_id} 状态已更新为{status_text} (延迟处理)')
|
||
else:
|
||
logger.error(f'[{pending_msg["msg_time"]}] 【{cookie_id}】{pending_msg["red_reminder"]}，但订单 {order_id} 状态更新失败 (延迟处理)')
|
||
|
||
# 清理临时订单ID的待处理更新
|
||
temp_order_id = pending_msg['temp_order_id']
|
||
if temp_order_id in self.pending_updates:
|
||
del self.pending_updates[temp_order_id]
|
||
logger.info(f"清理临时订单ID {temp_order_id} 的待处理更新")
|
||
|
||
# 如果队列为空，删除该账号的队列
|
||
if not self._pending_red_reminder_messages[cookie_id]:
|
||
del self._pending_red_reminder_messages[cookie_id]
|
||
else:
|
||
logger.error(f"订单 {order_id} ID已提取，但没有找到对应的待处理红色提醒消息")
|
||
|
||
|
||
# 创建全局实例
|
||
order_status_handler = OrderStatusHandler()
|