Pass request to page init callback

README.md

@@ -234,13 +234,13 @@ TWISTED_REACTOR = "twisted.internet.asyncioreactor.AsyncioSelectorReactor"
 * `playwright_page_init_callback` (type `Optional[Union[Callable, str]]`, default `None`)
 
     A coroutine function (`async def`) to be invoked immediately after creating
-    a page for the request. It receives the page as its only positional
-    argument. Useful for initialization code. Invoked only for newly created
+    a page for the request. It receives the page and the request as positional
+    arguments. Useful for initialization code. Invoked only for newly created
     pages, ignored if the page for the request already exists (e.g. by passing
     `playwright_page`).
 
     ```python
-    async def init_page(page):
+    async def init_page(page, request):
         await page.add_init_script(path="./custom_script.js")
 
     class AwesomeSpider(scrapy.Spider):

changelog.md

@@ -5,6 +5,7 @@
 
 * Remove deprecated code (`PageCoroutine` class, `playwright_page_coroutines` request meta key,
   `use_playwright_headers` function).
+* `playwright_page_init_callback` meta key (page initialization callback)
 
 
 ### [v0.0.21](https://github.com/scrapy-plugins/scrapy-playwright/releases/tag/v0.0.21) (2022-08-08)

examples/init_page.py

@@ -3,7 +3,7 @@
 import scrapy
 
 
-async def init_page(page):
+async def init_page(page, request):
     await page.set_extra_http_headers({"Asdf": "Qwerty"})
 
 

scrapy_playwright/handler.py

@@ -191,7 +191,7 @@ async def _create_page(self, request: Request) -> Page:
         if page_init_callback:
             try:
                 page_init_callback = load_object(page_init_callback)
-                await page_init_callback(page)
+                await page_init_callback(page, request)
             except Exception as ex:
                 logger.warning(
                     "[Context=%s] Page init callback exception for %s (%s)",

tests/test_playwright_requests.py

@@ -400,7 +400,7 @@ def should_abort_request_sync(request):
 
     @pytest.mark.asyncio
     async def test_page_initialization_ok(self, caplog):
-        async def init_page(page):
+        async def init_page(page, request):
             await page.set_extra_http_headers({"Extra-Header": "Qwerty"})
 
         settings_dict = {
@@ -421,7 +421,7 @@ async def init_page(page):
 
     @pytest.mark.asyncio
     async def test_page_initialization_fail(self, caplog):
-        async def init_page(page, unused_arg):
+        async def init_page(page, request, unused_arg):
             await page.set_extra_http_headers({"Extra-Header": "Qwerty"})
 
         settings_dict = {