# Copyright 2017 Google Inc.

#

# Licensed under the Apache License, Version 2.0 (the "License");

# you may not use this file except in compliance with the License.

# You may obtain a copy of the License at

#

# http://www.apache.org/licenses/LICENSE-2.0

#

# Unless required by applicable law or agreed to in writing, software

# distributed under the License is distributed on an "AS IS" BASIS,

# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

# See the License for the specific language governing permissions and

# limitations under the License.

import argparse

import googleapiclient.discovery

def list_clusters_and_nodepools(project_id, zone):

"""Lists all clusters and associated node pools."""

service = googleapiclient.discovery.build("container", "v1")

clusters_resource = service.projects().zones().clusters()

clusters_response = clusters_resource.list(

projectId=project_id, zone=zone

).execute()

for cluster in clusters_response.get("clusters", []):

print(

"Cluster: {}, Status: {}, Current Master Version: {}".format(

cluster["name"], cluster["status"], cluster["currentMasterVersion"]

)

)

nodepools_response = (

clusters_resource.nodePools()

.list(projectId=project_id, zone=zone, clusterId=cluster["name"])

.execute()

)

for nodepool in nodepools_response["nodePools"]:

print(

" -> Pool: {}, Status: {}, Machine Type: {}, "

"Autoscaling: {}".format(

nodepool["name"],

nodepool["status"],

nodepool["config"]["machineType"],

nodepool.get("autoscaling", {}).get("enabled", False),

)

)

if __name__ == "__main__":

parser = argparse.ArgumentParser(

description=__doc__, formatter_class=argparse.RawDescriptionHelpFormatter

)

subparsers = parser.add_subparsers(dest="command")

list_clusters_and_nodepools_parser = subparsers.add_parser(

"list_clusters_and_nodepools", help=list_clusters_and_nodepools.__doc__

)

list_clusters_and_nodepools_parser.add_argument("project_id")

list_clusters_and_nodepools_parser.add_argument("zone")

args = parser.parse_args()

if args.command == "list_clusters_and_nodepools":

list_clusters_and_nodepools(args.project_id, args.zone)

else:

parser.print_help()