o
    `j                     @   s6   d dl Z d dlmZ ddlmZmZ G dd dZdS )    N)AlterConfigOp   )add_resource_argumentsparse_resourcesc                   @   s,   e Zd ZdZdZedd Zedd ZdS )AlterConfigsalterzAlter Kafka Configsc              	   C   st   t | |jddtddddd |jdd	d
dd |jddddd | }|jdd
dd d |jdddd d d S )Nz-cz--configappendconfigsTzkey=value to alter)typeactiondestrequiredhelpz-vz--validate-only
store_trueF)r   defaultz--allow-unknownstore_falseraise_on_unknown)r   r   r   z--force-incrementalincrementalz--force-alter)r   add_argumentstradd_mutually_exclusive_group)clsparserr    r   [/home/djax/ivt_ai_plugin/venv/lib/python3.10/site-packages/kafka/cli/admin/configs/alter.pyadd_arguments   s   zAlterConfigs.add_argumentsc                 C   s   z6t dd |jD }d}tjtjtjtjd}|D ]}t||| }|r4|	 \}}	|| |	f||< qW n t
yE   t
d|j w t||d}
|j|
|j|j|jdS )Nc                 s   s    | ]}| d V  qdS )=N)split).0configr   r   r   	<genexpr>   s    z'AlterConfigs.command.<locals>.<genexpr>z^(set|del|add|sub)\((.*)\)$)setdeladdsubzUnable to parse configs! )r	   )validate_onlyr   r   )dictr	   r   SETDELETEAPPENDSUBTRACTrematchgroups
ValueErrorr   alter_configsr%   r   r   )r   clientargsr	   regexopskeyr,   op_strval	resourcesr   r   r   command   s0   zAlterConfigs.commandN)__name__
__module____qualname__COMMANDHELPclassmethodr   r8   r   r   r   r   r      s    
	r   )r+   kafka.adminr   commonr   r   r   r   r   r   r   <module>   s    