o iw@sdZddlZddlZddlZddlZddlmZm Z m Z ddl Z e e ZdZ dZdZddZd d Zd d Zd dZddZddddeddfddZdS)z Implements I/O streams over FTP.N)FTPFTP_TLS error_reply)ftpftps)zftp://username@host/path/filez&ftp://username:password@host/path/filez+ftp://username:password@host:port/path/filezftps://username@host/path/filez'ftps://username:password@host/path/filez,ftps://username:password@host:port/path/filecCs|otj|SN)urllibparseunquote)textr B/home/ubuntu/.local/lib/python3.10/site-packages/smart_open/ftp.py_unquote$srcCsNtj|}|jtvs Jt|jt|jt|j|j t |j pt t|j dS)N)schemeuri_pathuserhostportpassword)r r urlsplitrSCHEMESdictrpathusernamehostnameintr DEFAULT_PORTr) uri_as_string split_urir r r parse_uri(s  r cCsRtjt|t|}|d}|d}|dkrdnd}t||f||d|S)NrrrTF)secure_connectiontransport_params) smart_openutils check_kwargsopenr pop)urimoder" parsed_urirr secure_connr r ropen_uri5s  r,csDgdfdd|D}fdd|D}|r td||S)N)timeoutsource_addressencodingcsg|]}|vr|qSr r ).0ksupported_keywordsr r Jsz4convert_transport_params_to_args..csi|] \}}|vr||qSr r )r0r1vr2r r Ksz4convert_transport_params_to_args..z.ignoring unsupported ftp keyword arguments: %r)itemsloggerwarning)r"unsupported_keywordskwargsr r2r convert_transport_params_to_argsDsr<c Cst|}|rtjtjjd}tdd|i|}ntdi|}z|||Wnty:} zt d| d} ~ wwz| ||Wnt yV} zt d| d} ~ ww|r]| |S)N)purposecontextz@Unable to connect to FTP server: try checking the host and port!zFUnable to login to FTP server: try checking the username and password!r )r<sslcreate_default_contextPurpose SERVER_AUTHrrconnect Exceptionr8errorloginrprot_p) rrrrr!r"r; ssl_contextrer r r_connectUs0 rJrbFcCs|std|s td|si}t||||||}dddd} z| |\} } Wnty5td|w| |\} } |d|| d |} | | } d d }| j| _| | _|| _ t || | _| S) aiOpen a file for reading or writing via FTP/FTPS. Parameters ---------- path: str The path on the remote server mode: str Must be "rb" or "wb" host: str The host to connect to user: str The username to use for the connection password: str The password for the specified username port: int The port to connect to secure_connection: bool True for FTPS, False for FTP transport_params: dict Additional parameters for the FTP connection. Currently supported parameters: timeout, source_address, encoding. z'you must specify the host to connect tozyou must specify the user)RETRrK)STORwb)APPErN)rKrNabzunsupported mode: zTYPE I cSs ||j|jdSr) orig_closesocketcloseconn)selfr r r full_closes zopen..full_close) ValueErrorrJKeyErrorvoidcmd transfercmdmakefilerTrRrSrUtypes MethodType)rr)rrrrr!r"rUmode_to_ftp_cmdsftp_mode file_obj_moderSfobjrWr r rr&ms4     r&)__doc__loggingr? urllib.parser smart_open.utilsr#ftplibrrrr] getLogger__name__r8rr URI_EXAMPLESrr r,r<rJr&r r r rs2