1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
| async def control_playback(ws, session_id, action="toggle", index=0):
"""控制媒体播放"""
js = {
"play": f"document.querySelectorAll('video,audio')[{index}].play()",
"pause": f"document.querySelectorAll('video,audio')[{index}].pause()",
"toggle": f"""
(()=>{{const e=document.querySelectorAll('video,audio')[{index}];
e.paused?e.play():e.pause();return e.paused?'paused':'playing';}})()"""
}[action]
result = await cdp(ws, "Runtime.evaluate", {
"expression": js, "returnByValue": True
}, session_id=session_id)
return result.get("result", {}).get("value")
async def capture_video_frame(ws, session_id, index=0):
"""捕获视频帧（Canvas 方案）"""
js = f"""
(()=>{{const v=document.querySelectorAll('video')[{index}];
if(!v)return null;const c=document.createElement('canvas');
c.width=v.videoWidth||v.clientWidth;
c.height=v.videoHeight||v.clientHeight;
c.getContext('2d').drawImage(v,0,0);
return c.toDataURL('image/png');}})()
"""
result = await cdp(ws, "Runtime.evaluate", {
"expression": js, "returnByValue": True
}, session_id=session_id)
data_url = result.get("result", {}).get("value", "")
if data_url.startswith("data:image"):
import base64
return base64.b64decode(data_url.split(",")[1])
return None
async def save_video_frame(ws, session_id, path, index=0):
"""保存视频帧到文件"""
img = await capture_video_frame(ws, session_id, index)
if img:
with open(path, "wb") as f:
f.write(img)
print(f"帧已保存: {path} ({len(img)} bytes)")
return True
return False
async def inspect_media_element(ws, session_id, tag="video", index=0):
"""检查媒体元素状态"""
result = await cdp(ws, "Runtime.evaluate", {
"expression": f"""
(()=>{{const el=document.querySelectorAll('{tag}')[{index}];
if(!el)return null;
return{{src:el.currentSrc,duration:el.duration,
currentTime:el.currentTime,paused:el.paused,ended:el.ended,
volume:el.volume,readyState:el.readyState,
videoWidth:el.videoWidth,videoHeight:el.videoHeight}};}})()
""",
"returnByValue": True
}, session_id=session_id)
return result.get("result", {}).get("value")
|
