o `j`@s(ddlmZdddZGdddZdS)) defaultdict) preferreduncleanc@s,eZdZdZdZeddZeddZdS) ElectLeadersz elect-leadersz&Trigger leader election for partitionsc Csp|jdtdttdd|jddtddgd d |jd d tdd gdd |jdtddd|jddddddS)Nz--election-typerz"Election type (default: preferred))typedefaultchoiceshelpz-pz--topic-partitionappendtopic_partitionszZTOPIC:PARTITION pair (repeatable). Omit to elect leaders for all partitions of all topics.)ractiondestrr z-tz--topictopicszbElect leaders for all partitions of TOPIC (repeatable). Mutually exclusive with --topic-partition.z --timeout-mszRequest timeout in milliseconds)rrr z--no-raise-errors raise_errors store_falsezCDo not raise on partition-level errors; return the response instead)rr r ) add_argumentstrsorted_ELECTION_TYPESint)clsparserrf/home/djax/ivt_ai_plugin/venv/lib/python3.10/site-packages/kafka/cli/admin/partitions/elect_leaders.py add_arguments s. zElectLeaders.add_argumentscCs|jr |jr td|jr-tt}|jD]}|dd\}}||t|qt|}n |jr4|j}nd}|j t |j ||j |j dS)Nz4--topic-partition and --topic are mutually exclusive:r)r timeout_msr)r r ValueErrorrlistrsplitr rdict elect_leadersr election_typerr)rclientargstopic2partitionsspectopic partitionr rrrcommand"s&   zElectLeaders.commandN)__name__ __module__ __qualname__COMMANDHELP classmethodrr*rrrrrs rN) collectionsrrrrrrrs