ansible/library/cloud/rax_clb

#!/usr/bin/python -tt
# This file is part of Ansible
#
# Ansible 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.
#
# Ansible 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 Ansible.  If not, see <http://www.gnu.org/licenses/>.

DOCUMENTATION = '''
---
module: rax_clb
short_description: create / delete a load balancer in Rackspace Public Cloud
description:
     - creates / deletes a Rackspace Public Cloud load balancer.
version_added: "1.4"
options:
  state:
    description:
     - Indicate desired state of the resource
    choices: ['present', 'absent']
    default: present
  credentials:
    description:
     - File to find the Rackspace credentials in (ignored if C(api_key) and
       C(username) are provided)
    default: null
    aliases: ['creds_file']
  api_key:
    description:
     - Rackspace API key (overrides C(credentials))
  username:
    description:
     - Rackspace username (overrides C(credentials))
  name:
    description:
     - Name to give the load balancer
    default: null
  algorithm:
    description:
     - algorithm for the balancer being created
    choices: ['RANDOM', 'LEAST_CONNECTIONS', 'ROUND_ROBIN', 'WEIGHTED_LEAST_CONNECTIONS', 'WEIGHTED_ROUND_ROBIN']
    default: LEAST_CONNECTIONS
  port:
    description:
     - Port for the balancer being created
    default: 80
  protocol:
    description:
     - Protocol for the balancer being created
    choices: ['DNS_TCP', 'DNS_UDP' ,'FTP', 'HTTP', 'HTTPS', 'IMAPS', 'IMAPv4', 'LDAP', 'LDAPS', 'MYSQL', 'POP3', 'POP3S', 'SMTP', 'TCP', 'TCP_CLIENT_FIRST', 'UDP', 'UDP_STREAM', 'SFTP']
    default: HTTP
  timeout:
    description:
     - timeout for communication between the balancer and the node
    default: 30
  type:
    description:
     - type of interface for the balancer being created
    choices: ['PUBLIC', 'SERVICENET']
    default: PUBLIC
  region:
    description:
     - Region to create the load balancer in
    default: DFW
  wait:
    description:
     - wait for the balancer to be in state 'running' before returning
    default: "no"
    choices: [ "yes", "no" ]
  wait_timeout:
    description:
     - how long before wait gives up, in seconds
    default: 300
requirements: [ "pyrax" ]
author: Christopher H. Laco, Jesse Keating
notes:
  - The following environment variables can be used, C(RAX_USERNAME),
    C(RAX_API_KEY), C(RAX_CREDS), C(RAX_CREDENTIALS), C(RAX_REGION).
  - C(RAX_CREDENTIALS) and C(RAX_CREDS) points to a credentials file
    appropriate for pyrax
  - C(RAX_USERNAME) and C(RAX_API_KEY) obviate the use of a credentials file
  - C(RAX_REGION) defines a Rackspace Public Cloud region (DFW, ORD, LON, ...)
'''

EXAMPLES = '''
- name: Build a Load Balancer
  gather_facts: False

  tasks:
    - name: Provision Load Balancer
      rax_lb:
        credentials: ~/.raxpub
        name: my-lb
        port: 9000
        protocol: HTTP
        type: SERVICENET
        timeout: 30
        region: DFW
        wait: yes
        state: present
      register: my_lb
'''

import sys
import os
import time

try:
    import pyrax
    import pyrax.utils
    from pyrax import exc
except ImportError:
    print("failed=True msg='pyrax required for this module'")
    sys.exit(1)


def cloud_load_balancer(module, state, name, meta, algorithm, port, protocol, type, timeout,
                         wait, wait_timeout):
    for arg in (state, name, port, protocol, type):
        if not arg:
            module.fail_json(msg='%s is required for cloud_loadbalancers' % arg)

    changed = False
    balancer = None
    balancers = []
    instances = []

    for balancer in pyrax.cloud_loadbalancers.list():
        if name != balancer.name:
            continue

        balancers.append(balancer)

    if state == 'present':
        if not balancers:
            try:
                balancers=[pyrax.cloud_loadbalancers.create(
                    name,
                    metadata=meta,
                    algorithm=algorithm,
                    port=port,
                    protocol=protocol,
                    timeout=timeout,
                    virtual_ips=[pyrax.cloud_loadbalancers.VirtualIP(type=type)]
                )]
                changed = True
            except Exception, e:
                module.fail_json(msg='%s' % e.message)

        for balancer in balancers:
            if wait:
                pyrax.utils.wait_for_build(balancer, interval=5, attempts=5)

            if balancer.status == 'ACTIVE':
                instance = dict(
                    id=balancer.id,
                    algorithm=balancer.algorithm,
                    cluster=balancer.cluster,
                    connectionLogging=balancer.connectionLogging,
                    contentCaching=balancer.contentCaching,
                    halfClosed=balancer.halfClosed,
                    port=balancer.port,
                    protocol=balancer.protocol,
                    sourceAddresses=balancer.sourceAddresses,
                    name=balancer.name,
                    timeout=balancer.timeout,
                    status=balancer.status
                )
                instances.append(instance)
            elif balancer.status == 'ERROR':
                module.fail_json(msg='%s failed to build' % balancer.id)
            elif wait:
                module.fail_json(msg='Timeout waiting on %s' % balancer.id)

    elif state == 'absent':
        if balancer:
            try:
                balancer.delete()
                changed = True
            except Exception, e:
                module.fail_json(msg='%s' % e.message)

            instance = dict(
                id=balancer.id,
                algorithm=balancer.algorithm,
                cluster=None,
                connectionLogging=None,
                contentCaching=None,
                halfClosed=None,
                port=balancer.port,
                protocol=balancer.protocol,
                sourceAddresses=None,
                name=balancer.name,
                timeout=balancer.timeout,
                status=balancer.status
            )
            instances.append(instance)

    module.exit_json(changed=changed, balancers=instances)


def main():
    module = AnsibleModule(
        argument_spec=dict(
            state=dict(default='present',
                       choices=['present', 'absent']),
            credentials=dict(aliases=['creds_file']),
            api_key=dict(),
            username=dict(),
            region=dict(),
            name=dict(),
            meta=dict(type='dict', default={}),
            algorithm=dict(choices=['RANDOM', 'LEAST_CONNECTIONS', 'ROUND_ROBIN', 'WEIGHTED_LEAST_CONNECTIONS', 'WEIGHTED_ROUND_ROBIN'], default='LEAST_CONNECTIONS'),
            port=dict(type='int', default=80),
            protocol=dict(choices=['DNS_TCP', 'DNS_UDP' ,'FTP', 'HTTP', 'HTTPS', 'IMAPS', 'IMAPv4', 'LDAP', 'LDAPS', 'MYSQL', 'POP3', 'POP3S', 'SMTP', 'TCP', 'TCP_CLIENT_FIRST', 'UDP', 'UDP_STREAM', 'SFTP'], default='HTTP'),
            type=dict(choices=['PUBLIC', 'SERVICENET'], default='PUBLIC'),
            timeout=dict(type='int', default=30),
            wait=dict(type='bool'),
            wait_timeout=dict(default=300)
        )
    )

    credentials = module.params.get('credentials')
    api_key = module.params.get('api_key')
    username = module.params.get('username')
    region = module.params.get('region')
    state = module.params.get('state')
    name = module.params.get('name')
    meta = module.params.get('meta')
    algorithm = module.params.get('algorithm')
    port = module.params.get('port')
    protocol = module.params.get('protocol')
    type = module.params.get('type')
    timeout = int(module.params.get('timeout'))
    wait = module.params.get('wait')
    wait_timeout = int(module.params.get('wait_timeout'))

    try:
        username = username or os.environ.get('RAX_USERNAME')
        api_key = api_key or os.environ.get('RAX_API_KEY')
        credentials = (credentials or
                       os.environ.get('RAX_CREDENTIALS') or
                       os.environ.get('RAX_CREDS_FILE'))
        region = region or os.environ.get('RAX_REGION')

    except KeyError, e:
        module.fail_json(msg='Unable to load %s' % e.message)

    try:
        pyrax.set_setting("identity_type", "rackspace")
        if api_key and username:
            pyrax.set_credentials(username, api_key=api_key, region=region)
        elif credentials:
            credentials = os.path.expanduser(credentials)
            pyrax.set_credential_file(credentials, region=region)
        else:
            raise Exception('No credentials supplied!')
    except Exception, e:
        module.fail_json(msg='%s' % e.message)

    cloud_load_balancer(module, state, name, meta, algorithm, port, protocol, type, timeout, wait, wait_timeout)

# this is magic, see lib/ansible/module_common.py
#<<INCLUDE_ANSIBLE_MODULE_COMMON>>

main()