o i$@s"UdZddlZddlZddlZddlZddlmZmZm Z zddl Z Wn e y.dZ Ynwddl ZeeZiZdZ dZdZejdgZd d Zd d Zd efdefdefdefdefdefdZeee eeffed<ddZddZ ddZ!d!ddZ"        d"dd Z#dS)#aqImplements I/O streams over SSH. Examples -------- >>> with open('/proc/version_signature', host='1.2.3.4') as conn: ... print(conn.read()) b'Ubuntu 4.4.0-1061.70-aws 4.4.131' Similarly, from a command line:: $ python -c "from smart_open import ssh;print(ssh.open('/proc/version_signature', host='1.2.3.4').read())" b'Ubuntu 4.4.0-1061.70-aws 4.4.131' N)DictCallableTupleT)sshscpsftp)zssh://username@host/path/filezssh://username@host//path/filezscp://username@host/path/filezsftp://username@host/path/filez ~/.ssh/configcCs|otj|SN)urllibparseunquote)textrB/home/ubuntu/.local/lib/python3.10/site-packages/smart_open/ssh.py_unquoteDsrcCs(|dkrdS|dkr dStd|d)NnoFyesTzExpected 'yes' / 'no', got .) ValueError)stringrrr _str2boolHs rconnecttimeout compressiongssapiauthenticationgssapikeyexchangegssapidelegatecredentialsgssapitrustdns)timeoutcompressgss_authgss_kexgss_deleg_creds gss_trust_dns_PARAMIKO_CONFIG_MAPcCsTtj|}|jtvs Jt|jt|jt|j|j |j r"t |j ndt|j dS)N)schemeuri_pathuserhostportpassword) r r urlsplitr$SCHEMESdictrpathusernamehostnamer(intr)) uri_as_string split_urirrr parse_urids r3cCsHtjt|}t|}|d}|di||}t||fi|S)Nr%r$) smart_openutils check_kwargsopenr3pop)urimodetransport_paramskwargs parsed_urir% final_paramsrrropen_uriqs    r?cCsbt}||t|pi}d|vr|d||d||j||fi||S)N key_filenamer)r.)paramiko SSHClientload_system_host_keysset_missing_host_key_policy AutoAddPolicycopy setdefaultconnect)r/r.r(r)connect_kwargsrr<rrr _connect_sshzs   rJc Cs~tdd|||||fDs|||||fS|std|pi}ddtD}d}|D]{}ztj|} Wn ty?Yq,w|| vrGq,| |} |durV| dd}|s\| d}|durtzt | d }Wn t tfysYnw| d dur| d g} t | r| |d <tD]\} \} } | | dur| | vr| | | || <qq,|durt}|st}|r|}|||||fS) Ncss|]}|duVqdSr r).0argrrr sz&_maybe_fetch_config..z'you must specify the host to connect tocSsg|] }tj|r|qSr)osr-exists)rKfrrr sz'_maybe_fetch_config..r&r/r(r@ identityfile)anyrrF_SSH_CONFIG_FILESrA SSHConfig from_pathPermissionError get_hostnameslookupgetr0KeyErrorlenr#items DEFAULT_PORTgetpassgetuser)r'r.r)r(rIconnect_params config_filesactual_hostnameconfig_filenamecfgrS param_name sshcfg_namefrom_strrrr_maybe_fetch_configsT       rjrc  Cs0t|||||\}}}}}||f} d} t| D]g} zt| } | js3| t|||||} t| <WntyIt|||||} t| <Ynwz | } | }Wn)t j y~}z|j og|j ddk}| | dksp|sqt| =WYd}~qd}~ww|j |||d}||_ |dur|jdi||S)aOpen a file on a remote machine over SSH. Expects authentication to be already set up via existing keys on the local machine. Parameters ---------- path: str The path to the file to open on the remote machine. mode: str, optional The mode to use for opening the file. host: str, optional The hostname of the remote machine. May not be None. user: str, optional The username to use to login to the remote machine. If None, defaults to the name of the current user. password: str, optional The password to use to login to the remote machine. port: int, optional The port to connect to. connect_kwargs: dict, optional Any additional settings to be passed to paramiko.SSHClient.connect. prefetch_kwargs: dict, optional Any additional settings to be passed to paramiko.SFTPFile.prefetch. The presence of this dict (even if empty) triggers prefetching. buffer_size: int, optional Passed to the bufsize argument of paramiko.SFTPClient.open. Returns ------- A file-like object. Important --------- If you specify a previously unseen host, then its host key will be added to the local ~/.ssh/known_hosts *automatically*. If ``username`` or ``password`` are specified in *both* the uri and ``transport_params``, ``transport_params`` will take precedence rzSSH session not activeN)r:bufsizer)rjrange_SSH get_transportactivecloserJr\open_sftp_clientrA SSHExceptionargsr7nameprefetch)r-r:r'r&r)r(rIprefetch_kwargs buffer_sizekeyattemptsattemptr transport sftp_clientexconnection_timed_outfobjrrrr7s>2     r7)NNNN)rkNNNNNNrl)$__doc__r`rNlogging urllib.parser typingrrrrA ImportError MISSING_DEPSsmart_open.utilsr4 getLogger__name__loggerrqr+r_ URI_EXAMPLESr- expanduserrUrrfloatr#str__annotations__r3r?rJrjr7rrrrsR       Y