89 lines
3.3 KiB
Python
89 lines
3.3 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
# -*- coding: utf-8 -*-
|
||
|
|
"""
|
||
|
|
映射某个顶导航模块的侧边栏功能结构。
|
||
|
|
用法: python3 map_menu.py "<模块名>" [in_more]
|
||
|
|
in_more: 传 1 表示该模块在"更多"下拉里, 需先展开
|
||
|
|
输出: JSON {module, loc, groups:[{title, items:[{name,href}]}], flat:[...]}
|
||
|
|
"""
|
||
|
|
import sys, json, time, urllib.request
|
||
|
|
|
||
|
|
DAEMON = "http://127.0.0.1:10086/command"
|
||
|
|
SESSION = "2haohr"
|
||
|
|
|
||
|
|
def call(action, args):
|
||
|
|
body = json.dumps({"action": action, "args": args, "session": SESSION}).encode()
|
||
|
|
req = urllib.request.Request(DAEMON, data=body, headers={"Content-Type": "application/json"})
|
||
|
|
with urllib.request.urlopen(req, timeout=40) as resp:
|
||
|
|
return json.loads(resp.read().decode())
|
||
|
|
|
||
|
|
def ev(code):
|
||
|
|
return call("evaluate", {"code": code}).get("data", {}).get("value")
|
||
|
|
|
||
|
|
CLICK_TOP = """
|
||
|
|
(()=>{const name=%s;
|
||
|
|
const els=[...document.querySelectorAll("a,span,div,li")].filter(e=>{
|
||
|
|
const t=(e.innerText||"").trim();const r=e.getBoundingClientRect();
|
||
|
|
return t===name && r.top<60 && r.width>8 && r.width<160;});
|
||
|
|
if(els.length){els[0].click();return "top-clicked";}return "not-found";})()
|
||
|
|
"""
|
||
|
|
|
||
|
|
OPEN_MORE = """
|
||
|
|
(()=>{const els=[...document.querySelectorAll("a,span,div,li")].filter(e=>{
|
||
|
|
const t=(e.innerText||"").trim();const r=e.getBoundingClientRect();
|
||
|
|
return t==="更多" && r.top<60 && r.width>8 && r.width<160;});
|
||
|
|
if(!els.length)return "no-more";
|
||
|
|
const el=els[0];
|
||
|
|
["pointerover","mouseover","mouseenter","pointerenter"].forEach(tp=>{
|
||
|
|
el.dispatchEvent(new MouseEvent(tp,{bubbles:true,cancelable:true,view:window}));});
|
||
|
|
el.click();
|
||
|
|
return "more-hover";})()
|
||
|
|
"""
|
||
|
|
|
||
|
|
CLICK_ANY = """
|
||
|
|
(()=>{const name=%s;
|
||
|
|
const els=[...document.querySelectorAll("a,span,div,li")].filter(e=>(e.innerText||"").trim()===name && e.getBoundingClientRect().width>8);
|
||
|
|
if(els.length){els[els.length-1].click();return "clicked";}return "not-found";})()
|
||
|
|
"""
|
||
|
|
|
||
|
|
EXTRACT = r"""
|
||
|
|
(()=>{
|
||
|
|
const root=document.querySelector("[class*=leftMenu]")||document.querySelector("[class*=menu_]");
|
||
|
|
if(!root)return JSON.stringify({loc:location.href,groups:[],flat:[]});
|
||
|
|
const groups=[]; let cur={title:"",items:[]}; groups.push(cur);
|
||
|
|
const seen=new Set();
|
||
|
|
const walk=(el)=>{
|
||
|
|
for(const c of el.children){
|
||
|
|
const cls=(c.className||"").toString();
|
||
|
|
if(/nameTitle/.test(cls) && !c.closest("a")){ const t=(c.innerText||"").trim(); if(t){ cur={title:t,items:[]}; groups.push(cur);} }
|
||
|
|
if(c.tagName==="A"){ const h=c.getAttribute("href")||""; const t=(c.innerText||"").trim().replace(/\s+/g," ");
|
||
|
|
if(h&&h.length>1&&t&&!seen.has(h)){seen.add(h);cur.items.push({name:t,href:h});} }
|
||
|
|
walk(c);
|
||
|
|
}
|
||
|
|
};
|
||
|
|
walk(root);
|
||
|
|
const flat=[]; groups.forEach(g=>g.items.forEach(it=>flat.push({group:g.title,name:it.name,href:it.href})));
|
||
|
|
return JSON.stringify({loc:location.href,groups:groups.filter(g=>g.items.length),flat});
|
||
|
|
})()
|
||
|
|
"""
|
||
|
|
|
||
|
|
def main():
|
||
|
|
module = sys.argv[1]
|
||
|
|
in_more = len(sys.argv) > 2 and sys.argv[2] == "1"
|
||
|
|
name_js = json.dumps(module, ensure_ascii=False)
|
||
|
|
if in_more:
|
||
|
|
print("more:", ev(OPEN_MORE)); time.sleep(1.2)
|
||
|
|
print("click:", ev(CLICK_ANY % name_js))
|
||
|
|
else:
|
||
|
|
r = ev(CLICK_TOP % name_js)
|
||
|
|
print("top:", r)
|
||
|
|
if r == "not-found":
|
||
|
|
print("more:", ev(OPEN_MORE)); time.sleep(1.2)
|
||
|
|
print("click:", ev(CLICK_ANY % name_js))
|
||
|
|
time.sleep(3.0)
|
||
|
|
data = ev(EXTRACT)
|
||
|
|
print(data)
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
main()
|