li7BzddlZddlZddlZddlmZddlZddlmZddlm Z ddl m Z ddl m Z mZmZmZmZmZmZmZmZmZmZmZddlmZddlmZdd lmZmZdd l m!Z!dd l"m#Z#dd l$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*dd l+m,Z,ddl-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5ddl6m7Z7m8Z8erddl9Z9ee:e;fZ@%++- >Bquery_header_contextc:t|j||_yr1)rr-r4)r5r9s r6set_query_headerz&BaseConnectionManager.set_query_headerQs24<tjtfSr1)osgetpidrr8r6get_thread_identifierz+BaseConnectionManager.get_thread_identifierTsyy{IK''r8c|j}|j5||jvrt|t |j|j|cdddS#1swYyxYwr1)r@r3r2r)listr5keys r6get_thread_connectionz+BaseConnectionManager.get_thread_connectionZs`((* YY 0$111,S$t7N7N2OPP**3/ 0 0 0s |jr|j||j|j yy)z Clear any existing transactions.N)rEtransaction_open _rollbackbegincommitr5rFs r6clear_transactionz'BaseConnectionManager.clear_transactionrsB))+  $$t$ JJL KKM r8c|j}|,|jr|jr|j|yyyyr1)rKhandlerOrPrSs r6rollback_if_openz&BaseConnectionManager.rollback_if_open{s;!!#   0E0E NN4 1F r8sqlctd)a8Create a context manager that handles exceptions caused by database interactions. :param str sql: The SQL string that the block inside the context manager is executing. :return: A context manager that handles exceptions raised by the underlying database. z8`exception_handler` is not implemented for this adapter!rr5rXs r6exception_handlerz'BaseConnectionManager.exception_handlers""\]]r8namec||dn|}|j}|r |j|k(r|jdk(r|S|tt |j |t jdd|jj}t|j|_ |j|tt||j t!|S|jdk7rt|j|_ |j|k7r-|jxsd}||_tt#|||S) zCalled by 'acquire_connection' in BaseAdapter, which is called by 'connection_named'. Creates a connection for this thread if one doesn't already exist, and will rename an existing connection.NmasteropenF)typer]staterOrV credentials) conn_name conn_type node_info)orig_conn_namerd)rKr]rbrrr,rINITr-rcrr`rVrHrr%rr$)r5r]rdrFrhs r6set_connection_namez)BaseConnectionManager.set_connection_names &*\t !!# DII*tzzV/CK < *%**!& LL44 D%TYY/DK  & &t ,  TYYR_Rab  zzV#(3 yyI%&*iio2% +>U^_` r8 connectionconnectloggerretryable_exceptions retry_limit retry_timeout _attemptsc t|r||n|}|dkr td|dks|tjkDr'd|_t j |_td ||_t j|_|S#t|$r} |dkr0d|_t j |_tt| |jd|jd|d|d| t||j||||d z |||d z cYd} ~ Sd} ~ wt$r5} d|_t j |_tt| d} ~ wwxYw) aGiven a Connection, set its handle by calling connect. The calls to connect will be retried up to retry_limit times to deal with transient connection errors. By default, one retry will be attempted if retryable_exceptions is set. :param Connection connection: An instance of a Connection that needs a handle to be set, usually when attempting to open it. :param connect: A callable that returns the appropiate connection handle for a given adapter. This callable will be retried retry_limit times if a subclass of any Exception in retryable_exceptions is raised by connect. :type connect: Callable[[], AdapterHandle] :param AdapterLogger logger: A logger to emit messages on retry attempts or errors. When handling expected errors, we call debug, and call warning on unexpected errors or when all retry attempts have been exhausted. :param retryable_exceptions: An iterable of exception classes that if raised by connect should trigger a retry. :type retryable_exceptions: Iterable[Type[Exception]] :param int retry_limit: How many times to retry the call to connect. If this limit is exceeded before a successful call, a FailedToConnectError will be raised. Must be non-negative. :param retry_timeout: Time to wait between attempts to connect. Can also take a Callable that takes the number of attempts so far, beginning at 0, and returns an int or float to be passed to time.sleep. :type retry_timeout: Union[Callable[[int], SleepTime], SleepTime] = 1 :param int _attempts: Parameter used to keep track of the number of attempts in calling the connect function across recursive calls. Passed as an argument to retry_timeout if it is a Callable. This parameter should not be set by the initial caller. :raises dbt.adapters.exceptions.FailedToConnectError: Upon exhausting all retry attempts without successfully acquiring a handle. :return: The given connection with its appropriate state and handle attributes set depending on whether we successfully acquired a handle or not. rz;retry_timeout cannot be negative or return a negative time.Nzretry_limit cannot be negativez0Got a retryable error when attempting to open a z connection. z! attempts remaining. Retrying in z seconds. Error: )rkrlrmrorprnrq)callabler(sysgetrecursionlimitrVrFAILrbOPENtuplestrdebugr,rretry_connection Exception) clsrkrlrmrnrorprqtimeoutes r6r|z&BaseConnectionManager.retry_connectionssV/7}.E- *= Q;&M  ?kC,A,A,CC $J .33J &'GH H / ' J .33J  )* a$( !#2#7#7  *3q622 LLB388*N-@ J#  'N''%'!O+%9#a-(  / $J .33J &s1v. . /s+*"B E&BD%E&% E&10E!!E&ctd)z6Cancel all open connections on the adapter. (passable)z2`cancel_open` is not implemented for this adapter!rZr5s r6 cancel_openz!BaseConnectionManager.cancel_open s""VWWr8ctd)a/Open the given connection on the adapter and return it. This may mutate the given connection (in particular, its state and its handle). This should be thread-safe, or hold the lock if necessary. The given connection should not be in either in_use or available. z+`open` is not implemented for this adapter!rZr~rks r6r`zBaseConnectionManager.opens""OPPr8c|j5|j}| dddy ddd |jy#1swYxYw#t$r|j wxYwr1)r3rKcloser}rMrSs r6releasezBaseConnectionManager.releasess YY %%'D|      JJt      ( ( *  sAAAA-c |j5|jjD]r}|jdvr)t t t |jn(t tt |j|j|t|jjdddy#1swYyxYw)N>initclosed)rd) r3r2valuesrbrr#rr]r!rclearr5rks r6 cleanup_allz!BaseConnectionManager.cleanup_all.s YY ,"55<<> ' ##+==:[Q[Q`Q`Eabc8;zC_`a :&  '  # # ) ) + , , ,s B*CC ctd)zBegin a transaction. (passable)z,`begin` is not implemented for this adapter!rZrs r6rQzBaseConnectionManager.begin:s""PQQr8ctd)z Commit a transaction. (passable)z-`commit` is not implemented for this adapter!rZrs r6rRzBaseConnectionManager.commit?s""QRRr8c  |jjy#t$rGtt t |j tjtYywxYw)z&Perform the actual rollback operation.)rdexc_inforfN) rVrollbackr}rr'rr] traceback format_excrrs r6_rollback_handlez&BaseConnectionManager._rollback_handleDsU     & & (  )*//:&113+o  sA A-,A-c*t|jdrLttt |j t |jjyttt |j t y)z#Perform the actual close operation.rrdrfN) hasattrrVrr rr]rrr"rs r6 _close_handlez#BaseConnectionManager._close_handleRsg :$$g .  ;z+GS`Sbc     # # % ")*//:mo r8c|jdurtd|jdtt t |jt |j|d|_y)zRoll back the given connection.Fz-Tried to rollback transaction on connection "z!", but it does not have one open!rN)rOrr]rr&rrrrs r6rPzBaseConnectionManager._rollbackbsi  & &% /"OO$$EG  8k*//&Bmo^_ Z(&+ #r8cp|jtjtjhvr|S|jrN|j rBt tt|jt|j|d|_|j|tj|_|S)NrF) rbrCLOSEDrirOrVrr&rr]rrrrs r6rzBaseConnectionManager.closeps    6 68L8LM M   & &:+<+< x+joo*FR_Rab c   ,&+ # *%*11 r8cJ|j}|r|jyy)z?If the named connection exists, commit the current transaction.N)rKrRrs r6commit_if_has_connectionz.BaseConnectionManager.commit_if_has_connections!'')  KKM r8cT|j|S|jj|Sr1)r4addr[s r6_add_query_commentz(BaseConnectionManager._add_query_comments*    $J  $$S))r8 auto_beginfetchlimitz agate.Tablectd)aExecute the given SQL. :param str sql: The sql to execute. :param bool auto_begin: If set, and dbt is not currently inside a transaction, automatically begin one. :param bool fetch: If set, fetch results. :param int limit: If set, limits the result set :return: A tuple of the query status and results (empty if fetch=False). :rtype: Tuple[AdapterResponse, agate.Table] z.`execute` is not implemented for this adapter!rZ)r5rXrrrs r6executezBaseConnectionManager.executes$""RSSr8ctd)a This was added here because base.impl.BaseAdapter.get_column_schema_from_query expects it to be here. That method wouldn't work unless the adapter used sql.impl.SQLAdapter, sql.connections.SQLConnectionManager or defined this method on ConnectionManager before passing it in to Adapter. See https://github.com/dbt-labs/dbt-core/issues/8396 for more information. z7`add_select_query` is not implemented for this adapter!rZr[s r6add_select_queryz&BaseConnectionManager.add_select_querys""[\\r8 type_codectd)zBGet the string representation of the data type from the type_code.z=`data_type_code_to_name` is not implemented for this adapter!rZ)r~rs r6data_type_code_to_namez,BaseConnectionManager.data_type_code_to_names""abbr8)r/Nr1)rsrsr)FFN)<__name__ __module__ __qualname____doc__NotImplementedr,rz__annotations__rrr7r rr; staticmethodr r@rrErHrrKrMrTrWabcabstractmethodr r\rj classmethodr AdapterHandlerr rr}intr SleepTimer|r rr`rrrQrRrrrPrrrboolrrrrrr?r8r6r+r+;s D#C 5C<CTXC WT#s(^WW(8(( 0z0,*,, 4x 34 1 !   ^S ^^ ^ ^' ''RFGU/U/"m+,U/ U/ 'tI7 U/  U/XseY&67BCU/U/ U/U/n XXd3i0XX Qj QZ Q Q  , RR SS *    z d   ,: ,$ , , z j   *c*c*  !# T TT T } T  - . TT&]C]E*c/,B]cuS#Xc3ccr8r+) metaclass)@rr=rutimerrmultiprocessing.contextrmultiprocessing.synchronizer threadingrtypingrrr r r r r rrrrrdbt_common.events.contextvarsrdbt_common.events.functionsrdbt_common.exceptionsrrdbt_common.utilsrdbt.adapters.base.query_headersr!dbt.adapters.contracts.connectionrrrrrrdbt.adapters.events.loggingrdbt.adapters.events.typesr r!r"r#r$r%r&r'dbt.adapters.exceptionsr(r)agaterfloatrrABCMetar+r?r8r6rs 0-    82G(B6   Q #u*   rcckkrcr8