File tree Expand file tree Collapse file tree 1 file changed +6
-6
lines changed
src/plugins/bilibili/plugins/dynamic Expand file tree Collapse file tree 1 file changed +6
-6
lines changed Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments