Update playwright/README.md
This commit is contained in:
@@ -6,28 +6,32 @@ pip install playwright
|
|||||||
playwright install --with-deps
|
playwright install --with-deps
|
||||||
"""
|
"""
|
||||||
# %%
|
# %%
|
||||||
|
def sync(coro):
|
||||||
import asyncio, functools
|
import asyncio, functools
|
||||||
def _run_once(loop):
|
if not asyncio.iscoroutinefunction(coro): return coro
|
||||||
|
@functools.wraps(coro)
|
||||||
|
def wrapper(*args, **kwargs):
|
||||||
|
loop, future = asyncio.get_event_loop(), asyncio.ensure_future(coro(*args, **kwargs))
|
||||||
|
while not future.done():
|
||||||
loop._process_events(loop._selector.select(0))
|
loop._process_events(loop._selector.select(0))
|
||||||
if (ready := loop._ready) and (handle := ready.popleft())._cancelled is False:
|
if (ready := loop._ready) and (handle := ready.popleft())._cancelled is False:
|
||||||
task = (tasks := asyncio.tasks._current_tasks).pop(loop, None)
|
task = (tasks := asyncio.tasks._current_tasks).pop(loop, None)
|
||||||
handle._run(); tasks[loop] = task
|
handle._run(); tasks[loop] = task
|
||||||
(loop := asyncio.get_event_loop()).__class__._run_once = _run_once
|
return future.result()
|
||||||
asyncio.run = loop.run_until_complete; loop.__class__._check_running = lambda _: None
|
return wrapper
|
||||||
|
|
||||||
@(sync := lambda coro: functools.wraps(coro)(
|
@sync
|
||||||
lambda *args, **kwargs: asyncio.run(coro(*args, **kwargs))))
|
|
||||||
async def Page(headless=True):
|
async def Page(headless=True):
|
||||||
from playwright.async_api import async_playwright
|
from playwright.async_api import async_playwright
|
||||||
browser = await (await async_playwright().start()).firefox.launch(headless=headless)
|
browser = await (await async_playwright().start()).firefox.launch(headless=headless)
|
||||||
(page := await browser.new_page()).set_default_timeout(0)
|
(page := await browser.new_page()).set_default_timeout(0)
|
||||||
for attr in dir(page):
|
for attr in dir(page):
|
||||||
if asyncio.iscoroutinefunction(method := getattr(page, attr)):
|
if callable(method := getattr(page, attr)): setattr(page, attr, sync(method))
|
||||||
setattr(page, attr, sync(method))
|
|
||||||
try:
|
try:
|
||||||
from IPython.display import Image
|
from IPython.display import Image
|
||||||
page.goto = lambda url, goto=page.goto: goto(url) and Image(page.screenshot())
|
page.goto = lambda url, goto=page.goto: goto(url) and Image(page.screenshot())
|
||||||
finally: return page
|
finally:
|
||||||
|
return page
|
||||||
(page := Page()).goto('https://example.org')
|
(page := Page()).goto('https://example.org')
|
||||||
print(page.title()) # Example Domain
|
print(page.title()) # Example Domain
|
||||||
```
|
```
|
||||||
Reference in New Issue
Block a user