o
    `j                     @   sT   d dl Z d dlZd dlmZ d dlmZ ddlmZmZm	Z	 d
ddZ
ddd	ZdS )    N)KafkaConsumer)
KafkaError   )add_common_cli_argsconfigure_loggingbuild_connect_kwargsc              	   C   s   t j| pddd}t| |d}|jddtddd	d
d |jddtdd |jddtdd |jddtddd |jdtddd |S )Nzpython -m kafka.consumerzKafka console consumer)progdescriptionzconsumer optionsz-tz--topicappendtopicsTzsubscribe to topic)typeactiondestrequiredhelpz-gz--groupzconsumer group)r   r   z-iz--group-instance-idz"static group membership identifierz-fz--formatstrzoutput format: str|raw|full)r   defaultr   z
--encodingzutf-8z'encoding to use for str output decode())argparseArgumentParserr   add_argument_groupadd_argumentr   )r   parseroptions r   Y/home/djax/ivt_ai_plugin/venv/lib/python3.10/site-packages/kafka/cli/consumer/__init__.pymain_parser	   s4   
r   c                 C   s"  t |d}|| }|jdvrtd|j t| tt}t|}t	d|j
|jd|}||j zVz%|D ] }|jdkrJt|j|j q9|jdkrTt| q9t|j q9W n) tyo   |d Y W |  dS  ty   |jd	d
d Y W |  dS w W |  d S |  w )N)r   )r   rawfullzUnrecognized format: %s)group_idgroup_instance_idr   r   zBye!r   zError!T)exc_info   r   )r   
parse_argsformat
ValueErrorr   logging	getLogger__name__r   r   groupr   	subscriber   printvaluedecodeencodingKeyboardInterruptinfoclose	Exceptioncritical)argsr   r   configloggerkwargsconsumermr   r   r   run_cli!   sB   







r9   )N)NN)r   r%   kafkar   kafka.errorsr   commonr   r   r   r   r9   r   r   r   r   <module>   s    
