o `jP@s`ddlZddlZdddZddZddZd d Zd d Zd dZddZddZ dddZ dS)NTcCsl|d}|jddtd|dd|jddtd d d |jd d tddd |jddtd|jddtddS)N connectionz-bz--bootstrap-serversappendzGhost:port for cluster bootstrap server. Can be provided multiple times.)typeactionrequiredhelpz-Sz--security-protocol PLAINTEXTz(PLAINTEXT, SSL, SASL_PLAINTEXT, SASL_SSLrdefaultrz-Mz--sasl-mechanismPLAINz8PLAIN, GSSAPI, OAUTHBEARER, SCRAM-SHA-256, SCRAM-SHA-512z-Uz --sasl-user)rz-Pz--sasl-passwordadd_argument_group add_argumentstr)parserbootstrap_required connect_grouprN/home/djax/ivt_ai_plugin/venv/lib/python3.10/site-packages/kafka/cli/common.pyadd_connect_cli_argss"     rc Csti}|pgD]1}|d\}}zt|}Wn tyYnw|dkr&d}n |dkr-d}n|dkr3d}|||<q|S)N=NoneFalseFTrueT)splitint ValueError)propskwargspropkvrrr build_kwargss     r"cCsd|jstdg}|jD]}|dd|dDq t|j}|||j|j|j |j d|S)NzTpython -m kafka: error: the following arguments are required: -b/--bootstrap-serverscss |] }|r|VqdSN)strip).0srrr ,sz'build_connect_kwargs..,)bootstrap_serverssecurity_protocolsasl_mechanismsasl_plain_usernamesasl_plain_password) r)rextendrr" extra_configupdater*r+ sasl_user sasl_password)configr)entryrrrrbuild_connect_kwargs&s  r5cCs|d}|jddtddd|jddtd d d |jd d td dd |jdtddd|jdtddd|jdtddd|jdtddddS)Nloggingz-lz --log-levelCRITICALz,logging level, passed to logging.basicConfigr z-Lz--enable-loggerrzbenable a specific logger. Can be provided multiple times. If not provided, all loggers are enabledrrrz-Dz--disable-loggerz:disable a specific logger. Can be provided multiple times.z --log-formatz6log message format string, passed to logging.Formatterz--log-date-formatz3log date format string, passed to logging.Formatterz --log-filez)write logs to this file instead of stderrz --log-configapath to a logging configuration file for full control over handlers, formatters, etc. A .json (or .yaml/.yml, if PyYAML is installed) file is loaded as a logging.config.dictConfig; any other extension is loaded as a logging.config.fileConfig. The file owns handlers/formatters, so --log-format, --log-date-format and --log-file are ignored, but --enable-logger and --disable-logger still apply as logger level adjustments.r )r logging_grouprrradd_logging_cli_args8s:  r:cCs"|d}|jddtddddS)Nextendedz-Cz--extra-configrzcadditional configuration properties for client in "key=val" format. Can be provided multiple times.r8r )rextended_grouprrradd_extended_cli_argsVs  r=cCs|dr8zddl}Wntytd|fwt|}t||WddS1s1wYdS|dr_ddl}t|}t||WddS1sXwYdSt j j |dddS)zLConfigure logging from a dictConfig (.json/.yaml) or fileConfig (.ini) file.)z.yamlz.ymlrNz4PyYAML is required to load a YAML logging config: %sz.jsonF)disable_existing_loggers) endswithyaml ImportErrorropen _dict_config safe_loadjsonloadr6r3 fileConfig)pathr@frErrr_load_log_config]s    "  "rJcCs|ddtj|dS)Nr>F) setdefaultr6r3 dictConfig)cfgrrrrCqs rCcCs:ddddddd}||j}t|ddr>t|j|jdur=ttj d |jD]}t|}d |_ ||q-nJ|j pCtj }|jdur}|j rSt|j }nt}|||tj||jd |jD]}t|}||||qjn tj|||j|j d |jpgD] }t|tj d qdS) Nr (2)NOTSETDEBUGINFOWARNINGERRORr7 log_configF)datefmt)levelformatrZfilename) log_leveluppergetattrrJrX enable_loggerr6 getLoggersetLevelr7disabled log_format BASIC_FORMATlog_file FileHandler StreamHandler setFormatter Formatterlog_date_format addHandler basicConfigdisable_logger)r3_LOGGING_LEVELSr^nameloggerrehandlerrrrconfigure_loggingysH             rtcCst||t|t|dSr#)rr:r=)rrrrradd_common_cli_argss  ru)T) r6logging.configrr"r5r:r=rJrCrtrurrrrs -