Skip to content

Commit 94aa95c

Browse files
committed
fix(bilibili): remove all elements except target
1 parent 1bf1e72 commit 94aa95c

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

src/plugins/bilibili/plugins/dynamic/__init__.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -174,21 +174,21 @@ async def render_screenshot(dynamic: Dynamic) -> bytes:
174174
await page.wait_for_load_state("networkidle")
175175

176176
if "opus" in page.url:
177-
remove = ".opus-nav,.opus-float-btn,.openapp-dialog,.opus-read-more"
178177
target = ".opus-modules"
179178
await page.locator(".opus-module-content").evaluate(
180179
"e => e.classList.remove('limit')"
181180
)
182181
elif "dynamic" in page.url:
183-
remove = ".m-navbar,.dynamic-float-btn,.dyn-share"
184182
target = ".dyn-card"
183+
await page.locator(".dyn-share").evaluate(
184+
"e => e.parentNode.removeChild(e)"
185+
)
185186
else:
186-
remove = ""
187187
target = "body"
188188

189-
await page.locator(remove).evaluate_all(
190-
"es => es.forEach(e => e.parentNode.removeChild(e))"
191-
)
189+
await page.locator(
190+
f"body :not({target}, :has({target}), {target} *)"
191+
).evaluate_all("es => es.forEach(e => e.parentNode.removeChild(e))")
192192
screenshot = await page.locator(target).first.screenshot()
193193

194194
return screenshot

0 commit comments

Comments
 (0)