Synnefo

#!/usr/bin/env python

# Invalid name for type module. pylint: disable-msg=C0103

"""

Continuous Integration script for Synnefo.

"""

import os

import utils

from optparse import OptionParser

CREATE_SERVER_CMD = "create"

DELETE_SERVER_CMD = "delete"

BUILD_SYNNEFO_CMD = "build"

BUILD_DOCS_SYNNEFO_CMD = "docs"

DEPLOY_SYNNEFO_CMD = "deploy"

TEST_SYNNEFO_CMD = "test"

RUN_BURNIN_CMD = "burnin"

CREATE_X2GO_FILE = "x2goplugin"

SHELL_CONNECT = "shell"

ALL_CMDS = "all"

COMMANDS_IN_ALL_MODE = [

    CREATE_SERVER_CMD,

    BUILD_SYNNEFO_CMD,

    BUILD_DOCS_SYNNEFO_CMD,

    DEPLOY_SYNNEFO_CMD,

    TEST_SYNNEFO_CMD,

    RUN_BURNIN_CMD,

]

AVAILABLE_COMMANDS = [

    CREATE_X2GO_FILE,

    DELETE_SERVER_CMD,

    SHELL_CONNECT,

] + COMMANDS_IN_ALL_MODE

USAGE = """usage: %%prog [options] command[,command...]

command:

    * %s: Create the slave server

    * %s: Create debian packages for Synnefo in the created server

    * %s: Create documentation for Synnefo in the created server

    * %s: Deploy Synnefo in created server

    * %s: Run Synnefo unittests

    * %s: Run snf-burnin in the deployed Synnefo

    * %s: Create x2go plugin file

    * %s: Delete the slave server

    * %s: Connect to the server using ssh

    * %s: Run all the available commands

""" % tuple([CREATE_SERVER_CMD,

             BUILD_SYNNEFO_CMD,

             BUILD_DOCS_SYNNEFO_CMD,

             DEPLOY_SYNNEFO_CMD,

             TEST_SYNNEFO_CMD,

             RUN_BURNIN_CMD,

             CREATE_X2GO_FILE,

             DELETE_SERVER_CMD,

             SHELL_CONNECT,

             ALL_CMDS])

def main():  # pylint: disable=too-many-statements, too-many-branches

    """Parse command line options and run the specified actions"""

    parser = OptionParser(usage=USAGE)

    parser.add_option("-c", "--conf", dest="config_file", default=None,

                      help="Configuration file for SynnefoCI script")

    parser.add_option("--cloud", dest="kamaki_cloud", default=None,

                      help="Use specified cloud, as is in .kamakirc")

    parser.add_option("-f", "--flavor", dest="flavor", default=None,

                      help="Flavor to use for the server."

                           " Supports both search by name (reg expression)"

                           " with \"name:flavor name\" or by id with"

                           " \"id:flavor id\".")

    parser.add_option("-i", "--image", dest="image", default=None,

                      help="Image to use for the server."

                           " Supports both search by name (reg expression)"

                           " with \"name:image name\" or by id with"

                           " \"id:image id\".")

    parser.add_option("--ssh-keys", dest="ssh_keys", default=None,

                      help="Upload/Install the public ssh keys contained"

                           " in this file to the server")

    parser.add_option("--name", dest="server_name", default=None,

                      help="")

    parser.add_option("-n", "--build-id", dest="build_id", default=None,

                      type="int",

                      help="Specify a number to use to identify this build."

                           " One can later use this number to retrieve"

                           " information (such as IPs, passwords etc) about"

                           " the machines created. If not given this script"

                           " will create a new build-id.")

    parser.add_option("--fetch-packages", dest="fetch_packages",

                      default=None,

                      help="Download the debian packages that were created"

                           " during the '%s' step in this directory." %

                           BUILD_SYNNEFO_CMD)

    parser.add_option("--fetch-docs", dest="fetch_docs",

                      default=None,

                      help="Download the documentation that was created"

                           " during the '%s' step in this directory." %

                           BUILD_DOCS_SYNNEFO_CMD)

    parser.add_option("--schema", dest="schema", default=None,

                      help="Schema for snf-deploy.")

    parser.add_option("--local-repo", dest="local_repo", default=False,

                      action="store_true",

                      help="Instead of cloning from the official Synnefo"

                           " repo, copy and use the local one.")

    parser.add_option("--x2go-output", dest="x2go_output", default=None,

                      help="File where to save the x2go plugin html page.")

    parser.add_option("--no-colors", dest="use_colors",

                      default=True, action="store_false",

                      help="Don't use colorful output messages.")

    (options, args) = parser.parse_args()

    # ----------------------------------

    # Check arguments

    if len(args) != 1:

        msg = "ERROR: Command takes exactly one argument"

        parser.print_help()

        print

        print msg

        return

    commands = args[0]

    if commands == ALL_CMDS:

        for cmd in COMMANDS_IN_ALL_MODE:

            setattr(options, cmd, True)

    else:

        commands = commands.split(",")

        for command in commands:

            if command not in AVAILABLE_COMMANDS:

                msg = "ERROR: Unknown command: %s" % command

                parser.print_help()

                print

                print msg

                return

            else:

                setattr(options, command, True)

    # ----------------------------------

    # Initialize SynnefoCi

    utils.USE_COLORS = options.use_colors

    synnefo_ci = utils.SynnefoCI(config_file=options.config_file,

                                 build_id=options.build_id,

                                 cloud=options.kamaki_cloud)

    # ----------------------------------

    # Run commands

    if getattr(options, CREATE_SERVER_CMD, False):

        synnefo_ci.create_server(flavor=options.flavor,

                                 image=options.image,

                                 ssh_keys=options.ssh_keys,

                                 server_name=options.server_name)

        synnefo_ci.clone_repo(local_repo=options.local_repo)

    if getattr(options, BUILD_SYNNEFO_CMD, False):

        synnefo_ci.build_packages()

        if options.fetch_packages:

            dest = os.path.abspath(options.fetch_packages)

            synnefo_ci.fetch_packages(dest=dest)

    if getattr(options, BUILD_DOCS_SYNNEFO_CMD, False):

        synnefo_ci.build_documentation()

        if options.fetch_docs:

            dest = os.path.abspath(options.fetch_docs)

            synnefo_ci.fetch_documentation(dest=dest)

    if getattr(options, DEPLOY_SYNNEFO_CMD, False):

        synnefo_ci.deploy_synnefo(schema=options.schema)

    if getattr(options, TEST_SYNNEFO_CMD, False):

        synnefo_ci.unit_test()

    if getattr(options, RUN_BURNIN_CMD, False):

        synnefo_ci.run_burnin()

    if getattr(options, CREATE_X2GO_FILE, False):

        synnefo_ci.x2go_plugin(options.x2go_output)

    if getattr(options, DELETE_SERVER_CMD, False):

        synnefo_ci.destroy_server()

    if getattr(options, SHELL_CONNECT, False):

        synnefo_ci.shell_connect()

if __name__ == "__main__":

    main()