113 lines
3.6 KiB
Python
113 lines
3.6 KiB
Python
"""
|
|
企微内嵌 H5 成功后自动关页。
|
|
|
|
本文件职责：给附件上传成功、选线路成功页一段共用脚本。
|
|
手机端只调一次 WeixinJSBridge.closeWindow，关掉当前 H5 后留在询价对话里。
|
|
再调 ww/wx/window.close 会把询价小助手整个退出，回到企业微信首页。
|
|
电脑企微没有这套桥，加载 wecom-jssdk、用 /inquiry-form/jsapi-signature 注册后再关一次。
|
|
失败、过期页不要挂这段。
|
|
副作用：只产出 HTML 片段；关窗签名走浏览器短请求，不占智能体请求线程做慢活。
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from html import escape
|
|
|
|
from agent.config import get_settings
|
|
|
|
# 签名完成后再稍停，让销售看见「已成功」。
|
|
_CLOSE_DELAY_MS = 400
|
|
|
|
|
|
def wecom_auto_close_script() -> str:
|
|
"""
|
|
成功页脚：加载企微 JS-SDK，register 后关掉当前 H5 窗口。
|
|
|
|
线程：浏览器端执行。签名接口与复制页共用，ticket 已缓存在 Redis。
|
|
为何这样写：上一版只调 WeixinJSBridge，电脑端企微关不掉，会停在成功页。
|
|
"""
|
|
settings = get_settings()
|
|
corp = escape((settings.wecom_corp_id or "").strip(), quote=True)
|
|
agent = escape(str(settings.wecom_agent_id or "1000010").strip(), quote=True)
|
|
delay = _CLOSE_DELAY_MS
|
|
return f"""<script src="https://wwcdn.weixin.qq.com/node/wework/open/js/wecom-jssdk-2.3.2.js"></script>
|
|
<script>
|
|
(function () {{
|
|
var corpId = "{corp}";
|
|
var agentId = Number("{agent}") || 0;
|
|
|
|
function sign(kind) {{
|
|
return function (url) {{
|
|
return fetch("/inquiry-form/jsapi-signature?kind=" + encodeURIComponent(kind)
|
|
+ "&url=" + encodeURIComponent(url))
|
|
.then(function (r) {{
|
|
if (!r.ok) throw new Error("sign");
|
|
return r.json();
|
|
}});
|
|
}};
|
|
}}
|
|
|
|
var closed = false;
|
|
|
|
function closeOnce(fn) {{
|
|
if (closed) return;
|
|
closed = true;
|
|
try {{ fn(); }} catch (e) {{}}
|
|
}}
|
|
|
|
// 手机企微有 WeixinJSBridge：只关当前 H5。再调第二套关闭会把询价小助手整个退出，回到企业微信首页。
|
|
function closeMobile() {{
|
|
closeOnce(function () {{ WeixinJSBridge.call("closeWindow"); }});
|
|
}}
|
|
|
|
// 电脑企微没有 WeixinJSBridge，注册 JS-SDK 后只关一次。
|
|
function closeDesktop() {{
|
|
closeOnce(function () {{
|
|
if (window.ww && typeof ww.closeWindow === "function") ww.closeWindow();
|
|
else if (window.wx && typeof wx.closeWindow === "function") wx.closeWindow();
|
|
}});
|
|
}}
|
|
|
|
function boot() {{
|
|
if (typeof WeixinJSBridge !== "undefined") {{
|
|
setTimeout(closeMobile, {delay});
|
|
return;
|
|
}}
|
|
var sawBridge = false;
|
|
document.addEventListener("WeixinJSBridgeReady", function () {{
|
|
sawBridge = true;
|
|
setTimeout(closeMobile, {delay});
|
|
}}, false);
|
|
var ready = Promise.resolve();
|
|
if (window.ww && ww.register && corpId) {{
|
|
ready = ww.register({{
|
|
corpId: corpId,
|
|
agentId: agentId,
|
|
jsApiList: ["closeWindow"],
|
|
getConfigSignature: sign("corp"),
|
|
getAgentConfigSignature: sign("agent")
|
|
}}).catch(function () {{ return null; }});
|
|
}}
|
|
ready.then(function () {{
|
|
setTimeout(function () {{
|
|
if (sawBridge || typeof WeixinJSBridge !== "undefined") {{
|
|
if (!closed) closeMobile();
|
|
return;
|
|
}}
|
|
closeDesktop();
|
|
}}, {delay});
|
|
}}).catch(function () {{
|
|
setTimeout(function () {{
|
|
if (!closed && typeof WeixinJSBridge === "undefined") closeDesktop();
|
|
}}, {delay});
|
|
}});
|
|
}}
|
|
|
|
if (document.readyState === "loading") {{
|
|
document.addEventListener("DOMContentLoaded", boot);
|
|
}} else {{
|
|
boot();
|
|
}}
|
|
}})();
|
|
</script>"""
|