o
    `j                     @   s   d dl mZ G dd dZdS )    )TopicPartitionc                   @   s,   e Zd ZdZdZedd Zedd ZdS )AlterPartitionReassignmentszalter-reassignmentsz(Alter replica assignments for partitionsc              
   C   s0   |j ddtddg ddd |j dtd d	d
 d S )Nz-rz
--reassignappendreassignmentsTzTOPIC:PARTITION=BROKER_ID[,BROKER_ID...] to set a new replica set, or TOPIC:PARTITION=cancel to cancel an in-progress reassignment for that partition. Repeatable.)typeactiondestdefaultrequiredhelpz--timeout-mszRequest timeout in milliseconds)r   r	   r   )add_argumentstrint)clsparser r   l/home/djax/ivt_ai_plugin/venv/lib/python3.10/site-packages/kafka/cli/admin/partitions/alter_reassignments.pyadd_arguments   s   
z)AlterPartitionReassignments.add_argumentsc                 C   s   i }|j D ]0}|dd\}}|dd\}}t|t|}	| dkr)d ||	< qdd |dD ||	< q|j||jd}
d	d
 |
 D S )N=   :cancelc                 S   s   g | ]}|rt |qS r   )r   ).0br   r   r   
<listcomp>   s    z7AlterPartitionReassignments.command.<locals>.<listcomp>,)
timeout_msc                 S   s,   i | ]\}}d |j |jf |r|jndqS )z%s:%dN)topic	partition__name__)r   tperrr   r   r   
<dictcomp>"   s    z7AlterPartitionReassignments.command.<locals>.<dictcomp>)	r   rsplitr   r   lowersplitalter_partition_reassignmentsr   items)r   clientargsr   spectp_strreplicas_strr   r   r    resultsr   r   r   command   s   

z#AlterPartitionReassignments.commandN)r   
__module____qualname__COMMANDHELPclassmethodr   r.   r   r   r   r   r      s    
r   N)kafka.structsr   r   r   r   r   r   <module>   s    