o `j@sdZddlmZddlmZddlZddlZddlZddlm Z ddl m Z ddl m Z ddlmZmZe reZdZUdZded<ddZddZd d d Zd d d ZdS)UserAdminMixinz=Mixin providing user management methods for KafkaAdminClient.r_managerc sg}g}|D]2}t|tr||jt|jfqt|tr1||jt|j|j|j|j fqt dt |j t ||d}|j|IdH}i}|jD] }|jrV|jnd||j<qN|S)NzWalterations must be UserScramCredentialDeletion or UserScramCredentialUpsertion, got %s) deletions upsertions) isinstanceUserScramCredentialDeletionappenduserint mechanismUserScramCredentialUpsertion iterationssaltsalted_passwordrtype__name__rr sendresults error_code error_message) self alterationsr r altrequestresponseretresultr$P/home/djax/ivt_ai_plugin/venv/lib/python3.10/site-packages/kafka/admin/_users.py#_async_alter_user_scram_credentials s8   z2UserAdminMixin._async_alter_user_scram_credentialscC|j|j|S)aoAlter SCRAM credentials for one or more users. Arguments: alterations: A list of UserScramCredentialDeletion and/or UserScramCredentialUpsertion objects describing the credentials to delete and/or insert/update. Returns: A dict mapping user name -> error message (or None on success). )r runr&)rrr$r$r%alter_user_scram_credentials>s z+UserAdminMixin.alter_user_scram_credentialsNcs|durd}ndd|D}t|d}|j|IdH}t|j}|tjur0|d|jfi}|jD]}|jrD|jgd||j <q5ddd|j Dd||j <q5|S)NcSsg|]}|fqSr$r$).0rr$r$r% OszIUserAdminMixin._async_describe_user_scram_credentials..)usersz.DescribeUserScramCredentialsRequest failed: %s)errorcredential_infoscSsg|] }t|j|jdqS))rr)ScramMechanismrr)r*cir$r$r%r+cs ) rr rErrorsfor_coderNoErrorrrrr.)rr, users_fieldr r! error_typer"r#r$r$r%&_async_describe_user_scram_credentialsKs2     z5UserAdminMixin._async_describe_user_scram_credentialscCr')aDescribe SCRAM credentials for one or more users. Arguments: users (list of str, optional): User names to describe. If None, describe all users with SCRAM credentials. Returns: A dict mapping user name to a dict with keys ``'error'`` (None or error message) and ``'credential_infos'`` (list of {'mechanism': ScramMechanism, 'iterations': int}). )r r(r6)rr,r$r$r%describe_user_scram_credentialsms z.UserAdminMixin.describe_user_scram_credentials)N) r __module__ __qualname____doc____annotations__r&r)r6r7r$r$r$r%r s  "r c@s$eZdZdZdZdZeddZdS)r/rcCstjdtjdi|S)Nsha256sha512)r/ SCRAM_SHA_256 SCRAM_SHA_512rr$r$r% hash_names zScramMechanism.hash_nameN)rr8r9UNKNOWNr@rApropertyrCr$r$r$r%r/|s r/c@s eZdZdZddZddZdS)rzSpecifies that a SCRAM credential should be deleted. Arguments: user (str): The user name. mechanism (ScramMechanism or int or str): The SCRAM mechanism to delete for this user. cCsBt|tst|trt|dd}nt|}||_||_dS)N-_)r r/strupperreplacerr)rrrr$r$r%__init__s   z$UserScramCredentialDeletion.__init__cCsd|jd|jjdS)NzUserScramCredentialDeletion(, ))rrnamerBr$r$r%__repr__sz$UserScramCredentialDeletion.__repr__N)rr8r9r:rKrOr$r$r$r%rs rc@s&eZdZdZdZdddZddZdS) raHSpecifies that a SCRAM credential should be inserted or updated. Arguments: user (str): The user name. mechanism (ScramMechanism or int or str): The SCRAM mechanism. password (bytes or str): The plaintext password. The salted password sent to the broker is derived via PBKDF2-HMAC using the given salt and iteration count. Keyword Arguments: iterations (int, optional): PBKDF2 iteration count. Default: 4096. salt (bytes, optional): Salt to use. If omitted, a random 24-byte salt is generated. iNcCst|tst|trt|dd}nt|}|tjkr"td||_||_|dur.|n|j |_ |dur8|nt d|_ t|trH|d}t|j||j |j |_dS)NrFrGz#SCRAM mechanism must not be UNKNOWNzutf-8)r r/rHrIrJrDrrrDEFAULT_ITERATIONSrosurandomrencodehashlib pbkdf2_hmacrCr)rrrpasswordrrr$r$r%rKs      z%UserScramCredentialUpsertion.__init__cCs d|jd|jjd|jdS)NzUserScramCredentialUpsertion(rLz , iterations=rM)rrrNrrBr$r$r%rOs z%UserScramCredentialUpsertion.__repr__)NN)rr8r9r:rQrKrOr$r$r$r%rs   r)r: __future__renumrrUloggingrRtypingr kafka.errorserrorsr1rkafka.protocol.adminrrkafka.net.managerr getLoggerrlogr r/rrr$r$r$r%s"       `