g<>dZddlZddlZddlmZddlmZddlmZ dZ ej dZ ej d Z ej d Zej d Zej d Zej d Zej dZGddeZdS)z( Protocol version 3.0 client and tools. N)pq_message_stream)element3)xact3) Connection)SFATALC08P01)Mz2wire-data caused exception in protocol transaction)HzProtocol error detected.)r)r z--TOE)rzconnect timed out)rr )rz&server did not support SSL negotiation)rz&The server is probably not PostgreSQL.)rr z--SEC)rz6SSL was required, and the server could not accommodate)rr)rz SSL negotiation caused exception)r)r z--XXX)rzfailed to complete negotiation)rzONegotiation failed to completed, but no error was attributed on the connection.)rr 08006)rzunexpected EOF from server)Dz.Zero-length read from the connection's socket.ceZdZdZdZdZedieZdZddZddZ de dffdZ e j fd Zd ZeZd Zejfd ZeZd ZdZdZdZdZdZdZdZdZdZdZ ddZ!dS)rz A PQv3 connection. Operations are designed to not raise exceptions. The user of the connection must check for failures. This is done to encourage users to use their own Exception hierarchy. Nc4d}d}d}d}tS)Nc|jSN)_tracerselfs Q/granary/edev/quantum/lib/python3.11/site-packages/postgresql/protocol/client3.pyfgetzConnection.tracer..fgetNs ,cD||_|j|_|j|_dSr)rtraced_write_messageswrite_messagestraced_read_messages read_messages)rvalues rfsetzConnection.tracer..fsetPs%4<3414rc:|`|j|_|j|_dSr)rstandard_write_messagesr standard_read_messagesr"rs rfdelzConnection.tracer..fdelTs# |5434rzMCallable object to pass protocol trace strings to. (Normally a write method.))locals)rr$r(docs rtracerzConnection.tracerMs?222444 # /rc|j|tjtjf}||_|dS)z Explicitly send a Synchronize message to the backend. Useful for forcing the completion of lazily processed transactions. NOTE: This will not cause trash to be taken out. N)xactcomplete InstructionelementSynchronizeMessagerxs r synchronizezConnection.synchronize]sG Y==??? 2455!$)--/////rctj|j|j}||} |||dS#|wxYw)Ntimeout)r0 CancelRequest backend_idkeybytessocket_factorysendallclose)rr7cqss r interruptzConnection.interruptjsqT_dh77==??" G,,! 99R===77999991779999s A44B cZt|drdSd|_ |||_n#|jj|jjf$r}t j|j_d|j_||j_ |j |rt|j_ n?|j |}tjddd|pdff|j_ Yd}~dSd}~wwxYw| |x|_}|5d|j_t$|j_ t j|j_dS|s9|dur5d|j_t&|j_ t j|j_dS|r} |j|j|_nW#t*$rJ}||j_ d|j_t j|j_t,|j_ Yd}~dSd}~wwxYw|j}||jt jur+|j$|jj|_|jj|_dSt|jd s5d|j_t j|j_t8|j_ dSdS) a Establish the connection to the server. If `ssl` is None, the socket will not be secured. If `ssl` is True, the socket will be secured, but it will close the connection and return if SSL is not available. If `ssl` is False, the socket will attempt to be secured, but will continue even in the event of a server that does not support SSL. `timeout` will be passed directly to the configured `socket_factory`. socketNr6Tr)r 08004rzcould not connect error_message)hasattrrCr<timeout_exceptionfatal_exceptionr-Completestatefatal exception timed_outclient_connect_timeoutrEfatal_exception_messager0 ClientError negotiate_sslssl_negotiation not_pq_error no_ssl_errorsecure Exceptionssl_failed_errorr.killinfor:pidr9partial_connection_error)rsslr7errerrmsg supportednegxacts rconnectzConnection.connectrs T8  6$+ $$w$774;;(&    ]49?49?49 ##C(( 4DI  8 8 = =F&1 F ))* 3DI  66666) , _'+&8&8&:&::4)DIO*DImDIO F  t DIO*DImDIO F   &--dk::T[[    TYTY_}TY_/TY VVVVV   I'--/// ]dm## (="48%)4??? 49o . .649?]49?549 66s.7C6BC11C6 $F11 H;?HHreturnctj}|r$||j|d}|$|jd}|dkrdS|dkrdSdS)z Negotiate SSL If SSL is available--received b'S'--return True. If SSL is unavailable--received b'N'--return False. Otherwise, return None. Indicates non-PQv3 endpoint. Nrr TNF)r0NegotiateSSLMessager;rCsendrecv)rrstatuss rrQzConnection.negotiate_ssls!''))!   !    1  ;  A  & t^^ $ ~~ % rcF|jj}|jj}|jj}|j}|j}|s|j||jd|_) |||_nm#|jj $r[}|j |}|9||_ d|_ ||_ tjddd|ff|_Yd}~dSd}~wwxYw|jdkr||_ d|_ t"|_dS||j|_|dS)zC read data from the wire and write it into the message buffer. NTrrrFr)message_buffer has_messagewriterCrfrecvsizer- read_datar<rHrOrJrKrLr0rPrE eof_error) rrIBUFFER_HAS_MSGBUFFER_WRITE_MSG RECV_DATA RECV_BYTESXACTemsgs r read_intozConnection.read_intosm&2.(.k)}* $ N  #5 n T^$$$DN  Yz**DNN   -     5 5a 8 8C TZTZT^!- Sk/T EEEEE  & nDJDJ"D 5%$T^444>G N  #5H sA..C=ACCCc|d}|js2|}|j|_|S)z9read more messages into self.read when self.read is emptyT)readrwrj)rrgs rr'z!Connection.standard_read_messagess= ! * ~~1"''))49 (rc|jj} |jr)|j||jd|_|j)n#|jj$ry}|j|}|Wt j|j_d|j_ ||j_ tj ddd|ff|j_ Yd}~dSd}~wwxYwdS)zk send all `message_data`. If an exception occurs, it will check if the exception is fatal or not. NTrrrF)rCre message_datar<rHrOr-rIrJrKrLr0rPrE)r SEND_DATArurvs rsend_message_datazConnection.send_message_datask) )Yt !!""D     ,    4 4Q 7 73 omDIODIODI%1 C[3DI 55555  s0?CA-CCCc|j|jur*|xj||jz c_|j|_||jur1||_|xj||jz c_|j|_|S)zprotocol message writer)writingwrittenr{r})rmessages cat_messagess rr&z"Connection.standard_write_messages:s \%% T\222,4< T\!!4< T\222,4<   ! !!rc |D]}t|dd}|z|}|dt |ddt |t |t j|dt ||t jfz||S)z message_writer used when tracingtypeNu↑ {type}({lend}): {data}{nl})rlenddatanlu↑__(%d): %r%s) getattrr;rformatreprlenoslinesepr&)rrrvtdata_outs rrz Connection.traced_write_messagesIs csFD!!1myy{{HLL188 GGAbDM MM NN * 9 LL"XXsBJ%  % %h / //rc |}|jD]G}|d|dt|d|dtjfzH|S)z message_reader used when tracingu↓ %r(%d): %r%srr)r'ryrrrr)rrgrvs rr!zConnection.traced_read_messages\sp !!##! Yc<<"FCAKKQ%-- (rcg}d}|jD].}|tj||dz }/d}|jD].}|tj||dz }/|tjtj|}||_|jd|=|jd|=| dS)z6 close cursors and statements slated for closure. rrN) garbage_cursorsappendr0 ClosePortalgarbage_statementsCloseStatementr1r-r/r.)rxmcursorsr3 statementss rtake_out_trashzConnection.take_out_trashes " '  a99W  # #$$$ a<77*  "a99W #A & &''' ?::))G &''' r!$) 8G8$ kzk*--/////rc|jtjurdS|j||jF|js|jr||j||_|dSdSdS)z2 setup the given transaction to be processed. N)rJr-rIr.rrrstepr2s rpushzConnection.pushys W  6 Y==??? Y !5  iDIIIKKKKK rcl|j} |j\}}|tjur%||j|n|tjurR||j||jd|_t|d|j|_ntdt|zn8#|j j $r&}|j |rYd}~dSd}~wwxYw|jtjur!t|jdddur d|_dSdSdS)z Make a single transition on the transaction. This should be used during COPY TO STDOUT or large result sets to stream information out. N last_readyz!unexpected PQ transaction state: rKT)r-rJSendingr r Receivingr"ryr RuntimeErrorrr<try_again_exception try_againrI)rr3diroprus rrzConnection.stepsU i! W73 T\ ###BDDDD t~ ""TY--..)DIL$*55DJJ (4994     0    ##A&&  FFFFF  W  $)Wd##4//4999//sBz,Connection.register_cursor..suu\?R?Rr) trash_cursorweakrefrr)rcursorrrs `@rregister_cursorzConnection.register_cursors7  %&{63R3R3R3R3RSS$,|rcn |j|=n#t$rYnwxYw|j|dSr)rKeyErrorrr)rrs rrzConnection.trash_cursorsQ |L!! 4l+++++  cZ|jtj|fd|j<dS)NcSrr)rpq_statement_idrs rrz/Connection.register_statement..soH^H^r)trash_statementrrr)r statementrrs `@rregister_statementzConnection.register_statements7  %%,[<^<^<^<^<^%_%_$//"""rcn |j|=n#t$rYnwxYw|j|dSr)rrrr)rrs rrzConnection.trash_statementsQ '' 4  11111rcTt|dr|jdurd}n|jdurd}nd}d|t |j|j}t |jdz|zd ztj z| zS) NrRTSSLFzNOSSL after SSLNOSSLz -> ()) rFrRjoinexception_stringrrLstrr<rrstrip)rr[excstrs r__str__zConnection.__str__s T$%% d"" CC %% C 3 774((dn)=)=t~NN O O& T ! !    Z ,,.. !!rrcHd|_d|_||_tjt j|||_i|_i|_g|_ g|_ t|_ d|_ d|_d|_d|_d|_d|_d|_dS)z] Create a connection. This does not establish the connection, it only initializes it. Ni rr INITIALIZED)r:r9r<r- Negotiationr0Startuprrrrrrjrmryrnr{rrrJ)rr<startuppasswords r__init__zConnection.__init__s $($/&$ ?7X$)$,$/$$)++$$-$)$.$$,$,$***rrr)NN)r)"__name__ __module__ __qualname____doc__rr+propertyr4rAr`boolrQr-rIrwr'r"r}r0rr&r rr!rrrr.rrrrrrrrrrrDs     (  VVXX         Y6Y6Y6Y6vT4L&!% ----^   (B% " " " "*000&   (*###J(((TTTT,,,```222 ! ! !######rr)rrrbufferrrrr0rr-__all__rPrrNrSrTrWrZroobjectrrrrrs[ %%%%%%!!!!!! !4!46""-,. #w"$ #w"$ '7&(/7.0 G !   ``````````r