From: Michael Jeanson Date: Mon, 17 Aug 2015 23:05:14 +0000 (-0400) Subject: Standalone nodes upgrade script X-Git-Url: http://git.lttng.org./?a=commitdiff_plain;h=92150710909225057c6855dca21e1c9269a68f43;p=lttng-ci.git Standalone nodes upgrade script --- diff --git a/automation/update_standalone.py b/automation/update_standalone.py new file mode 100644 index 0000000..6f7094c --- /dev/null +++ b/automation/update_standalone.py @@ -0,0 +1,192 @@ +#!/usr/bin/python +# -*- coding: utf-8 -*- +# +# Copyright (C) 2015 - Michael Jeanson +# +# This program is free software: you can redistribute it and/or modify +# it under the terms of the GNU General Public License as published by +# the Free Software Foundation, either version 3 of the License, or +# (at your option) any later version. +# +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with this program. If not, see . + +""" This script is used to upgrade the base snapshot of standalone ci slaves """ + +USERNAME = '' +PASSWORD = '' +JENKINS_URL = 'https://ci.lttng.org' + +DISTRO_LIST = ['el', 'sles', 'ubuntu'] +DEFAULT_DISTRO = 'ubuntu' +DISTRO_COMMAND = { + 'el': 'yum update -y && yum clean all', + 'sles': 'zypper --non-interactive refresh && zypper --non-interactive patch --auto-agree-with-licenses --with-interactive', + 'ubuntu': 'apt-get update && apt-get dist-upgrade -V -y && apt-get clean && apt-get --purge autoremove -y', +} + +BASESNAP = 'base-configuration' + +SNAPSHOTXML = """ + + %s + Snapshot of OS install and updates + + +""" % BASESNAP + +import argparse +import sys +import libvirt +#from jenkinsapi.jenkins import Jenkins +from time import sleep +import paramiko +import select + + +def main(): + """ Main """ + + parser = argparse.ArgumentParser(description='Update base snapshot.') + parser.add_argument('instance_name', metavar='INSTANCE', type=str, + help='the shortname of the instance to update') + parser.add_argument('vmhost_name', metavar='VMHOST', type=str, + help='the hostname of the VM host') + parser.add_argument('--distro', choices=DISTRO_LIST, + default=DEFAULT_DISTRO, type=str, + help='the distro of the target instance') + + args = parser.parse_args() + + instance_name = args.instance_name + vmhost_name = args.vmhost_name + distro = args.distro + + + # Get jenkibs connexion + #jenkins = Jenkins(JENKINS_URL, username=USERNAME, password=PASSWORD) + + # Get jenkins node + #node = jenkins.get_node(instance_name) + + # Set node temporarily offline + #node.toggle_temporarily_offline('Down for upgrade to base snapshot') + + # Get libvirt connexion + print("Opening libvirt connexion to %s..." % vmhost_name) + vmhost = libvirt.open("qemu+ssh://root@%s/system" % vmhost_name) + + if not vmhost: + print("Could not connect to libvirt on %s" % vmhost_name) + sys.exit(1) + + # Get instance + print("Getting instance %s from libvirt..." % instance_name) + vminstance = vmhost.lookupByName(instance_name) + + if not vminstance: + print("Could not get instance %s on %s" % (instance_name, vmhost_name)) + sys.exit(1) + + # If instance is running, shutdown + print("Checking if instance %s is running..." % instance_name) + if vminstance.isActive(): + try: + print("Shutting down instance %s" % instance_name) + vminstance.destroy() + except: + print("Failed to shutdown %s", instance_name) + sys.exit(1) + + + # Revert to base snapshot + print("Getting base snapshot...") + basesnap = vminstance.snapshotLookupByName(BASESNAP) + if not basesnap: + print("Could not find base snapshot %s" % BASESNAP) + sys.exit(1) + + #if not basesnap.isCurrent(): + # print("Not current snapshot") + + print("Reverting to base snapshot...") + try: + vminstance.revertToSnapshot(basesnap) + except: + print("Failed to revert to base snapshot %s" % basesnap.getName()) + sys.exit(1) + + # Launch instance + try: + print("Starting instance %s.." % instance_name) + vminstance.create() + except: + print("Failed to start instance %s" % instance_name) + sys.exit(1) + + + # Wait for instance to boot + print("Waiting for instance to boot...") + sleep(10) + + # Run dist-upgrade + print("Running upgrade command...") + client = paramiko.SSHClient() + client.set_missing_host_key_policy(paramiko.AutoAddPolicy()) + client.load_system_host_keys() + client.connect(instance_name, username="root") + stdin, stdout, stderr = client.exec_command(DISTRO_COMMAND[distro]) + while not stdout.channel.exit_status_ready(): + if stdout.channel.recv_ready(): + rl, wl, xl = select.select([stdout.channel], [], [], 0.0) + if len(rl) > 0: + print(stdout.channel.recv(1024)), + + if stdout.channel.recv_exit_status() != 0: + print("Update command failed!") + sys.exit(1) + + # Close ssh connexion + client.close() + + # Shutdown VM + print("Shutting down instance...") + try: + vminstance.shutdown() + except: + print("Failed to shutdown instance %s" % instance_name) + sys.exit(1) + + while vminstance.isActive(): + sleep(1) + print("Waiting for instance to shutdown...") + + # Delete original base snapshot + print("Deleting current base snapshot...") + try: + basesnap.delete() + except: + print("Failed to delete base snapshot %s" % basesnap.getName()) + sys.exit(1) + + # Create new base snapshot + print("Creating new base snapshot...") + try: + vminstance.snapshotCreateXML(SNAPSHOTXML) + except: + print("Failed to create new snapshot.") + sys.exit(1) + + # And we're done! + print("All done!") + + +if __name__ == "__main__": + main() + +# EOF