# utils/chat_utils.py import json import logging import requests from django.conf import settings from orders.models import Dingdan, DingdanPingtai, DingdanShangjia from users.models import UserDashou, UserBoss, UserShangjia from gvsdsdk.models import User logger = logging.getLogger('chat_utils') # ========== 辅助配置获取 ========== def _get_self_goeasy_appkey(): return getattr(settings, 'GOEASY_APPKEY', '') def _get_self_goeasy_secret(): return getattr(settings, 'GOEASY_SECRET', '') # ========== 头像拼接 ========== def _full_local_avatar(relative_url): if not relative_url: return '' if relative_url.startswith('http'): return relative_url return relative_url # ========== GoEasy 订阅 ========== def _subscribe_users_to_group(user_ids, group_ids, appkey=None, secret=None): if not appkey: appkey = _get_self_goeasy_appkey() if not secret: secret = _get_self_goeasy_secret() if not appkey: logger.error("GoEasy AppKey 未配置") return False url = 'https://rest-hangzhou.goeasy.io/v2/im/subscribe-groups' body = {"appkey": appkey, "userIds": user_ids, "groupIds": group_ids} headers = {"Content-Type": "application/json"} if secret: headers["Authorization"] = f"Bearer {secret}" try: resp = requests.post(url, headers=headers, json=body, timeout=10) if resp.status_code == 200: return True logger.error(f"订阅失败: {resp.status_code} {resp.text}") return False except Exception as e: logger.error(f"订阅异常: {e}", exc_info=True) return False # ========== 发送群消息（完整参数） ========== def _send_group_message(appkey, secret, group_id, sender_id, sender_name, sender_avatar, message_text, custom_payload=None, group_name=None, order_id=None): if not appkey: appkey = _get_self_goeasy_appkey() if not secret: secret = _get_self_goeasy_secret() if not appkey: logger.error("GoEasy AppKey 未配置") return False url = 'https://rest-hangzhou.goeasy.io/v2/im/message' # to.data 包含群聊展示信息及业务字段 to_data = { "name": group_name or group_id, "avatar": sender_avatar or "", } if order_id: to_data["orderId"] = order_id payload_str = custom_payload.get("text", message_text) if custom_payload else message_text request_body = { "appkey": appkey, "senderId": sender_id, "senderData": {"avatar": sender_avatar, "name": sender_name}, "to": { "type": "group", "id": group_id, "data": to_data }, "type": "text", "payload": payload_str } headers = {"Content-Type": "application/json"} if secret: headers["Authorization"] = f"Bearer {secret}" try: resp = requests.post(url, headers=headers, json=request_body, timeout=10) if resp.status_code == 200: return True logger.error(f"群聊消息发送失败，状态码:{resp.status_code}，响应:{resp.text}") return False except Exception as e: logger.error(f"发送群聊消息异常: {e}", exc_info=True) return False # ========== 核心入口 ========== def establish_order_chat(dingdan_id): try: order = Dingdan.objects.select_related('pingtai_kuozhan', 'shangjia_kuozhan').get(dingdan_id=dingdan_id) except Dingdan.DoesNotExist: logger.error(f"订单 {dingdan_id} 不存在") return False dashou_uid = order.jiedan_dashou_id dashou_goeasy_id = f"Ds{dashou_uid}" dashou_nickname = f'打手{dashou_uid[:6]}' dashou_avatar_relative = '' try: dashou_user = User.objects.get(UserUID=dashou_uid) dashou_avatar_relative = dashou_user.avatar or '' dashou_profile = UserDashou.objects.get(user=dashou_user) if dashou_profile.nicheng: dashou_nickname = dashou_profile.nicheng except Exception as e: logger.warning(f"获取打手信息失败: {e}") dashou_avatar = _full_local_avatar(dashou_avatar_relative) appkey = _get_self_goeasy_appkey() secret = _get_self_goeasy_secret() if not appkey: logger.error("GoEasy AppKey 未配置") return False # 本地订单处理 return _handle_local_order(order, dashou_goeasy_id, dashou_nickname, dashou_avatar, appkey, secret) # ========== 普通订单 / 我方派单 ========== def _handle_local_order(order, dashou_goeasy_id, dashou_name, dashou_avatar, appkey, secret): group_id = f"group_{order.dingdan_id}" group_name = (order.jieshao[:20] + '…') if order.jieshao and len(order.jieshao) > 20 else (order.jieshao or f"订单{order.dingdan_id}") group_avatar = _full_local_avatar(order.tupian) if order.tupian else _full_local_avatar('') partner_goeasy_id, partner_name, partner_avatar = _get_local_partner_info(order) if not partner_goeasy_id: return False # 订阅双方 if not _subscribe_users_to_group([dashou_goeasy_id, partner_goeasy_id], [group_id], appkey, secret): return False # 1. 打手发送初始化消息 init_msg_text = f"订单已接单，内容：{order.jieshao}，备注：{order.beizhu}，游戏ID：{order.nicheng}" success1 = _send_group_message( appkey, secret, group_id, dashou_goeasy_id, dashou_name, group_avatar, init_msg_text, None, group_name=group_name, order_id=order.dingdan_id ) # 2. 【新增】派单方也发一条消息，确保他也能看到群聊 partner_msg = f"订单已确认，请开始服务。" success2 = _send_group_message( appkey, secret, group_id, partner_goeasy_id, partner_name, partner_avatar, partner_msg, None, group_name=group_name, order_id=order.dingdan_id ) return success1 and success2 def _get_local_partner_info(order): """本地订单下单方：老板或商家""" if order.fadan_pingtai == 1: # 老板 try: ext = order.pingtai_kuozhan laoban_id = ext.laoban_id avatar_full = _full_local_avatar('') nickname = f'老板{laoban_id[:6]}' try: boss_user = User.objects.get(UserUID=laoban_id) avatar_full = _full_local_avatar(boss_user.avatar or '') boss_profile = UserBoss.objects.get(user=boss_user) if boss_profile.nickname: nickname = boss_profile.nickname except Exception: pass return f"Boss{laoban_id}", nickname, avatar_full except Exception as e: logger.error(f"获取老板信息失败: {e}") return None, None, None elif order.fadan_pingtai == 2: # 商家 try: ext = order.shangjia_kuozhan shangjia_id = ext.shangjia_id nickname = ext.sjnicheng or f'商家{shangjia_id[:6]}' avatar_full = _full_local_avatar('') try: sj_user = User.objects.get(UserUID=shangjia_id) avatar_full = _full_local_avatar(sj_user.avatar or '') except Exception: pass return f"Sj{shangjia_id}", nickname, avatar_full except Exception as e: logger.error(f"获取商家信息失败: {e}") return None, None, None else: if order.user1_id: return order.user1_id, "用户", _full_local_avatar('') return None, None, None