chore(deps): update mypy requirement from ~=0.800 to ~=0.812 (appium#589)

dependabot[bot] · ki4070ma · web-flow · commit db035dd0f4cc · 2021-02-24T23:49:55.000+09:00
* chore(deps): update mypy requirement from ~=0.800 to ~=0.812 Updates the requirements on [mypy](https://github.com/python/mypy) to permit the latest version. - [Release notes](https://github.com/python/mypy/releases) - [Commits](python/mypy@v0.800...v0.812) Signed-off-by: dependabot[bot] <support@github.com> * Fix mypy error with mypy v0.812 (appium#590) * chore(deps): update mypy requirement from ~=0.800 to ~=0.812 Updates the requirements on [mypy](https://github.com/python/mypy) to permit the latest version. - [Release notes](https://github.com/python/mypy/releases) - [Commits](python/mypy@v0.800...v0.812) Signed-off-by: dependabot[bot] <support@github.com> * fix mypy error * review comment Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com> Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com> Co-authored-by: Mori Atsushi <ki4070ma@gmail.com>
diff --git a/Pipfile b/Pipfile
@@ -24,4 +24,4 @@ pylint = "~=2.6"
 astroid = "~=2.4"
 isort = "~=5.7"
 
-mypy = "~=0.800"
+mypy = "~=0.812"
diff --git a/appium/webdriver/switch_to.py b/appium/webdriver/switch_to.py
@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from typing import TypeVar
+from typing import Optional, TypeVar
 
 from selenium.webdriver.remote.switch_to import SwitchTo
 
@@ -22,8 +22,9 @@
 
 
 class MobileSwitchTo(SwitchTo):
-    def context(self, context_name: str) -> T:
+    def context(self, context_name: Optional[str]) -> T:
         """Sets the context for the current session.
+        Passing `None` is equal to switching to native context.
 
         Args:
             context_name: The name of the context to switch to.
diff --git a/appium/webdriver/webdriver.py b/appium/webdriver/webdriver.py
@@ -160,7 +160,6 @@ def __init__(
             self._addCommands()
 
         self.error_handler = MobileErrorHandler()
-        self._switch_to = MobileSwitchTo(self)
 
         if direct_connection:
             self._update_command_executor(keep_alive=keep_alive)
@@ -344,6 +343,19 @@ def set_value(self, element: MobileWebElement, value: str) -> T:
         self.execute(Command.SET_IMMEDIATE_VALUE, data)
         return self
 
+    @property
+    def switch_to(self) -> MobileSwitchTo:
+        """Returns an object containing all options to switch focus into
+
+        Override for appium
+
+        Returns:
+            `appium.webdriver.switch_to.MobileSwitchTo`
+
+        """
+
+        return MobileSwitchTo(self)
+
     # pylint: disable=protected-access
 
     def _addCommands(self) -> None:
