o
    `jG                     @   s   d dl Z d dlZd dlZd dlmZ d dlmZ ddlmZ ddlm	Z	 ddl
mZ ddlmZ dd	lmZ dd
lmZ ddlmZ ddlmZ ddlmZmZmZ d dlmZ ee	eeeeeefZdddZdddZdddZdS )    N)pprint)KafkaAdminClient   )ACLsCommandGroup)ClusterCommandGroup)ConfigsCommandGroup)GroupsCommandGroup)PartitionsCommandGroup)TopicsCommandGroup)TransactionsCommandGroup)UsersCommandGroup   )add_common_cli_argsconfigure_loggingbuild_connect_kwargs)BrokerResponseErrorc                 C   s   t t| dS )Nprog)build_parserDEFAULT_COMMAND_GROUPSr    r   V/home/djax/ivt_ai_plugin/venv/lib/python3.10/site-packages/kafka/cli/admin/__init__.pymain_parser   s   r   r   c           
      C   s   t j|pddd}t|dd |djdtdd	d
 |jdddd}| D ]9}|j|j|j	d}|j
|d |jdddd}|jD ]}|j|j|j	d}|d}	||	 |j
|jd qAq%|S )Nzpython -m kafka.adminzKafka Admin Client)r   descriptionF)bootstrap_requiredoutputz--formatrawzoutput format: raw|json)typedefaulthelpgroupGROUPzAvailable command groups)destmetavartitle)r   )r    commandCOMMANDzAvailable commandszcommand options)r%   )argparseArgumentParserr   add_argument_groupadd_argumentstradd_subparsers
add_parserr!   HELPset_defaultsCOMMANDSr&   add_argumentsr%   )
groupsr   parser
groups_subr    group_parsercommands_subcmdcommand_parseroptionsr   r   r   r      s*   



r   c           	   
   C   s  t |d}|| }|js|  dS |js|j  dS |jdvr,td|j  dS t| t	t
}zt|}W n! ty\ } z|  t|j d|  W Y d }~dS d }~ww tdi |}zz-|||}|jdkrvt| n|jdkrt|dr| }tt| W W |  d	S  ty } zt| W Y d }~W |  dS d }~w ty } zt|jd	  W Y d }~W |  dS d }~w ty } z|| |  W Y d }~W |  dS d }~w ty   |jd
dd Y W |  dS w |  w )Nr   r   )r   jsonzUnrecognized format: z: r   r:   to_dictr   zError!T)exc_infor   )r   
parse_argsr    
print_helpr%   formatprintr   logging	getLogger__name__r   
ValueErrorprint_usager   r   r   hasattrr;   r:   dumpscloser   argsAttributeError	exception	Exceptioncritical)	rI   r   r3   configloggerkwargsexcclientresultr   r   r   run_cli4   sl   








	

rT   )N)r   N)NN) r'   r:   rA   r   kafka.admin.clientr   aclsr   clusterr   configsr   r2   r   
partitionsr	   topicsr
   transactionsr   usersr   commonr   r   r   kafka.errorsr   r   r   r   rT   r   r   r   r   <module>   s,    

