2021-01-01 15:17:27 -08:00
|
|
|
---
|
|
|
|
id: extensibility
|
|
|
|
title: "Extensibility"
|
|
|
|
---
|
2020-12-30 18:04:51 -08:00
|
|
|
|
2021-01-01 15:17:27 -08:00
|
|
|
<!-- TOC -->
|
2020-12-30 18:04:51 -08:00
|
|
|
|
|
|
|
## Custom selector engines
|
|
|
|
|
|
|
|
Playwright supports custom selector engines, registered with [`method: Selectors.register`].
|
|
|
|
|
|
|
|
Selector engine should have the following properties:
|
2021-01-14 15:01:39 -08:00
|
|
|
- `create` function to create a relative selector from `root` (root is either a `Document`, `ShadowRoot` or `Element`)
|
|
|
|
to a `target` element.
|
2020-12-30 18:04:51 -08:00
|
|
|
- `query` function to query first element matching `selector` relative to the `root`.
|
|
|
|
- `queryAll` function to query all elements matching `selector` relative to the `root`.
|
|
|
|
|
2021-01-14 15:01:39 -08:00
|
|
|
By default the engine is run directly in the frame's JavaScript context and, for example, can call an
|
|
|
|
application-defined function. To isolate the engine from any JavaScript in the frame, but leave access to the DOM,
|
|
|
|
register the engine with `{contentScript: true}` option. Content script engine is safer because it is protected from any
|
|
|
|
tampering with the global objects, for example altering `Node.prototype` methods. All built-in selector engines run as
|
|
|
|
content scripts. Note that running as a content script is not guaranteed when the engine is used together with other
|
|
|
|
custom engines.
|
2020-12-30 18:04:51 -08:00
|
|
|
|
|
|
|
An example of registering selector engine that queries elements based on a tag name:
|
2021-01-14 15:01:39 -08:00
|
|
|
|
2020-12-30 18:04:51 -08:00
|
|
|
```js
|
|
|
|
// Must be a function that evaluates to a selector engine instance.
|
|
|
|
const createTagNameEngine = () => ({
|
|
|
|
// Returns the first element matching given selector in the root's subtree.
|
|
|
|
query(root, selector) {
|
|
|
|
return root.querySelector(selector);
|
|
|
|
},
|
|
|
|
|
|
|
|
// Returns all elements matching given selector in the root's subtree.
|
|
|
|
queryAll(root, selector) {
|
|
|
|
return Array.from(root.querySelectorAll(selector));
|
|
|
|
}
|
|
|
|
});
|
|
|
|
|
|
|
|
// Register the engine. Selectors will be prefixed with "tag=".
|
|
|
|
await selectors.register('tag', createTagNameEngine);
|
|
|
|
|
|
|
|
// Now we can use 'tag=' selectors.
|
|
|
|
const button = await page.$('tag=button');
|
|
|
|
|
|
|
|
// We can combine it with other selector engines using `>>` combinator.
|
|
|
|
await page.click('tag=div >> span >> "Click me"');
|
|
|
|
|
|
|
|
// We can use it in any methods supporting selectors.
|
|
|
|
const buttonCount = await page.$$eval('tag=button', buttons => buttons.length);
|
|
|
|
```
|
2021-01-14 15:01:39 -08:00
|
|
|
|
2021-03-01 09:18:44 -08:00
|
|
|
```java
|
|
|
|
// Must be a script that evaluates to a selector engine instance.
|
|
|
|
String createTagNameEngine = "{\n" +
|
|
|
|
" // Returns the first element matching given selector in the root's subtree.\n" +
|
|
|
|
" query(root, selector) {\n" +
|
|
|
|
" return root.querySelector(selector);\n" +
|
|
|
|
" },\n" +
|
|
|
|
"\n" +
|
|
|
|
" // Returns all elements matching given selector in the root's subtree.\n" +
|
|
|
|
" queryAll(root, selector) {\n" +
|
|
|
|
" return Array.from(root.querySelectorAll(selector));\n" +
|
|
|
|
" }\n" +
|
|
|
|
"}";
|
|
|
|
|
|
|
|
// Register the engine. Selectors will be prefixed with "tag=".
|
|
|
|
playwright.selectors().register("tag", createTagNameEngine);
|
|
|
|
|
|
|
|
// Now we can use "tag=" selectors.
|
|
|
|
ElementHandle button = page.querySelector("tag=button");
|
|
|
|
|
|
|
|
// We can combine it with other selector engines using ">>" combinator.
|
|
|
|
page.click("tag=div >> span >> \"Click me\"");
|
|
|
|
|
|
|
|
// We can use it in any methods supporting selectors.
|
|
|
|
int buttonCount = (int) page.evalOnSelectorAll("tag=button", "buttons => buttons.length");
|
|
|
|
```
|
|
|
|
|
2021-01-14 15:01:39 -08:00
|
|
|
```python async
|
|
|
|
tag_selector = """
|
|
|
|
// Must evaluate to a selector engine instance.
|
|
|
|
{
|
|
|
|
// Returns the first element matching given selector in the root's subtree.
|
|
|
|
query(root, selector) {
|
|
|
|
return root.querySelector(selector);
|
|
|
|
},
|
|
|
|
|
|
|
|
// Returns all elements matching given selector in the root's subtree.
|
|
|
|
queryAll(root, selector) {
|
|
|
|
return Array.from(root.querySelectorAll(selector));
|
|
|
|
}
|
|
|
|
}"""
|
|
|
|
|
|
|
|
# register the engine. selectors will be prefixed with "tag=".
|
|
|
|
await playwright.selectors.register("tag", tag_selector)
|
|
|
|
|
|
|
|
# now we can use "tag=" selectors.
|
|
|
|
button = await page.query_selector("tag=button")
|
|
|
|
|
|
|
|
# we can combine it with other selector engines using `>>` combinator.
|
|
|
|
await page.click("tag=div >> span >> "click me"")
|
|
|
|
|
|
|
|
# we can use it in any methods supporting selectors.
|
|
|
|
button_count = await page.eval_on_selector_all("tag=button", buttons => buttons.length)
|
|
|
|
```
|
|
|
|
|
|
|
|
```python sync
|
|
|
|
tag_selector = """
|
|
|
|
// Must evaluate to a selector engine instance.
|
|
|
|
{
|
|
|
|
// Returns the first element matching given selector in the root's subtree.
|
|
|
|
query(root, selector) {
|
|
|
|
return root.querySelector(selector);
|
|
|
|
},
|
|
|
|
|
|
|
|
// Returns all elements matching given selector in the root's subtree.
|
|
|
|
queryAll(root, selector) {
|
|
|
|
return Array.from(root.querySelectorAll(selector));
|
|
|
|
}
|
|
|
|
}"""
|
|
|
|
|
|
|
|
# register the engine. selectors will be prefixed with "tag=".
|
|
|
|
playwright.selectors.register("tag", tag_selector)
|
|
|
|
|
|
|
|
# now we can use "tag=" selectors.
|
|
|
|
button = page.query_selector("tag=button")
|
|
|
|
|
|
|
|
# we can combine it with other selector engines using `>>` combinator.
|
|
|
|
page.click("tag=div >> span >> "click me"")
|
|
|
|
|
|
|
|
# we can use it in any methods supporting selectors.
|
|
|
|
button_count = page.eval_on_selector_all("tag=button", buttons => buttons.length)
|
|
|
|
```
|