^j~ddlZddlmZddlmZmZddlmZddlm Z ddl m Z ddl m Z dd lmZGd d e e e Zy) N)Optional)InvalidArgumentExceptionUnknownMethodException)Self)CanExecuteCommands)CanExecuteScripts)CanRememberExtensionPresence) MobileCommandc ZeZdZdedefdZdedefdZd dedeedeedefd Zd d Z y) RemoteFSpathreturncd} |j|j|d|iS#t$r7|j|j t j d|idcYSwxYw)zRetrieves the file at `path`. Args: path: the path to the file on the device Returns: The file's contents encoded as Base64. zmobile: pullFile remotePathrvalue)assert_extension_existsexecute_scriptrmark_extension_absenceexecuteCommand PULL_FILEselfrext_names Y/home/ubuntu/.local/lib/python3.12/site-packages/appium/webdriver/extensions/remote_fs.py pull_filezRemoteFS.pull_filesv& m//9HHT`bfSgh h% m..x8@@ARARU[]aTbcdkl l m"'=A'&A'cd} |j|j|d|iS#t$r7|j|j t j d|idcYSwxYw)zRetrieves a folder at `path`. Args: path: the path to the folder on the device Returns: The folder's contents zipped and encoded as Base64. zmobile: pullFolderrrr)rrrrrr PULL_FOLDERrs r pull_folderzRemoteFS.pull_folder-sv( o//9HHT`bfSgh h% o..x8@@ATATW]_cVdefmn n orNdestination_path base64data source_pathc| | td|J t|d5}|j}dddt j j d}d} |j|j|||d|S#1swYVxYw#t$r}d|d}t||d}~wwxYw#t$r6|j|jtj||d Y|SwxYw) a]Puts the data from the file at `source_path`, encoded as Base64, in the file specified as `path`. Specify either `base64data` or `source_path`, if both specified default to `source_path` Args: destination_path: the location on the device/simulator where the local file contents should be saved base64data: file contents, encoded as Base64, to be written to the file on the device/simulator source_path: local file path for the file to be loaded on device Returns: Union['WebDriver', 'RemoteFS']: Self instance Nz1Must either pass base64 data or a local file pathrbz source_path "z3" could not be found. Are you sure the file exists?zutf-8zmobile: pushFile)rpayload)rdata)ropenreadIOErrorbase64 b64encodedecoderrrrrr PUSH_FILE) rr"r#r$f file_dataemessagers r push_filezRemoteFS.push_file=s  :#5*+^_ _  " ?+t,) !I)  )))4;;GDJ%   ( ( 2 A A"2)   3)) ?)+6ij.w7Q> ?&   ' ' 1 9 9!!,&   s? BBB$B4B B B1B,,B14;C32C3c|jjtjdd|jjtjdd|jjtj ddy)NPOSTz+/session/$sessionId/appium/device/pull_filez-/session/$sessionId/appium/device/pull_folderz+/session/$sessionId/appium/device/push_file)command_executor add_commandrrr r/)rs r _add_commandszRemoteFS._add_commandsks_ ))'*;*;VErs ))'*=*=vGvw ))'*;*;VErs)NN)rN) __name__ __module__ __qualname__strrr!rrr4r9r:rr r sbmcmcm ooo ,#,8C=,^fgj^k,w{,\tr:r )r,typingrselenium.common.exceptionsrrtyping_extensionsr/appium.protocols.webdriver.can_execute_commandsr.appium.protocols.webdriver.can_execute_scriptsr:appium.protocols.webdriver.can_remember_extension_presencer mobilecommandr rr r?r:rrGs6W"NLc4Rt!#46RRtr: