o `j@s,ddlmZGdddZGdddZdS))TopicPartitionReplicac@,eZdZdZdZeddZeddZdS)DescribeLogDirszdescribe-log-dirsz#Get topic log directories and statscCs,|jdtdddd|jdtdddddS) Nz--brokerappendbrokerszQuery specific broker(s))typeactiondesthelpz--topictopicsz Get data about specific topic(s)) add_argumentintstrclsparserr^/home/djax/ivt_ai_plugin/venv/lib/python3.10/site-packages/kafka/cli/admin/cluster/log_dirs.py add_argumentsszDescribeLogDirs.add_argumentscCs|j|j|jdS)N)topic_partitionsr)describe_log_dirsr r)rclientargsrrrcommand szDescribeLogDirs.commandN__name__ __module__ __qualname__COMMANDHELP classmethodrrrrrrrs rc@r) AlterLogDirszalter-log-dirsz>Move replicas between log directories on their hosting brokersc Cs|jddtddgddddS)Nz-az --assignmentr assignmentsTzTOPIC:PARTITION:BROKER_ID=/absolute/log/dir/path (repeatable). Instructs BROKER_ID to move its replica of TOPIC:PARTITION into the given log directory.)rrr defaultrequiredr )r rrrrrrs  zAlterLogDirs.add_argumentsc Csji}|jD]!}|dd\}}|dd\}}} t|t|t| } ||| <q||} dd| DS)N=:cSs.i|]\}}|jd|jd|j|jqS)r')topic partition broker_idr).0tprerrrrr (sz(AlterLogDirs.command..)r"rsplitrr alter_replica_log_dirsitems) rrrr"spectpr_strlog_dirr)r*r+r-resultrrrrs   zAlterLogDirs.commandNrrrrrr!s r!N) kafka.structsrrr!rrrrs