Update playwright/README.md

This commit is contained in:
2024-12-24 06:47:59 +00:00
parent 8ef1bdb65b
commit 212bbdf2f5

View File

@@ -5,33 +5,34 @@ Linux (WSL Debian)
pip install playwright pip install playwright
playwright install --with-deps playwright install --with-deps
""" """
# %% def Page():
def sync(coro): def sync(func):
import asyncio, functools if not hasattr(sync, 'call'):
if not asyncio.iscoroutinefunction(coro): return coro sync.call = __import__('queue').Queue()
@functools.wraps(coro) def server():
while sync.call.get():
func, args, kwargs = sync.obj
sync.obj = func(*args, **kwargs)
sync.call.task_done()
__import__('asyncio').set_event_loop_policy(None)
__import__('threading').Thread(target=server).start()
@__import__('functools').wraps(func)
def wrapper(*args, **kwargs): def wrapper(*args, **kwargs):
loop, future = asyncio.get_event_loop(), asyncio.ensure_future(coro(*args, **kwargs)) sync.obj = func, args, kwargs
while not future.done(): sync.call.put(True); sync.call.join()
loop._process_events(loop._selector.select(0)) return sync.obj
if (ready := loop._ready) and (handle := ready.popleft())._cancelled is False:
task = (tasks := asyncio.tasks._current_tasks).pop(loop, None)
handle._run(); tasks[loop] = task
return future.result()
return wrapper return wrapper
@sync @sync
async def Page(headless=True): def Page():
from playwright.async_api import async_playwright from playwright.sync_api import sync_playwright
browser = await (await async_playwright().start()).firefox.launch(headless=headless) browser = sync_playwright().start().firefox.launch()
(page := await browser.new_page()).set_default_timeout(0) (context := browser.new_context()).set_default_timeout(0)
for attr in dir(page): for attr in dir(page := context.new_page()):
if callable(method := getattr(page, attr)): setattr(page, attr, sync(method)) if attr[0] != '_' and callable(method := getattr(page, attr)):
try: setattr(page, attr, sync(method))
from IPython.display import Image page._repr_png_ = page.screenshot
page.goto = lambda url, goto=page.goto: goto(url) and Image(page.screenshot())
finally:
return page return page
return Page()
(page := Page()).goto('https://example.org') (page := Page()).goto('https://example.org')
print(page.title()) # Example Domain print(page.title()) # Example Domain
``` ```