Combine broker_version/api_versions -> versions.py

kafka/cli/admin/cluster/__init__.py

@@ -1,8 +1,7 @@
-from .api_versions import GetApiVersions
-from .broker_version import GetBrokerVersion
 from .describe import DescribeCluster
-from .log_dirs import DescribeLogDirs, AlterLogDirs
 from .features import DescribeFeatures, UpdateFeatures
+from .log_dirs import DescribeLogDirs, AlterLogDirs
+from .versions import GetApiVersions, GetBrokerVersion
 
 
 class ClusterCommandGroup:

kafka/cli/admin/cluster/api_versions.py → kafka/cli/admin/cluster/versions.py

@@ -16,3 +16,18 @@ def command(cls, client, args):
         api_versions = client.api_versions()
         return {(k.value if args.raw else k.name): v for k, v in api_versions.items()
                 if k in api_keys}
+
+
+class GetBrokerVersion:
+    COMMAND = 'broker-version'
+    HELP = 'Get Version for Broker'
+
+    @classmethod
+    def add_arguments(cls, parser):
+        parser.add_argument('--broker', required=True)
+
+    @classmethod
+    def command(cls, client, args):
+        broker_id = int(args.broker)
+        bvd = client.get_broker_version_data(broker_id)
+        return {broker_id: '.'.join(map(str, bvd.broker_version))}