Source code for

# Licensed to the Software Freedom Conservancy (SFC) under one
# or more contributor license agreements.  See the NOTICE file
# distributed with this work for additional information
# regarding copyright ownership.  The SFC licenses this file
# to you under the Apache License, Version 2.0 (the
# "License"); you may not use this file except in compliance
# with the License.  You may obtain a copy of the License at
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# KIND, either express or implied.  See the License for the
# specific language governing permissions and limitations
# under the License.

from .command import Command

class _ConnectionType:
    def __init__(self, mask):
        self.mask = mask

    def airplane_mode(self):
        return self.mask % 2 == 1

    def wifi(self):
        return (self.mask / 2) % 2 == 1

    def data(self):
        return (self.mask / 4) > 0

[docs]class Mobile: ConnectionType = _ConnectionType ALL_NETWORK = ConnectionType(6) WIFI_NETWORK = ConnectionType(2) DATA_NETWORK = ConnectionType(4) AIRPLANE_MODE = ConnectionType(1) def __init__(self, driver): import weakref self._driver = weakref.proxy(driver) @property def network_connection(self): return self.ConnectionType(self._driver.execute(Command.GET_NETWORK_CONNECTION)["value"])
[docs] def set_network_connection(self, network): """Set the network connection for the remote device. Example of setting airplane mode:: """ mode = network.mask if isinstance(network, self.ConnectionType) else network return self.ConnectionType( self._driver.execute( Command.SET_NETWORK_CONNECTION, {"name": "network_connection", "parameters": {"type": mode}} )["value"] )
@property def context(self): """Returns the current context (Native or WebView).""" return self._driver.execute(Command.CURRENT_CONTEXT_HANDLE) @context.setter def context(self, new_context) -> None: """Sets the current context.""" self._driver.execute(Command.SWITCH_TO_CONTEXT, {"name": new_context}) @property def contexts(self): """Returns a list of available contexts.""" return self._driver.execute(Command.CONTEXT_HANDLES)