diff --git a/playwright/README.md b/playwright/README.md index 8d1661b..734253e 100644 --- a/playwright/README.md +++ b/playwright/README.md @@ -9,18 +9,20 @@ def Page(): def sync(f): import asyncio, threading, queue, functools; asyncio.set_event_loop_policy(None) def thread(): - while sync.q.get(): f, a, k = sync.p; sync.p = f(*a, **k); sync.q.task_done() - if not hasattr(sync, 'q'): sync.q = queue.Queue(); threading.Thread(target=thread).start() - def wrapper(*a, **k): sync.p = f, a, k; sync.q.put(True); sync.q.join(); return sync.p - return functools.wraps(f)(wrapper) + while sync.q.get(): f,a,k = sync.p; sync.p = f(*a, **k); sync.q.task_done() + if not hasattr(sync, 'q'): + sync.q = queue.Queue(); threading.Thread(target=thread).start() + def fun(*a, **k): sync.p = f,a,k; sync.q.put(True); sync.q.join(); return sync.p + return functools.wraps(f)(fun) @sync def Page(): from playwright.sync_api import sync_playwright as P (context := P().start().firefox.launch().new_context()).set_default_timeout(0) for attr in dir(page := context.new_page()): - if attr[0] != '_' and callable(f := getattr(page, attr)): setattr(page, attr, sync(f)) - page._repr_png_ = page.screenshot - page.goto = lambda url, goto=page.goto: goto(url if '//' in url else f'https://{url}') + if attr[0] != '_' and callable(f := getattr(page, attr)): + setattr(page, attr, sync(f)) + page._repr_png_ = page.screenshot; goto = page.goto + page.goto = lambda url: goto(url if '//' in url else f'https://{url}') return page return Page() page = Page()