o
    `j%                     @   s   G d d dZ dS )c                   @   s,   e Zd ZdZdZedd Zedd ZdS )CreatePartitionscreatez0Create additional partitions for existing topicsc              
   C   s@   |j ddtddg ddd |j dtd d	d
 |j dddd d S )Nz-pz--topic-partitionsappendtopic_partitionsTz-TOPIC:TOTAL_PARTITION_COUNT pair (repeatable))typeactiondestdefaultrequiredhelpz--timeout-mszRequest timeout in milliseconds)r   r   r
   z--validate-only
store_truez9Validate the request without actually creating partitions)r   r
   )add_argumentstrint)clsparser r   _/home/djax/ivt_ai_plugin/venv/lib/python3.10/site-packages/kafka/cli/admin/partitions/create.pyadd_arguments   s   
zCreatePartitions.add_argumentsc                 C   s@   i }|j D ]}|dd\}}t|||< q|j||j|jdS )N:   )
timeout_msvalidate_only)r   rsplitr   create_partitionsr   r   )r   clientargsr   spectopiccountr   r   r   command   s   
zCreatePartitions.commandN)__name__
__module____qualname__COMMANDHELPclassmethodr   r   r   r   r   r   r      s    
r   N)r   r   r   r   r   <module>   s    