o
    `j                     @   sP   d dl Z d dlZd dl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)KafkaProducer   )add_common_cli_argsconfigure_loggingbuild_connect_kwargsc                 C   sN   t j| pddd}t| |d}|jddtddd	 |jd
tddd |S )Nzpython -m kafka.producerzKafka console producer)progdescriptionzproducer optionsz-tz--topicTzpublish to topic)typerequiredhelpz
--encodingzutf-8z#byte encoding for produced messages)r	   defaultr   )argparseArgumentParserr   add_argument_groupadd_argumentstr)r   parseroptions r   Y/home/djax/ivt_ai_plugin/venv/lib/python3.10/site-packages/kafka/cli/producer/__init__.pymain_parser	   s   
r   c                    s  t |d}|| }t| tt t|}tdi |} fdd}z]z3	 zt }W n t	yH   t
j d}|sFY W W |  dS Y nw |j|j||jd| q' tyl    d Y W |  dS  ty    jd	dd
 Y W |  dS w |  w )N)r   c                    s,   t | tr jd| d d S  d|  d S )NzError producing messageexc_infozMessage produced: %s)
isinstance	Exceptionerrorinfo)
res_or_errloggerr   r   
log_result#   s   
zrun_cli.<locals>.log_resultT
r   )valuezBye!zError!r      r   )r   
parse_argsr   logging	getLogger__name__r   r   inputEOFErrorsysstdinreadrstripclosesendtopicencodeencodingadd_bothKeyboardInterruptr   r   critical)argsr   r   configkwargsproducerr    r"   r   r   r   run_cli   s<   



	

r:   )N)NN)r   r%   r*   kafkar   commonr   r   r   r   r:   r   r   r   r   <module>   s    
