o RDi@sddlmZ ddlZddlZddlmZmZddlmZm Z ddl m Z ddl m Z mZmZddlmZdd lmZmZmZdd lmZdd lmZe rSdd lmZeeZ d ZdZ e GdddeZ!GdddeZ"dS)) annotationsN)ABCabstractmethod)Enumunique)getenv) TYPE_CHECKINGAnyIterator)ER_FAILED_TO_CONNECT_TO_DB) DatabaseErrorErrorOperationalError)'SQLSTATE_CONNECTION_WAS_NOT_ESTABLISHED)TimeoutBackoffCtx)SnowflakeConnectionxc@s8eZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d S) AuthType SNOWFLAKEEXTERNALBROWSER SNOWFLAKE_JWTOAUTHID_TOKENUSERNAME_PASSWORD_MFAOKTAPROGRAMMATIC_ACCESS_TOKENNO_AUTHWORKLOAD_IDENTITYPAT_WITH_EXTERNAL_SESSIONN)__name__ __module__ __qualname__DEFAULTEXTERNAL_BROWSERKEY_PAIRrr USR_PWD_MFArPATrrr r)r)[/var/www/Datamplify/venv/lib/python3.10/site-packages/snowflake/connector/auth/by_plugin.pyr,src@seZdZdZ  d/d0d d Zed1d d Zejd2dd Zeed3ddZ eed4ddZ ed5dd Z ed6d#d$Z ed2d%d&Z ed7d(d)Zd8d+d,Zd9d-d.ZdS): AuthByPluginz!External Authenticator interface.Ntimeout int | Nonebackoff_generatorIterator | NonereturnNonec Ks>d|_t|dur |nt|dttdt|d|_d|_dS)NFmax_retry_attemptsMAX_CON_RETRY_ATTEMPTS)r,r2r.) consent_cache_id_tokenrDEFAULT_AUTH_CLASS_TIMEOUTgetintrDEFAULT_MAX_CON_RETRY_ATTEMPTS _retry_ctx_socket_timeout)selfr,r.kwargsr)r)r*__init__>s  zAuthByPlugin.__init__r7cCs|jjS)zYThe timeout of _retry_ctx is guaranteed not to be None during AuthByPlugin initialization)r9r,r;r)r)r*r,TszAuthByPlugin.timeoutcCstddS)NzdAttempting to mutate timeout of AuthByPlugin. Create a new instance with desired parameters instead.)loggerwarningr>r)r)r*r,YsrcCt)z1Return the Snowflake friendly name of auth class.NotImplementedErrorr>r)r)r*type__szAuthByPlugin.type_strcCrA)zReturn a safe version of the information used to authenticate with Snowflake. This is used for logging, useful for printing temporary tokens, but make sure to mask secrets. rBr>r)r)r*assertion_contenteszAuthByPlugin.assertion_contentconnr authenticator service_name str | Noneaccountuserpasswordr<r cKrA)zPrepare for authentication. This function is useful for situations where we need to reach out to a 3rd-party service before authenticating with Snowflake. rB)r;rGrHrIrKrLrMr<r)r)r*prepareoszAuthByPlugin.preparebodydict[Any, Any]cCrA)z.Update the body of the authentication request.rB)r;rOr)r)r* update_bodyzAuthByPlugin.update_bodycCrA)zReset secret members.rBr>r)r)r* reset_secretsrRzAuthByPlugin.reset_secretsdict[str, Any]cKrA)zRe-perform authentication. The difference between this and authentication is that secrets will be removed from memory by the time this gets called. rB)r;rGr<r)r)r*reauthenticates zAuthByPlugin.reauthenticateretc Ks@t|dtdj|jj|jj|ddt|ddt ddS)aHandles a failure when an issue happens while connecting to Snowflake. If the user returns from this function execution will continue. The argument data can be manipulated from within this function and so recovery is possible from here. Nz1Failed to connect to DB: {host}:{port}, {message}message)hostportrWcode)msgerrnosqlstate) rerrorhandler_wrapperr format_rest_host_portr7r6r)r;rGrVr<r)r)r*_handle_failures zAuthByPlugin._handle_failurecKs|ddr ~~~~~td|jjs$td|jjddtd}|td|jjdd t t |jj |j d S) a Default timeout handler. This will trigger if the authenticator hasn't implemented one. By default we retry on timeouts and use jitter to deduce the time to sleep before retrying. The sleep time ranges between 1 and 16 seconds. delete_paramsTz1Default timeout handler invoked for authenticatorz-Could not connect to Snowflake backend after rz attempt(s).Aborting)r\r]z'Hit connection timeout, attempt number z. Will retry in a bit...N) popr?debugr9 should_retryrcurrent_retry_countr timesleepfloatcurrent_sleep_time increment)r;rHrIrKrLrMr<errorr)r)r*handle_timeouts   zAuthByPlugin.handle_timeout)NN)r,r-r.r/r0r1)r0r7)r0r1)r0r)r0rE)rGrrHrErIrJrKrErLrErMrJr<r r0rJ)rOrPr0r1)rGrr<r r0rT)rGrrVrPr<r r0r1)rHrErIrJrKrErLrErMrEr<r r0r1)r!r"r#__doc__r=propertyr,setterrrDrFrNrQrSrUrdrpr)r)r)r*r+;s2        r+)# __future__rloggingrjabcrrenumrrosrtypingrr r errorcoder errorsr rrr^r time_utilrr getLoggerr!r?r8r5rr+r)r)r)r*s*