use external command

appium · Oct 18, 2024 · 0564815 · 0564815
1 parent 4a10690
commit 0564815
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/appium/webdriver/extensions/context.py b/appium/webdriver/extensions/context.py
@@ -58,8 +58,6 @@ def context(self) -> str:
         return self.current_context
 
     def _add_commands(self) -> None:
-        # noinspection PyProtectedMember,PyUnresolvedReferences
-        commands = self.command_executor._commands
-        commands[Command.CONTEXTS] = ('GET', '/session/$sessionId/contexts')
-        commands[Command.GET_CURRENT_CONTEXT] = ('GET', '/session/$sessionId/context')
-        commands[Command.SWITCH_TO_CONTEXT] = ('POST', '/session/$sessionId/context')
+        self.command_executor.extra_commands[Command.CONTEXTS] = ('GET', '/session/$sessionId/contexts')
+        self.command_executor.extra_commands[Command.GET_CURRENT_CONTEXT] = ('GET', '/session/$sessionId/context')
+        self.command_executor.extra_commands[Command.SWITCH_TO_CONTEXT] = ('POST', '/session/$sessionId/context')