from googleapiclient import discovery
from oauth2client.client import GoogleCredentials

credentials = GoogleCredentials.get_application_default()

containerClient = discovery.build('container', 'v1', credentials=credentials)

location = 'projects/pingcap-gardener/locations/' \
           'asia-northeast1/clusters/staging-seed-asia-northeast1-83d8caa'
nodePoolId = location + '/nodePools/gardener-node-pool-9a09a04'
request= containerClient.projects().locations().clusters().nodePools().setSize(
    name=nodePoolId,
    body={'nodeCount': 1},
)
request.execute()


location = 'projects/pingcap-gardener/locations/' \
           'asia-southeast1/clusters/staging-seed-asia-southeast1-b7bc5b9'
nodePoolId = location + '/nodePools/gardener-node-pool-f4fd362'
request= containerClient.projects().locations().clusters().nodePools().setSize(
    name=nodePoolId,
    body={'nodeCount': 1},
)
request.execute()


location = 'projects/pingcap-gardener/locations/' \
           'us-central1/clusters/staging-seed-us-central1-6159679'
nodePoolId = location + '/nodePools/gardener-node-pool-1eb7a40'
request= containerClient.projects().locations().clusters().nodePools().setSize(
    name=nodePoolId,
    body={'nodeCount': 1},
)
request.execute()

location = 'projects/pingcap-gardener/locations/' \
           'us-west1/clusters/staging-seed-us-west1-73cb72c'
nodePoolId = location + '/nodePools/gardener-node-pool-8734cb0'
request= containerClient.projects().locations().clusters().nodePools().setSize(
    name=nodePoolId,
    body={'nodeCount': 2},
)
request.execute()

def scaleup():
    return "ok"