Skip to content

Commit 477e78b

Browse files
committed
Minor PR cleanup
1 parent 80a86c7 commit 477e78b

File tree

7 files changed

+21
-34
lines changed

7 files changed

+21
-34
lines changed

playwright/_impl/_browser.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -106,7 +106,7 @@ def _did_create_context(self, context: BrowserContext) -> None:
106106

107107
def _setup_browser_context(self, context: BrowserContext) -> None:
108108
context._tracing._traces_dir = self._traces_dir
109-
self._browser_type._playwright.selectors._contextsForSelectors.add(context)
109+
self._browser_type._playwright.selectors._contexts_for_selectors.add(context)
110110

111111
def _on_close(self) -> None:
112112
self._is_connected = False

playwright/_impl/_browser_context.py

Lines changed: 7 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -116,7 +116,7 @@ def __init__(
116116
self._bindings: Dict[str, Any] = {}
117117
self._timeout_settings = TimeoutSettings(None)
118118
self._owner_page: Optional[Page] = None
119-
self._options: Dict[str, Any] = initializer.get("options", {})
119+
self._options: Dict[str, Any] = initializer["options"]
120120
self._background_pages: Set[Page] = set()
121121
self._service_workers: Set[Worker] = set()
122122
self._tracing = cast(Tracing, from_channel(initializer["tracing"]))
@@ -313,9 +313,6 @@ async def _initialize_har_from_options(
313313
if not record_har_path:
314314
return
315315
record_har_path = str(record_har_path)
316-
if len(record_har_path) == 0:
317-
return
318-
# Forcibly provide type to satisfy mypy
319316
default_policy: HarContentPolicy = (
320317
"attach" if record_har_path.endswith(".zip") else "embed"
321318
)
@@ -571,16 +568,15 @@ def expect_event(
571568
def _on_close(self) -> None:
572569
self._closing_or_closed = True
573570
if self._browser:
574-
try:
571+
if self in self._browser._contexts:
575572
self._browser._contexts.remove(self)
576-
except KeyError:
577-
pass
578-
try:
579-
self._browser._browser_type._playwright.selectors._contextsForSelectors.remove(
573+
if (
574+
self
575+
in self._browser._browser_type._playwright.selectors._contexts_for_selectors
576+
):
577+
self._browser._browser_type._playwright.selectors._contexts_for_selectors.remove(
580578
self
581579
)
582-
except KeyError:
583-
pass
584580

585581
self._dispose_har_routers()
586582
self._tracing._reset_stack_counter()

playwright/_impl/_browser_type.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -326,8 +326,10 @@ async def _prepare_browser_context_params(self, params: Dict) -> None:
326326
params["clientCertificates"] = await to_client_certificates_protocol(
327327
params["clientCertificates"]
328328
)
329-
params["selectorEngines"] = self._playwright.selectors._selectorEngines
330-
params["testIdAttributeName"] = self._playwright.selectors._testIdAttributeName
329+
params["selectorEngines"] = self._playwright.selectors._selector_engines
330+
params["testIdAttributeName"] = (
331+
self._playwright.selectors._test_id_attribute_name
332+
)
331333

332334
# Remove HAR options
333335
params.pop("recordHarPath", None)

playwright/_impl/_connection.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -617,10 +617,6 @@ def _filter_none(d: Mapping) -> Dict:
617617
filtered_v = _filter_none(v)
618618
if filtered_v:
619619
result[k] = filtered_v
620-
elif isinstance(v, Mapping):
621-
filtered_v = _filter_none(v)
622-
if filtered_v:
623-
result[k] = filtered_v
624620
else:
625621
result[k] = v
626622
return result

playwright/_impl/_frame.py

Lines changed: 0 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -857,10 +857,3 @@ def _locals_to_params_with_navigation_timeout(self, args: Dict) -> Dict:
857857
params = locals_to_params(args)
858858
params["timeout"] = self._navigation_timeout(params.get("timeout"))
859859
return params
860-
861-
def _locals_to_params_without_timeout(self, args: Dict) -> Dict:
862-
params = locals_to_params(locals())
863-
# Timeout is deprecated and does nothing
864-
if "timeout" in params:
865-
del params["timeout"]
866-
return params

playwright/_impl/_selectors.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -25,10 +25,10 @@
2525
class Selectors:
2626
def __init__(self, loop: asyncio.AbstractEventLoop, dispatcher_fiber: Any) -> None:
2727
self._loop = loop
28-
self._contextsForSelectors: Set[BrowserContext] = set()
29-
self._selectorEngines: List[Dict] = []
28+
self._contexts_for_selectors: Set[BrowserContext] = set()
29+
self._selector_engines: List[Dict] = []
3030
self._dispatcher_fiber = dispatcher_fiber
31-
self._testIdAttributeName: Optional[str] = None
31+
self._test_id_attribute_name: Optional[str] = None
3232

3333
async def register(
3434
self,
@@ -44,16 +44,16 @@ async def register(
4444
engine: Dict[str, Any] = dict(name=name, source=script)
4545
if contentScript:
4646
engine["contentScript"] = contentScript
47-
for context in self._contextsForSelectors:
47+
for context in self._contexts_for_selectors:
4848
await context._channel.send(
49-
"registerSelectorEngine", dict(selectorEngine=engine)
49+
"registerSelectorEngine", {"selectorEngine": engine}
5050
)
51-
self._selectorEngines.append(engine)
51+
self._selector_engines.append(engine)
5252

5353
def set_test_id_attribute(self, attributeName: str) -> None:
5454
set_test_id_attribute_name(attributeName)
55-
self._testIdAttributeName = attributeName
56-
for context in self._contextsForSelectors:
55+
self._test_id_attribute_name = attributeName
56+
for context in self._contexts_for_selectors:
5757
context._channel.send_no_reply(
5858
"setTestIdAttributeName", {"testIdAttributeName": attributeName}
5959
)

tests/sync/test_route_web_socket.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -340,7 +340,7 @@ def _handle_ws(ws: WebSocketRoute) -> None:
340340
f"message: data=echo origin=ws://localhost:{server.PORT} lastEventId=",
341341
],
342342
)
343-
page.close()
343+
context.close()
344344

345345

346346
def test_should_work_with_no_trailing_slash(page: Page, server: Server) -> None:

0 commit comments

Comments
 (0)