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