chore: refactor mapping wrappers (#93)

Author: pavelfeldman

playwright/accessibility.py

@@ -62,12 +62,6 @@ async def snapshot(
     ) -> Optional[Dict[str, Any]]:
         root = root._channel if root else None
         result = await self._channel.send(
-            "accessibilitySnapshot",
-            dict(root=root, interestingOnly=interestingOnly),
-            unpack_first_key=False,
-        )
-        return (
-            _ax_node_from_protocol(result["rootAXNode"])
-            if result.get("rootAXNode")
-            else None
+            "accessibilitySnapshot", dict(root=root, interestingOnly=interestingOnly),
         )
+        return _ax_node_from_protocol(result) if result else None

playwright/connection.py

@@ -30,21 +30,20 @@ def __init__(self, scope: "ConnectionScope", guid: str) -> None:
         self._guid = guid
         self._object: Optional[ChannelOwner] = None
 
-    async def send(
-        self, method: str, params: dict = None, unpack_first_key: bool = True
-    ) -> Any:
+    async def send(self, method: str, params: dict = None) -> Any:
         if params is None:
             params = dict()
         result = await self._scope.send_message_to_server(self._guid, method, params)
         # Protocol now has named return values, assume result is one level deeper unless
         # there is explicit ambiguity.
-        if unpack_first_key and isinstance(result, dict) and len(result) <= 1:
-            if len(result) == 1:
-                key = next(iter(result))
-                return result[key]
-            else:
-                return None
-        return result
+        if not result:
+            return None
+        assert isinstance(result, dict)
+        if len(result) == 0:
+            return None
+        assert len(result) == 1
+        key = next(iter(result))
+        return result[key]
 
     def send_no_reply(self, method: str, params: dict = None) -> None:
         if params is None: