Skip to content

Commit 4cf9fe6

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

File tree

1 file changed

+4
-6
lines changed

1 file changed

+4
-6
lines changed

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

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -174,21 +174,19 @@ 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("e => e.parentNode.removeChild(e)")
185184
else:
186-
remove = ""
187185
target = "body"
188186

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

194192
return screenshot

0 commit comments

Comments
 (0)