a d*Z?d+Z@d,ZAdZBd(ZCdZDd-ZEdZFd.ZGdZHdJd0d1ZIeIeCZJeIeDZKeIeGZLeIe:ZMeIeBe)edZNdZOd+ZPd2ZQd3ZQd4ZRd5ZSd6ZTd ZUd7ZVd8ZWd9ZXd:ZYd;ZZdd?Z`d@dAZadBdCZbdDdEZcGdFdGdGZdGdHdIdIZedS)KN) defaultdictdeque)md5)IOBase TextIOBase)count)Struct)PG_PY_ENCODINGSPG_TYPESPY_TYPES make_params string_in) DatabaseErrorInterfaceErrorz1.31.4cCstd|}|j|jfS)N!)rpack unpack_from)fmtZstrucrF/var/www/html/efn/efnconsultor/selenium/selenium-4.34.2/pg8000/core.py pack_funcssriHiiZihihihZciZbhZcccczMathieu FenniakNRSKZTEDC123sntAcdGHIBPQpXcCs|tt|d|S)N)i_packlen)codedatarrr_create_messagemsr;SVCMDPpqWFLRc Cs@z |Wn.ty:}ztd|WYd}~n d}~00dSN network error)flushOSErrorr)sockerrr_flushs rNc Cs|t}z>t||krB||t|}|dkr6td||qWn.tyr}ztd|WYd}~n d}~00t|S)Nr5rI) bytearrayr8readrextendrKbytes)rLsizebuffblockrMrrr_reads  rVc CsBz||Wn.ty<}ztd|WYd}~n d}~00dSrH)writerKr)rLdrMrrr_writesrYc Cs$|dur|durtdzPttds,tdttjtj}|||||rf|tjtj dWn@tj y} z&|dur| td| WYd} ~ n d} ~ 00n|dur|}n|durBzt ||f||}WnLtj y&} z0td|d|d|d |d | WYd} ~ n d} ~ 00|rJ|tjtj dntd d} |d urzd dl } |dusv|dur| } d | _| j| _n|} |tdd|dd} | dkr| j||d}td|} n$|dur|dur| tdWntytdYn0| |fS)Nz+If unix_sock is provided, sock must be NoneAF_UNIXz9attempt to connect to unix socket on unsupported platformzcommunication errorz"Can't create a connection to host z and port z (timeout is z and source_address is z).z/one of host, sock or unix_sock must be providedFrTi/asciir<)Zserver_hostnameztls-server-end-pointzServer refuses SSLzFSSL required but ssl module not available in this python installation.)rhasattrsocketrZZ SOCK_STREAMZ settimeoutconnectZ setsockoptZ SOL_SOCKETZ SO_KEEPALIVEerrorcloseZcreate_connectionsslZcreate_default_contextZcheck_hostnameZ CERT_NONEZ verify_modeZsendallii_packZrecvdecodeZ wrap_socketscrampZmake_channel_binding ImportError) unix_sockZ orig_sockhostporttimeoutsource_address tcp_keepaliveZorig_ssl_contextrLrMchannel_bindingrc ssl_contextZresprrr _make_socketsx    "        rpc @sLeZdZddZddZdTd d Zd d Zd dZddZddZ ddZ ddZ ddZ ddZ ddZddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd+d,Zd-d.Zd/d0ZdUd2d3Zd4d5Zd6d7Zd8d9ZdVd:d;ZdWdd?Zd@dAZ dBdCZ!dDdEZ"dFdGZ#dHdIZ$dJdKZ%dLdMZ&dNdOZ'dPdQZ(dRdSZ)dS)XCoreConnectioncCs|SNrselfrrr __enter__szCoreConnection.__enter__cCs |dSrr)rb)rtexc_type exc_value tracebackrrr__exit__szCoreConnection.__exit__ localhostN8Tc*Csd|_d|_tdd|_tdd|_i|_|dur:td||| | d}| durTin| }||@}t|dkr|td| |t | D]Z\}}t |t r|d||<q|dur||=qt |ttfstd |d t|d q|d |_t |t r|d|_n||_d|_t|_i|_t||||| || |\|_|_|jjd d|_d|_tddt|_ t!t"|_#t$|j%t&|j't(|j)t*|j+t,|j-t.|j/t0|j1t2|j3t4|j5t6|j7t8|j9t:|j;t<|j=t>|j?t@|jAtB|jCtD|jEtF|jGtH|jItJ|jKtL|jMi|_Nd}ttO|}| D]&\}}|P|dtQ|tQq|RdtS|jtOt|dtS|j|tT|jzdd}tUd}|t,t0fvrtVtW|jd\}}|jN|tW|j|d|q|jXdur|jXWn2tYy }z|Z|WYd}~n d}~00d|_[dS)Nutf8)sINSERTsDELETEsUPDATEsMOVEsFETCHsCOPYsSELECTd)maxlenz.The 'user' connection parameter cannot be None)userdatabaseapplication_name replicationrzjThe parameters '{common_params}' can't appear in startup_params, they must be set using keyword arguments.zThe parameter z can't be of type .rZrwb)modecSstSrr)r rrrrXr5z)CoreConnection.__init__..ir]r6)\_client_encodingZ_commands_with_countr notificationsnoticesparameter_statusesrkeysr8updatetupleitems isinstancestrencoderRrOtyperpassword_xidset_statement_numsZ_cachesrprn_usockZmakefile_sock_backend_key_datarr pg_typesdictr py_typesNOTICE_RESPONSEhandle_NOTICE_RESPONSEAUTHENTICATION_REQUESThandle_AUTHENTICATION_REQUESTPARAMETER_STATUShandle_PARAMETER_STATUSBACKEND_KEY_DATAhandle_BACKEND_KEY_DATAREADY_FOR_QUERYhandle_READY_FOR_QUERYROW_DESCRIPTIONhandle_ROW_DESCRIPTIONERROR_RESPONSEhandle_ERROR_RESPONSEEMPTY_QUERY_RESPONSEhandle_EMPTY_QUERY_RESPONSEDATA_ROWhandle_DATA_ROWCOMMAND_COMPLETEhandle_COMMAND_COMPLETEPARSE_COMPLETEhandle_PARSE_COMPLETE BIND_COMPLETEhandle_BIND_COMPLETECLOSE_COMPLETEhandle_CLOSE_COMPLETEPORTAL_SUSPENDEDhandle_PORTAL_SUSPENDEDNO_DATAhandle_NO_DATAPARAMETER_DESCRIPTIONhandle_PARAMETER_DESCRIPTIONNOTIFICATION_RESPONSEhandle_NOTIFICATION_RESPONSE COPY_DONEhandle_COPY_DONE COPY_DATAhandle_COPY_DATACOPY_IN_RESPONSEhandle_COPY_IN_RESPONSECOPY_OUT_RESPONSEhandle_COPY_OUT_RESPONSE message_typesr7rQ NULL_BYTEappendrYrNContext ci_unpackrVra BaseExceptionrb_transaction_status)rtrrirrjrrlrhrorkrmrrstartup_paramsrLZ init_paramsZ start_paramsZ common_paramskvprotocolvalr9contextdata_lenrMrrr__init__s                 zCoreConnection.__init__cCs||j|<dSrr)r)rttypZout_funcrrrregister_out_adaptersz#CoreConnection.register_out_adaptercCs||j|<dSrr)r)rtoidZin_funcrrrregister_in_adaptersz"CoreConnection.register_in_adaptercs&fdd|tD}t||_dS)Ncs<i|]4}|dkr|ddd|ddjjddqS)r5Nr[r]replace)errors)rer.0srsrr sz8CoreConnection.handle_ERROR_RESPONSE..)splitrrra)rtr:rmsgrrsrrs z$CoreConnection.handle_ERROR_RESPONSEcCsdSrrrrtr:rrrrrsz*CoreConnection.handle_EMPTY_QUERY_RESPONSEcCsdSrrrrrrrrsz$CoreConnection.handle_CLOSE_COMPLETEcCsdSrrrrrrrrsz$CoreConnection.handle_PARSE_COMPLETEcCsdSrrrrrrrrsz#CoreConnection.handle_BIND_COMPLETEcCsdSrrrrrrrrsz&CoreConnection.handle_PORTAL_SUSPENDEDcCsdSEhttps://www.postgresql.org/docs/current/protocol-message-formats.htmlNrrrrrrsz+CoreConnection.handle_PARAMETER_DESCRIPTIONcCsdSrrrrrrrrszCoreConnection.handle_COPY_DONEcsjt|\}}jdur tdnFtjtr\|r:tdqft|jfdd}|_n jj _dS)rNz7An output stream is required for the COPY OUT response.z@The COPY OUT stream is binary, but the stream parameter is text.csj|ddS)Nr)streamrW)r:rrerrwsz2CoreConnection.handle_COPY_OUT_RESPONSE..w) bh_unpackrrrrcodecs getdecoderr stream_writerW)rtr:r is_binarynum_colsrrrrrs    z'CoreConnection.handle_COPY_OUT_RESPONSEcCs||dSrr)rrrrrrszCoreConnection.handle_COPY_DATAc s,t|\}}jdurtdtjtrtjtrX|rDtdq`fdd}|}njj}td}||}|dkrzqtj t tj t |dtj |d|t j qhnJjD]B} t| t r|rtd | j} n| } t | t j qtj ttj tt j dS) rNzxThe 'stream' parameter is required for the COPY IN response. The 'stream' parameter can be an I/O stream or an iterable.zHThe COPY IN stream is binary, but the stream parameter is a text stream.cs*||jdjt|S)Ni)clearrQrrPrrr8)bffrrrtrrris z2CoreConnection.handle_COPY_IN_RESPONSE..rii rr6zZThe COPY IN stream is binary, but the stream parameter is an iterable with str type items.)rrrrrrreadintorOrYrrr7rNrrr _send_message COPY_DONE_MSGSYNC_MSG) rtr:rrrrrrZ bytes_readrbrrrrsF           z&CoreConnection.handle_COPY_IN_RESPONSEcCsZt|d}d}|t|}|||d}||ddd}|j|||fdS)rrr6r]r[N)i_unpackfindrrerr)rtr:rZ backend_pididxZnull_idxZchannelZpayloadrrrrs   z+CoreConnection.handle_NOTIFICATION_RESPONSEcCs|jdurtdzzt|jtt|jWzPz|jWn0tjyl}ztd|WYd}~n d}~00Wd|_d|_qd|_d|_0nbzPz|jWn0tjy}ztd|WYd}~n d}~00Wd|_d|_nd|_d|_00dS)zCloses the database connection. This function is part of the `DBAPI 2.0 specification `_. Nconnection is closedrI) rrrYr TERMINATE_MSGrNrbr_ra)rtrMrrrrb"s*   " "zCoreConnection.closec Cst|d}|dkrn|dkrR|jdur2td|t|jtt|jn|dkrdt |d}|jdur|tdd t t |j|j  d | d }|t|tt|jn4|d kr`d d |ddtD}tj||j d|jd|jd|_|j d}|jj d t}|t|tt||t|jn|dkr|j|ddd|j d} |t| t|jnR|dkr|j|dddn,|dvrtd|dntd|ddS)rrNzGserver requesting password authentication, but no password was providedrr5r6zKserver requesting MD5 password authentication, but no password was providedsmd5r] cSsg|]}|dqS)r])re)rmrrr Vr5z@CoreConnection.handle_AUTHENTICATION_REQUEST..r|)rn )r6r\ zAuthentication method z not supported by pg8000.z not recognized by pg8000.)rrrrPASSWORDrrNrjoin cccc_unpackrrZ hexdigestrrrfZ ScramClientrernZauthZget_client_firstZmechanism_namer7r8Zset_server_firstZget_client_finalZset_server_final) rtr:rZ auth_codeZsaltpwdZ mechanismsZinitZmechrrrrr8sb              z,CoreConnection.handle_AUTHENTICATION_REQUESTcCs ||_dSrr)rrrrrr|sz%CoreConnection.handle_READY_FOR_QUERYcCs ||_dSrr)rrrrrrsz&CoreConnection.handle_BACKEND_KEY_DATAc Cst|d}d}g}g}t|D]r}|||t|}|t|d7}ttdt||} ||j | d<|d7}| | | |j | dq ||_ ||_ |jdurg|_dS)Nrrr[)Z table_oidZcolumn_attrnumtype_oidZ type_sizeZ type_modifierformatnamer )H_unpackrangerrr8rzip ihihih_unpackrerrrcolumns input_funcsrows) rtr:rrrrrrrZfieldrrrrs*     z%CoreConnection.handle_ROW_DESCRIPTIONrcCspt|}|||jt|tt||D]}|t|dkrJdn|q4|t |t |j t dS)Nrr) rOrQrrrH_packr8r7rPARSErYr FLUSH_MSG)rtstatement_name_bin statementoidsrrrrr send_PARSEs zCoreConnection.send_PARSEcCs |tt|t|jtdSrr)rDESCRIBE STATEMENTrYrr)rtrrrrsend_DESCRIBE_STATEMENTsz&CoreConnection.send_DESCRIBE_STATEMENTcCs|t||jtdSrr)rQUERYrrr)rtsqlrrr send_QUERYszCoreConnection.send_QUERYcCs*t|}||t|j|||Srr)rr#rNrhandle_messages)rtrrrrrexecute_simples    zCoreConnection.execute_simplec Cst||d}|t||t|jtt|j|||tt|jtzt|jWn<t y}z$|jdurt dn|WYd}~n d}~00t |j |}| t||||t|jtt|j|||S)N)rr)rrrrYrrrNr$r AttributeErrorrr r send_BIND send_EXECUTE)rtrvalsrrrrMparamsrrrexecute_unnameds*              zCoreConnection.execute_unnamedc CstD]6}d|}|dt}||jvr|j|q>q||||||t|jt zt |jWn<t y}z$|jdurt dn|WYd}~n d}~00t |}||||j|jfS)NZpg8000_statement_r]r)rrrraddrr rYrrrNr&rrr$rr)rtrrrZstatement_namerrMrrrrprepare_statements$         z CoreConnection.prepare_statementcCsFt|||d}||||t|jtt|j|||S)N)rrr)rr'r(rYrrrNr$)rtrr*rrrrrrr execute_nameds    zCoreConnection.execute_namedc Cst|}|tt|d||zt|jt|WnVtyz}z&t|dkrbt dn|WYd}~n"d}~0t yt dYn0dS)Nr6zwrite to closed filer) rOrQr7r8rYrrR ValueErrorrrr&)rtr9r:rTrMrrrrs    zCoreConnection._send_messagecCstt|tdtt|}|D]D}|dur@|tdq$||j}|tt|||q$|td|t |t |j t dS)rrNr) rOrrr8rQr7rrrBINDrYrr)rtrr*Zretvalvaluerrrrr' s   zCoreConnection.send_BINDcCst|jtt|jtdSr)rYr EXECUTE_MSGrrsrrrr(s zCoreConnection.send_EXECUTEcCsdSrrr)rtrrrrrr#szCoreConnection.handle_NO_DATAcCs|jtkr>|jdur>|jdd}|dkr>td|_|ddd}z0t|d}|j dkrp||_ n|j |7_ Wnt yYn0dS)Nr;ZROLLBACKzin failed transaction blockr ) rIN_FAILED_TRANSACTIONrarrrstripupperrint row_countr/)rtr:rr"valuesr9rrrr&s    z&CoreConnection.handle_COMMAND_COMPLETEcCsxd}g}|jD]X}t||d}|d7}|dkr6d}n&|t|||||jd}||7}||q|j|dS)Nrrr6r)encoding)rrrrrr)rtr:rrrowfuncZvlenrrrrr6s  zCoreConnection.handle_DATA_ROWcCsRd}|tkr>tt|jd\}}|j|t|j|d|q|jdurN|jdS)Nrr6)rrrVrrra)rtrr9rrrrr$Ds  zCoreConnection.handle_messagescCsT|tt|t|jtt|jtt|jtd}| ||j |dSr) rCLOSErrYrrrrNrr$rremove)rtrrrrrclose_prepared_statementOs    z'CoreConnection.close_prepared_statementcCs |jdd|tDdS)rcSs"i|]}|dd|ddqS)rr[Nrrrrrr[r5z9CoreConnection.handle_NOTICE_RESPONSE..N)rrrrrrrrrYsz%CoreConnection.handle_NOTICE_RESPONSEcCs|t}|d|d||dd|j}}||j|<|dkr`|}t|||_n|dkrt|dkr|q|n|dkr|dS)Nr]r[rZclient_encodingZinteger_datetimesonZserver_version)rrrerrlowerr get)rtr:rposkeyr1r;rrrr]s   z&CoreConnection.handle_PARAMETER_STATUS) rzNr{NNNNNTNNNN)r)rrN)N)*__name__ __module__ __qualname__ruryrrrrrrrrrrrrrrrrbrrrrrr r#r%r+r-r.rr'r(rrrr$r@rrrrrrrqsj  > D      rqc@seZdZdddZdS)rNcCsF||_|durdng|_d|_||_||_|dur6gn||_d|_dS)Nr)rrr9rrrra)rtrrrrrrrrsszContext.__init__)NNN)rFrGrHrrrrrrrsr)r5)frr_ collectionsrrhashlibriorr itertoolsrstructrrfpg8000.convertersr r r r r pg8000.exceptionsrrverrr7rrrrdZ ii_unpackZ ihihih_packrZci_packrZbh_packrZ cccc_packr  __author__rrrrrrrrrrrrrrrrrrrrrrr0rr!ZEXECUTEZFLUSHZSYNCrrZ TERMINATEr>r;rrrrr2rZPORTALZRESPONSE_SEVERITYZ RESPONSE_CODEZ RESPONSE_MSGZRESPONSE_DETAILZ RESPONSE_HINTZRESPONSE_POSITIONZRESPONSE__POSITIONZRESPONSE__QUERYZRESPONSE_WHEREZ RESPONSE_FILEZ RESPONSE_LINEZRESPONSE_ROUTINEZIDLEIN_TRANSACTIONr5rNrVrYrprqrrrrrs           Qz