暾j%, UdZddlmZddlZddlZddlZddlZddlm Z ddl m Z m Z m Z  ddlZddlZe r ddlmZddlmZej.eZiZd ed <d Z d Zd Zee djAgZ!d!dZ"d"dZ#de$fde#fde#fde#fde#fde#fdZ%ded<Gdde Z&d#dZ'd$dZ( d%dZ) d& d'dZ* d( d)d Z+y#e$rdZYwxYw)*aImplements I/O streams over SSH; GSSAPI auth (``gss_*`` options) needs ``paramiko[gssapi]``. Example: >>> 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' ) annotationsN)Path) TYPE_CHECKINGAny TypedDictT)Callable)TransportParamsz)dict[tuple[str, str], paramiko.SSHClient]_SSH)sshscpsftp)zssh://username@host/path/filezssh://username@host//path/filezscp://username@host/path/filezsftp://username@host/path/filez ~/.ssh/configcH|xrtjj|SN)urllibparseunquote)texts X/home/spazio/kingdom_calls/work5/transenv/lib/python3.12/site-packages/smart_open/ssh.py_unquoterCs  .FLL((..c<|dk(ry|dk(ryd|d}t|)NnoFyesTzExpected 'yes' / 'no', got .) ValueError)stringmsgs r _str2boolrGs. ~  'xq 1C S/rconnecttimeout compressiongssapiauthenticationgssapikeyexchangegssapidelegatecredentialsgssapitrustdns)timeoutcompressgss_authgss_kexgss_deleg_creds gss_trust_dnszdict[str, tuple[str, Callable]]_PARAMIKO_CONFIG_MAPcJeZdZUded<ded<ded<ded<ded<ded <y ) _SSHUristrscheme str | Noneuri_pathuserhost int | NoneportpasswordN)__name__ __module__ __qualname____annotations__rrr.r.ds% K   rr.cVtjj|}|jtvsJ|jt |j t |j|j|jrt|jndt |jdS)zJParse an ``ssh://``/``scp://``/``sftp://`` URI into connection components.N)r0r2r3r4r6r7) rrurlsplitr0SCHEMESrpathusernamehostnamer6intr7) uri_as_string split_uris r parse_urirFms %%m4I   w && &""Y^^,++,""'0~~INN#4Y//0  rctjjt|}t t |}|j d}|j di||}t||fi|S)zCOpen an SSH/SCP/SFTP URI using the given mode and transport params.r2r0) smart_openutils check_kwargsopendictrFpop)urimodetransport_paramskwargs parsed_urir2 final_paramss ropen_urirT{sd    * *41A BF!%in!5J~~j)HNN8+j+F+L $ /, //rcDtj}|j|jtj|xsij }d|vr|j d||j d||j|||ntfi||S)N key_filenamer7rA) paramiko SSHClientload_system_host_keysset_missing_host_key_policy AutoAddPolicycopy setdefaultconnect DEFAULT_PORT)rBrAr6r7connect_kwargsr rQs r _connect_sshras    C##H$:$:$<=" ( ( *FV#*h/ j(+CKK$"2$ OO Jrctd|||||fDs|||||fS|s d}t||xsij}tDcgc]}t |j s| }}d} |D]} t jj| } || jvr7| j|} || jdd}| s| d} |5tjtt5t!| d}ddd|jd"| jdg} t#| r| |d<t$j'D]-\} \}}|j| || vs || ||| </|t(}|st+j,}| r| }|||||fScc}w#t$rYNwxYw#1swYxYw) Nc3$K|]}|du ywrr<).0args r z&_maybe_fetch_config..sWssd{Wsz'you must specify the host to connect tor3rBr6rV identityfile)anyrr\_SSH_CONFIG_FILESrexistsrW SSHConfig from_pathPermissionError get_hostnameslookupget contextlibsuppressKeyErrorrClenr,itemsr_getpassgetuser)r4rAr7r6r`rconnect_paramsf config_filesactual_hostnameconfig_filenamecfgrh param_name sshcfg_namefrom_strs r_maybe_fetch_configrs WtXx~&VW WXx~== 7o"%*002N0E!DGNN4DAELE O'!H $$..?C s((* * jj  wwvt,H!*oO <$$Xz: (3v;' (   n - 577>26L< 1=~.3G3M3M3O H /J/h!!*-5+:L-5c+6F-Gz* H?!HP | ??$ 8T> 99qF    ( (s* F0)F08F56G5 GGG c t|||||\}}}}}|J|J|J||f} d} d} t| D]z} t| } | j}|J|js(| j t |||||x} t| < | j}|J|j} n| J| j|||}||_ ||jdi||S#t$rt |||||x} t| <YwxYw#tj$r=}|jxr|jddk(}| | dz k(s|st| =Yd}~$d}~wwxYw)awOpen a file on a remote machine over SSH. Expects authentication to be already set up via existing keys on the local machine. Args: path: The path to the file to open on the remote machine. mode: The mode to use for opening the file. host: The hostname of the remote machine. May not be None. user: The username to use to login to the remote machine. If None, defaults to the name of the current user. password: The password to use to login to the remote machine. port: The port to connect to. connect_kwargs: Any additional settings to be passed to paramiko.SSHClient.connect. prefetch_kwargs: Any additional settings to be passed to paramiko.SFTPFile.prefetch. The presence of this dict (even if empty) triggers prefetching. buffer_size: Passed to the bufsize argument of paramiko.SFTPClient.open. Returns: A file-like object. Raises: paramiko.SSHException: If the SSH session is no longer active and cannot be re-established within the retry limit. Note: 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. NrzSSH session not active)rObufsizer<)rranger get_transportactivecloserartopen_sftp_clientrW SSHExceptionargsrKnameprefetch)r@rOr4r3r7r6r`prefetch_kwargs buffer_sizekey sftp_clientattemptsattemptr transportexconnection_timed_outfobjs rrKrKsT2E dHdN2.D$$          ,C.2KH? Ws)C))+I( ((## ".tT4>"ZZd3i ))+I( ((#446K #:  "" "   Dt[  ADDI" (( K1 W*4tX~V VC$s) W$$ #%77#Urwwqz=U/U (Q,&.B S  s*AC' $D'!D  D E!2EE)rr1returnr1)rr/rbool)rDr/rr.)rNr/rOr/rPr rparamiko.SFTPFile) rBr/rAr1r6r5r7r1r`dict[str, Any] | Nonerzparamiko.SSHClient)NNNN) r4r1rAr1r7r1r6r5r`rrzLtuple[str | None, str | None, str | None, int | None, dict[str, Any] | None])rNNNNNN)r@r1rOr/r4r1r3r1r7r1r6r5r`rrrrrCrr),__doc__ __future__rrrrwlogging urllib.parserpathlibrtypingrrrrW ImportError MISSING_DEPSsmart_open.utilsrHcollections.abcrsmart_open._typingr getLoggerr8loggerr r;r?r_ URI_EXAMPLESr/ expanduserrjrrfloatr,r.rFrTrarrKr<rrrs #00(2   8 $ 35/4   o.99;<=/(!%( *'3#Y/3Y?& 2 95i 0   *   ( ,0 W: W:W:W:  W: * W: R W:x,0-1W W W W  W  W  W*W+WWWcLsC88DD