^jK*~ddlmZmZmZddlmZmZddlmZddl m Z ddl m Z ddl mZddlmZGd d e e eZy ) )AnyDictUnion)InvalidArgumentExceptionUnknownMethodException)Self)CanExecuteCommands)CanExecuteScripts)CanRememberExtensionPresence) MobileCommandc eZdZdedefdZdedefdZdede defdZ d ede defd Z d ede defd Z d edefd Z d edefd ZddeedfdeedfdeeeffdZddZy) Applicationssecondsreturncd}d|i} |j|j|||S#t$r3|j|j t j |Y|SwxYw)aPuts the application in the background on the device for a certain duration. Args: seconds: the duration for the application to remain in the background. Providing a negative value will continue immediately after putting the app under test to the background. Returns: Union['WebDriver', 'Applications']: Self instance zmobile: backgroundAppr)assert_extension_existsexecute_scriptrmark_extension_absenceexecuteCommand BACKGROUND)selfrext_nameargss \/home/ubuntu/.local/lib/python3.12/site-packages/appium/webdriver/extensions/applications.pybackground_appzApplications.background_appst+7# T  ( ( 2 A A(D Q & T  ' ' 1 9 9':L:Ld S  Ts!+8A'&A' bundle_idcd} |j|j|||dS#ttf$r7|j |j t jd|idcYSwxYw)zChecks whether the application specified by `bundle_id` is installed on the device. Args: bundle_id: the id of the application to query Returns: `True` if app is installed zmobile: isAppInstalled)bundleIdappIdr value)rrrrrrrIS_APP_INSTALLED)rrrs ris_app_installedzApplications.is_app_installed/s, //9HH )& '(@A ..x8@@((     #(AA.-A.app_pathoptionsc d} |j|jd||d|xsi|S#ttf$rLd|i}|r|j d|i|j |j tj|Y|SwxYw)aInstall the application found at `app_path` on the device. Args: app_path: the local or remote path to the application to install Keyword Args: replace (bool): [Android only] whether to reinstall/upgrade the package if it is already present on the device under test. True by default timeout (int): [Android only] how much time to wait for the installation to complete. 60000ms by default. allowTestPackages (bool): [Android only] whether to allow installation of packages marked as test in the manifest. False by default useSdcard (bool): [Android only] whether to use the SD card to install the app. False by default grantPermissions (bool): [Android only] whether to automatically grant application permissions on Android 6+ after the installation completes. False by default Returns: Union['WebDriver', 'Applications']: Self instance zmobile: installApp)appappPathr*r') rrrrupdaterrr INSTALL_APP)rr&r'rdatas r install_appzApplications.install_appJs(( U  ( ( 2 A A$#'}"  '(@A U$-x#8D Y01  ' ' 1 9 9':M:Mt T  U*0AB  B app_idc d} |j|j|||d|xsi|S#ttf$rLd|i}|r|j d|i|j |j tj|Y|SwxYw)aRemove the specified application from the device. Args: app_id: the application id to be removed Keyword Args: keepData (bool): [Android only] whether to keep application data and caches after it is uninstalled. False by default timeout (int): [Android only] how much time to wait for the uninstall to complete. 20000ms by default. Returns: Union['WebDriver', 'Applications']: Self instance zmobile: removeAppr!r r!r') rrrrr+rrr REMOVE_APPrr0r'rr-s r remove_appzApplications.remove_appps' T  ( ( 2 A A# &}"  '(@A T$+V#4D Y01  ' ' 1 9 9':L:Ld S  Tr/c d} |j|j|||d|xsiS#ttf$rNd|i}|r|j d|i|j |j tj|dcYSwxYw)aeTerminates the application if it is running. Args: app_id: the application id to be terminates Keyword Args: `timeout` (int): [Android only] how much time to wait for the uninstall to complete. 500ms by default. Returns: True if the app has been successfully terminated zmobile: terminateAppr2r!r'r") rrrrr+rrr TERMINATE_APPr4s r terminate_appzApplications.terminate_apps* g//9HH# &}" '(@A g$+V#4D Y01..x8@@AVAVX\]^ef f  gs).AB  B cd} |j|j|||d|S#ttf$r5|j |j t jd|iY|SwxYw)zActivates the application if it is not running or is running in the background. Args: app_id: the application id to be activated Returns: Union['WebDriver', 'Applications']: Self instance zmobile: activateAppr2r!)rrrrrrr ACTIVATE_APPrr0rs r activate_appzApplications.activate_apps) c  ( ( 2 A A# &  '(@A c  ' ' 1 9 9':N:NQXZ`Pa b  cs$*AA.-A.cd} |j|j|||dS#ttf$r7|j |j t jd|idcYSwxYw)zQueries the state of the application. Args: app_id: the application id to be queried Returns: One of possible application state constants. See ApplicationState class for more details. zmobile: queryAppStater2r!r")rrrrrrrQUERY_APP_STATEr;s rquery_app_statezApplications.query_app_states+ //9HH# & '(@A ..x8@@''V    r%Nlanguage string_filecd}i}|||d<|||d< |j|j||S#t$r5|j|j t j |dcYSwxYw)aIReturns the application strings from the device for the specified language. Args: language: strings language code string_file: the name of the string file to query. Only relevant for XCUITest driver Returns: The key is string id and the value is the content. zmobile: getAppStringsr@ stringFiler")rrrrrrGET_APP_STRINGS)rr@rArr-s r app_stringszApplications.app_stringss+  'D   "!,D  i//9HHSWX X% i..x8@@AXAXZ^_`gh h is 5;A32A3c|jjtjdd|jjtjdd|jjtj dd|jjtj dd|jjtjdd|jjtjdd|jjtjdd|jjtjdd y) NPOSTz)/session/$sessionId/appium/app/backgroundz//session/$sessionId/appium/device/app_installedz-/session/$sessionId/appium/device/install_appz,/session/$sessionId/appium/device/remove_appz//session/$sessionId/appium/device/terminate_appz./session/$sessionId/appium/device/activate_appz+/session/$sessionId/appium/device/app_statez&/session/$sessionId/appium/app/strings) command_executor add_commandrrr#r,r3r7r:r>rD)rs r _add_commandszApplications._add_commandss ))'*<*rZs6$#W"NLc4xu%'8:VxurK