o `šjQ ã@s^ddlmZmZddlZddlmZddlmZe e¡Z Gdd„deƒZ Gdd „d eƒZ dS) é)ÚABCÚabstractmethodNé)Ú SaslMechanism)ÚKafkaConfigurationErrorc@sDeZdZdd„Zdd„Zdd„Zdd„Zd d „Zd d „Zd d„Z dS)ÚSaslMechanismOAuthcKsFd|vrtdƒ‚t|dtƒstdƒ‚|d|_d|_d|_d|_dS)NÚsasl_oauth_token_providerz7sasl_oauth_token_provider required for OAUTHBEARER saslzSsasl_oauth_token_provider must implement kafka.net.sasl.oauth.AbstractTokenProviderF)rÚ isinstanceÚAbstractTokenProviderÚtoken_providerÚ_errorÚ_is_doneÚ_is_authenticated©ÚselfÚconfig©rúR/home/djax/ivt_ai_plugin/venv/lib/python3.10/site-packages/kafka/net/sasl/oauth.pyÚ__init__ s  zSaslMechanismOAuth.__init__cCs0|jr|jS|j ¡}| ¡}d ||¡ d¡S)Nzn,,auth=Bearer {}{}úutf-8)r r ÚtokenÚ_token_extensionsÚformatÚencode)rrÚ extensionsrrrÚ auth_bytess  zSaslMechanismOAuth.auth_bytescCs8|dkr| d¡}t d|¡d|_dSd|_d|_dS)NórzCSending x01 response to server after receiving SASL OAuth error: %sóT)ÚdecodeÚlogÚdebugr r r)rrÚerrorrrrÚreceives     zSaslMechanismOAuth.receivecCó|jS©N)r ©rrrrÚis_done(ózSaslMechanismOAuth.is_donecCr#r$)rr%rrrÚis_authenticated+r'z#SaslMechanismOAuth.is_authenticatedcCs2|j ¡}d dd„| ¡Dƒ¡}|rd|SdS)zŽ Return a string representation of the OPTIONAL key-value pairs that can be sent with an OAUTHBEARER initial request. úcSsg|] \}}d ||¡‘qS)z{}={})r)Ú.0ÚkÚvrrrÚ 5sz8SaslMechanismOAuth._token_extensions..Ú)r rÚjoinÚitems)rrÚmsgrrrr.s z$SaslMechanismOAuth._token_extensionscCs|jstdƒ‚dS)NzNot authenticated yet!zAuthenticated via SASL / OAuth)r(Ú RuntimeErrorr%rrrÚ auth_details8szSaslMechanismOAuth.auth_detailsN) Ú__name__Ú __module__Ú __qualname__rrr"r&r(rr3rrrrr s   rc@s,eZdZdZdd„Zedd„ƒZdd„ZdS) r a A Token Provider must be used for the SASL OAuthBearer protocol. The implementation should ensure token reuse so that multiple calls at connect time do not create multiple tokens. The implementation should also periodically refresh the token in order to guarantee that each call returns an unexpired token. A timeout error should be returned after a short period of inactivity so that the broker can log debugging info and retry. Token Providers MUST implement the token() method cKsdSr$rrrrrrLszAbstractTokenProvider.__init__cCsdS)zY Returns a (str) ID/Access Token to be sent to the Kafka client. Nrr%rrrrOszAbstractTokenProvider.tokencCsiS)aa This is an OPTIONAL method that may be implemented. Returns a map of key-value pairs that can be sent with the SASL/OAUTHBEARER initial client request. If not implemented, the values are ignored. This feature is only available in Kafka >= 2.1.0. All returned keys and values should be type str rr%rrrrWs z AbstractTokenProvider.extensionsN)r4r5r6Ú__doc__rrrrrrrrr >s    r ) ÚabcrrÚloggingrÚ kafka.errorsrÚ getLoggerr4rrr rrrrÚs   3