-
-
Save sremc/fd1ba994b39f2ee654b4ddbc924b82aa to your computer and use it in GitHub Desktop.
Upgrade service via Rancher API using python
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
import os | |
import requests | |
import json | |
import time | |
environmentName = 'some-env' | |
serviceName = 'some-service' | |
newImage = 'docker:arkka/some-image:tag' | |
# Find stack based on their name | |
r = requests.get(os.environ['RANCHER_URL'] + 'v1/environments?name=' + environmentName, | |
auth=(os.environ['RANCHER_ACCESS_KEY'], os.environ['RANCHER_SECRET_KEY'])) | |
environment = r.json()['data'][0] | |
# Find service based on their name and environmentId | |
r = requests.get(os.environ['RANCHER_URL'] + 'v1/services?name=' + serviceName + '&environmentId=' + environment['id'], | |
auth=(os.environ['RANCHER_ACCESS_KEY'], os.environ['RANCHER_SECRET_KEY'])) | |
service = r.json()['data'][0] | |
launchConfig = service['launchConfig'] | |
# Update launchConfig with newImage | |
launchConfig['imageUuid'] = newImage | |
# Construct payload for upgrade | |
payload = { | |
'inServiceStrategy': { | |
'batchSize': 1, | |
'intervalMillis': 2000, | |
'startFirst': False, | |
'launchConfig': launchConfig | |
} | |
} | |
headers = {'content-type': 'application/json'} | |
# Upgrade the service with payload | |
r = requests.post(os.environ['RANCHER_URL'] + 'v1/services/' + service['id'] + '/?action=upgrade', | |
data=json.dumps(payload), headers=headers, | |
auth=(os.environ['RANCHER_ACCESS_KEY'], os.environ['RANCHER_SECRET_KEY'])) | |
# Pool service upgrade status | |
state = 'upgrading' | |
sleep = 30 | |
retry = 10 | |
while (state != 'upgraded'): | |
print "service: " + service['name'] + " [upgrading]" | |
time.sleep(sleep) | |
r = requests.get(os.environ['RANCHER_URL'] + 'v1/services/' + service['id'], | |
auth=(os.environ['RANCHER_ACCESS_KEY'], os.environ['RANCHER_SECRET_KEY'])) | |
state = r.json()['state'] | |
retry -= 1 | |
if (retry <= 0): sys.exit() | |
print "service: " + service['name'] + " [upgraded]" | |
# Finish Upgrade | |
r = requests.post(os.environ['RANCHER_URL'] + 'v1/services/' + service['id'] + '/?action=finishupgrade', | |
headers=headers, auth=(os.environ['RANCHER_ACCESS_KEY'], os.environ['RANCHER_SECRET_KEY'])); |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment