rebootRouterVM.py

#!/usr/bin/python

#      Copyright 2015, Schuberg Philis BV
#
#      Licensed to the Apache Software Foundation (ASF) under one
#      or more contributor license agreements.  See the NOTICE file
#      distributed with this work for additional information
#      regarding copyright ownership.  The ASF licenses this file
#      to you 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.

# Script to upgrade a router VM to the new template
# Remi Bergsma - rbergsma@schubergphilis.com

import sys
import getopt
from cloudstackops import cloudstackops
import os.path

# Function to handle our arguments


def handleArguments(argv):
    global DEBUG
    DEBUG = 0
    global DRYRUN
    DRYRUN = 1
    global vmname
    vmname = ''
    global toCluster
    toCluster = ''
    global configProfileName
    configProfileName = ''
    global isProjectVm
    isProjectVm = 0
    global onlyRequired
    onlyRequired = 0
    global CLEANUP
    CLEANUP = 0

    # Usage message
    help = "Usage: ./" + os.path.basename(__file__) + ' [options] ' + \
        '\n  --config-profile -c <profilename>\tSpecify the CloudMonkey profile name to get the credentials from (or specify in ./config file)' + \
        '\n  --routerinstance-name -r <name>\tWork with this router (r-12345-VM)' + \
        '\n  --is-projectrouter\t\t\tThe specified router belongs to a project' + \
        '\n  --only-when-required\t\t\tOnly reboot when the RequiresUpgrade flag is set' + \
        '\n  --cleanup\t\t\t\tRestart router with cleanup' + \
        '\n  --debug\t\t\t\tEnable debug mode' + \
        '\n  --exec\t\t\t\tExecute for real'

    try:
        opts, args = getopt.getopt(
            argv, "hc:r:p", [
                "config-profile=", "routerinstance-name=", "debug", "exec", "cleanup", "is-projectrouter", "only-when-required"])
    except getopt.GetoptError:
        print help
        sys.exit(2)
    for opt, arg in opts:
        if opt == '-h':
            print help
            sys.exit()
        elif opt in ("-c", "--config-profile"):
            configProfileName = arg
        elif opt in ("-r", "--routerinstance-name"):
            vmname = arg
        elif opt in ("--cleanup"):
            CLEANUP = 1
        elif opt in ("--debug"):
            DEBUG = 1
        elif opt in ("--exec"):
            DRYRUN = 0
        elif opt in ("--is-projectrouter"):
            isProjectVm = 1
        elif opt in ("--only-when-required"):
            onlyRequired = 1

    # Default to cloudmonkey default config file
    if len(configProfileName) == 0:
        configProfileName = "config"

    # We need at least these vars
    if len(vmname) == 0:
        print help
        sys.exit()

# Parse arguments
if __name__ == "__main__":
    handleArguments(sys.argv[1:])

# Init CloudStack class
c = cloudstackops.CloudStackOps(DEBUG, DRYRUN)
c.instance_name = "N/A"

if DEBUG == 1:
    print "Warning: Debug mode is enabled!"

if DRYRUN == 1:
    print "Warning: dry-run mode is enabled, not running any commands!"

# make credentials file known to our class
c.configProfileName = configProfileName

# Init the CloudStack API
c.initCloudStackAPI()

if DEBUG == 1:
    print "DEBUG: API address: " + c.apiurl
    print "DEBUG: ApiKey: " + c.apikey
    print "DEBUG: SecretKey: " + c.secretkey

# Check cloudstack IDs
if DEBUG == 1:
    print "DEBUG: Checking CloudStack IDs of provided input.."

if isProjectVm == 1:
    projectParam = "true"
else:
    projectParam = "false"

# check routerID
routerID = c.checkCloudStackName({'csname': vmname,
                                  'csApiCall': 'listRouters',
                                  'listAll': 'true',
                                  'isProjectVm': projectParam})

# get router data
routerData = c.getRouterData({'name': vmname, 'isProjectVm': projectParam})
router = routerData[0]

if DEBUG == 1:
    print routerData

print "Note: Found router " + router.name + " that belongs to account " + str(router.account) + " with router ID " + router.id
print "Note: This router has " + str(len(router.nic)) + " nics."

# Pretty Slack messages
c.instance_name = router.name
c.slack_custom_title = "Domain"
c.slack_custom_value = router.domain
hostData = c.getHostData({'hostid': router.hostid})[0]
clusterData = c.listClusters({'id': hostData.clusterid})
c.cluster = clusterData[0].name

# Does this router need an upgrade?
if onlyRequired == 1 and router.requiresupgrade == 0:
    print "Note: This router does not need to be upgraded. Won't reboot because of --only-when-required flag. When you remove the flag and run the script again it will reboot anyway."
    sys.exit(0)

print "Note: Let's reboot the router VM.."

if DRYRUN == 1:
    print "Note: Would have rebooted router " + router.name + " (" + router.id + ")"
else:
    # Restart network with clean up
    if CLEANUP == 1:
        # If the network is a VPC
        if router.vpcid:
            c.task = "Restart VPC with clean up"
            message = "Restarting router " + router.name + " with clean up (" + router.id + ")"
            c.print_message(message=message, message_type="Note", to_slack=True)
            result = c.restartVPC(router.vpcid)
            if result == 1:
                print "Restarting failed, will try again!"
                result = c.restartVPC(router.vpcid)
                if result == 1:
                    message = "Restarting (" + router.id + ") with clean up failed.\nError: investigate manually!"
                    c.print_message(message=message, message_type="Error", to_slack=True)
                    sys.exit(1)
                else:
                    message = "Successfully restarted " + router.name + " (" + router.id + ")"
                    c.print_message(message=message, message_type="Note", to_slack=True)
            else:
                message = "Successfully restarted " + router.name + " (" + router.id + ")"
                c.print_message(message=message, message_type="Note", to_slack=True)
        # If the network is a Isolated network
        else:
            c.task = "Restart isolated network with clean up"
            message = "Restarting isolated network router " + router.name + " with clean up (" + router.id + ")"
            c.print_message(message=message, message_type="Note", to_slack=True)
            result = c.restartNetwork(router.guestnetworkid)
            if result == 1:
                print "Restarting failed, will try again!"
                result = c.restartNetwork(router.guestnetworkid)
                if result == 1:
                    message = "Restarting (" + router.id + ") with clean up failed.\nError: investigate manually!"
                    c.print_message(message=message, message_type="Error", to_slack=True)
                    sys.exit(1)
                else:
                    message = "Successfully restarted " + router.name + " (" + router.id + ")"
                    c.print_message(message=message, message_type="Note", to_slack=True)
            else:
                message = "Successfully restarted " + router.name + " (" + router.id + ")"
                c.print_message(message=message, message_type="Note", to_slack=True)
    else:
        # Reboot router
        c.task = "Reboot virtual router"
        message = "Rebooting router " + router.name + " (" + router.id + ")"
        c.print_message(message=message, message_type="Note", to_slack=True)
        result = c.rebootRouter(router.id)
        if result == 1:
            print "Rebooting failed, will try again!"
            result = c.rebootRouter(router.id)
            if result == 1:
                message = "Rebooting (" + router.id + ") failed.\nError: Investigate manually!"
                c.print_message(message=message, message_type="Error", to_slack=True)
                sys.exit(1)
        else:
            message = "Successfully rebooted " + router.name + " (" + router.id + ")"
            c.print_message(message=message, message_type="Note", to_slack=True)

print "Note: We're done!"