Merge branch 'rax-autoscale' of https://github.com/sivel/ansible into sivel-rax-autoscale

library/cloud/rax_scaling_group

@@ -0,0 +1,403 @@
+#!/usr/bin/python
+# 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/>.
+
+# This is a DOCUMENTATION stub specific to this module, it extends
+# a documentation fragment located in ansible.utils.module_docs_fragments
+DOCUMENTATION = '''
+---
+module: rax_scaling_group
+short_description: Manipulate Rackspace Cloud Autoscale Groups
+description:
+    - Manipulate Rackspace Cloud Autoscale Groups
+version_added: 1.7
+options:
+  cooldown:
+    description:
+      - The period of time, in seconds, that must pass before any scaling can
+        occur after the previous scaling. Must be an integer between 0 and
+        86400 (24 hrs).
+  disk_config:
+    description:
+      - Disk partitioning strategy
+    choices:
+      - auto
+      - manual
+    default: auto
+  files:
+    description:
+      - 'Files to insert into the instance. Hash of C(remotepath: localpath)'
+    default: null
+  flavor:
+    description:
+      - flavor to use for the instance
+    required: true
+  image:
+    description:
+      - image to use for the instance. Can be an C(id), C(human_id) or C(name)
+    required: true
+  key_name:
+    description:
+      - key pair to use on the instance
+    default: null
+  loadbalancers:
+    description:
+      - List of load balancer C(id) and C(port) hashes
+  max_entities:
+    description:
+      - The maximum number of entities that are allowed in the scaling group.
+        Must be an integer between 0 and 1000.
+    required: true
+  meta:
+    description:
+      - A hash of metadata to associate with the instance
+    default: null
+  min_entities:
+    description:
+      - The minimum number of entities that are allowed in the scaling group.
+        Must be an integer between 0 and 1000.
+    required: true
+  name:
+    description:
+      - Name to give the scaling group
+    required: true
+  networks:
+    description:
+      - The network to attach to the instances. If specified, you must include
+        ALL networks including the public and private interfaces. Can be C(id)
+        or C(label).
+    default:
+      - public
+      - private
+  server_name:
+    description:
+      - The base name for servers created by Autoscale
+    required: true
+  state:
+    description:
+      - Indicate desired state of the resource
+    choices:
+      - present
+      - absent
+    default: present
+author: Matt Martz
+extends_documentation_fragment: rackspace
+'''
+
+EXAMPLES = '''
+---
+- hosts: localhost
+  gather_facts: false
+  connection: local
+  tasks:
+    - rax_scaling_group:
+        credentials: ~/.raxpub
+        region: ORD
+        cooldown: 300
+        flavor: performance1-1
+        image: bb02b1a3-bc77-4d17-ab5b-421d89850fca
+        min_entities: 5
+        max_entities: 10
+        name: ASG Test
+        server_name: asgtest
+        loadbalancers:
+            - id: 228385
+              port: 80
+      register: asg
+'''
+
+import os
+
+from uuid import UUID
+
+try:
+    import pyrax
+    HAS_PYRAX = True
+except ImportError:
+    HAS_PYRAX = False
+
+NON_CALLABLES = (basestring, bool, dict, int, list, type(None))
+PUBLIC_NET_ID = "00000000-0000-0000-0000-000000000000"
+SERVICE_NET_ID = "11111111-1111-1111-1111-111111111111"
+
+
+def asg_to_dict(obj):
+    instance = {}
+    for key in dir(obj):
+        value = getattr(obj, key)
+        if key == 'policies' and isinstance(value, list):
+            policies = []
+            for policy in value:
+                policies.append(asg_to_dict(policy))
+            instance[key] = policies
+        elif (isinstance(value, NON_CALLABLES) and not key.startswith('_')):
+            instance[key] = value
+    return instance
+
+
+def rax_asg(module, cooldown=300, disk_config=None, files={}, flavor=None,
+            image=None, key_name=None, loadbalancers=[], meta={},
+            min_entities=0, max_entities=0, name=None, networks=[],
+            server_name=None, state='present'):
+    changed = False
+
+    au = pyrax.autoscale
+    cnw = pyrax.cloud_networks
+    if not au or not cnw:
+        module.fail_json(msg='Failed to instantiate clients. This '
+                             'typically indicates an invalid region or an '
+                             'incorrectly capitalized region name.')
+
+    if state == 'present':
+        # Normalize and ensure all metadata values are strings
+        if meta:
+            for k, v in meta.items():
+                if isinstance(v, list):
+                    meta[k] = ','.join(['%s' % i for i in v])
+                elif isinstance(v, dict):
+                    meta[k] = json.dumps(v)
+                elif not isinstance(v, basestring):
+                    meta[k] = '%s' % v
+
+        # Check if the provided image is a UUID and if not, search for an
+        # appropriate image using human_id and name
+        if image:
+            try:
+                UUID(image)
+            except ValueError:
+                try:
+                    image = cs.images.find(human_id=image)
+                except(cs.exceptions.NotFound,
+                       cs.exceptions.NoUniqueMatch):
+                    try:
+                        image = cs.images.find(name=image)
+                    except (cs.exceptions.NotFound,
+                            cs.exceptions.NoUniqueMatch):
+                        module.fail_json(msg='No matching image found (%s)' %
+                                             image)
+
+            image = pyrax.utils.get_id(image)
+
+        # Check if the provided network is a UUID and if not, search for an
+        # appropriate network using label
+        nics = []
+        if networks:
+            for network in networks:
+                try:
+                    UUID(network)
+                except ValueError:
+                    if network.lower() == 'public':
+                        nics.extend(cnw.get_server_networks(PUBLIC_NET_ID))
+                    elif network.lower() == 'private':
+                        nics.extend(cnw.get_server_networks(SERVICE_NET_ID))
+                    else:
+                        try:
+                            network_obj = cnw.find_network_by_label(network)
+                        except (pyrax.exceptions.NetworkNotFound,
+                                pyrax.exceptions.NetworkLabelNotUnique):
+                            module.fail_json(msg='No matching network found '
+                                                 '(%s)' % network)
+                        else:
+                            nics.extend(cnw.get_server_networks(network_obj))
+                else:
+                    nics.extend(cnw.get_server_networks(network))
+
+            for nic in nics:
+                nic.update(uuid=nic['net-id'])
+                del nic['net-id']
+
+        # Handle the file contents
+        personality = []
+        if files:
+            for rpath in files.keys():
+                lpath = os.path.expanduser(files[rpath])
+                try:
+                    f = open(lpath, 'r')
+                    personality.append({
+                        'path': rpath,
+                        'contents': f.read()
+                    })
+                    f.close()
+                except Exception, e:
+                    module.fail_json(msg='Failed to load %s' % lpath)
+
+        lbs = []
+        if loadbalancers:
+            for lb in loadbalancers:
+                lb_id = lb.get('id')
+                port = lb.get('port')
+                if not lb_id or not port:
+                    continue
+                lbs.append((lb_id, port))
+
+        try:
+            sg = au.find(name=name)
+        except pyrax.exceptions.NoUniqueMatch, e:
+            module.fail_json(msg='%s' % e.message)
+        except pyrax.exceptions.NotFound:
+            try:
+                sg = au.create(name, cooldown=cooldown,
+                               min_entities=min_entities,
+                               max_entities=max_entities,
+                               launch_config_type='launch_server',
+                               server_name=server_name, image=image,
+                               flavor=flavor, disk_config=disk_config,
+                               metadata=meta, personality=files,
+                               networks=nics, load_balancers=lbs,
+                               key_name=key_name)
+                changed = True
+            except Exception, e:
+                module.fail_json(msg='%s' % e.message)
+
+        if not changed:
+            # Scaling Group Updates
+            group_args = {}
+            if cooldown != sg.cooldown:
+                group_args['cooldown'] = cooldown
+
+            if min_entities != sg.min_entities:
+                group_args['min_entities'] = min_entities
+
+            if max_entities != sg.max_entities:
+                group_args['max_entities'] = max_entities
+
+            if group_args:
+                changed = True
+                sg.update(**group_args)
+
+            # Launch Configuration Updates
+            lc = sg.get_launch_config()
+            lc_args = {}
+            if server_name != lc.get('name'):
+                lc_args['name'] = server_name
+
+            if image != lc.get('image'):
+                lc_args['image'] = image
+
+            if flavor != lc.get('flavor'):
+                lc_args['flavor'] = flavor
+
+            if disk_config != lc.get('disk_config'):
+                lc_args['disk_config'] = disk_config
+
+            if meta != lc.get('metadata'):
+                lc_args['metadata'] = meta
+
+            if files != lc.get('personality'):
+                lc_args['personality'] = files
+
+            if nics != lc.get('networks'):
+                lc_args['networks'] = nics
+
+            if lbs != lc.get('load_balancers'):
+                # Work around for https://github.com/rackspace/pyrax/pull/393
+                lc_args['load_balancers'] = sg.manager._resolve_lbs(lbs)
+
+            if key_name != lc.get('key_name'):
+                lc_args['key_name'] = key_name
+
+            if lc_args:
+                # Work around for https://github.com/rackspace/pyrax/pull/389
+                if 'flavor' not in lc_args:
+                    lc_args['flavor'] = lc.get('flavor')
+                changed = True
+                sg.update_launch_config(**lc_args)
+
+            sg.get()
+
+        module.exit_json(changed=changed, autoscale_group=asg_to_dict(sg))
+
+    else:
+        try:
+            sg = au.find(name=name)
+            sg.delete()
+            changed = True
+        except pyrax.exceptions.NotFound, e:
+            sg = {}
+        except Exception, e:
+            module.fail_json(msg='%s' % e.message)
+
+        module.exit_json(changed=changed, autoscale_group=asg_to_dict(sg))
+
+
+def main():
+    argument_spec = rax_argument_spec()
+    argument_spec.update(
+        dict(
+            cooldown=dict(type='int', default=300),
+            disk_config=dict(choices=['auto', 'manual']),
+            files=dict(type='list', default=[]),
+            flavor=dict(required=True),
+            image=dict(required=True),
+            key_name=dict(),
+            loadbalancers=dict(type='list'),
+            meta=dict(type='dict', default={}),
+            min_entities=dict(type='int', required=True),
+            max_entities=dict(type='int', required=True),
+            name=dict(required=True),
+            networks=dict(type='list', default=['public', 'private']),
+            server_name=dict(required=True),
+            state=dict(default='present', choices=['present', 'absent']),
+        )
+    )
+
+    module = AnsibleModule(
+        argument_spec=argument_spec,
+        required_together=rax_required_together(),
+    )
+
+    if not HAS_PYRAX:
+        module.fail_json(msg='pyrax is required for this module')
+
+    cooldown = module.params.get('cooldown')
+    disk_config = module.params.get('disk_config')
+    if disk_config:
+        disk_config = disk_config.upper()
+    files = module.params.get('files')
+    flavor = module.params.get('flavor')
+    image = module.params.get('image')
+    key_name = module.params.get('key_name')
+    loadbalancers = module.params.get('loadbalancers')
+    meta = module.params.get('meta')
+    min_entities = module.params.get('min_entities')
+    max_entities = module.params.get('max_entities')
+    name = module.params.get('name')
+    networks = module.params.get('networks')
+    server_name = module.params.get('server_name')
+    state = module.params.get('state')
+
+    if not 0 <= min_entities <= 1000 or not 0 <= max_entities <= 1000:
+        module.fail_json(msg='min_entities and max_entities must be an '
+                             'integer between 0 and 1000')
+
+    if not 0 <= cooldown <= 86400:
+        module.fail_json(msg='cooldown must be an integer between 0 and 86400')
+
+    setup_rax_module(module, pyrax)
+
+    rax_asg(module, cooldown=cooldown, disk_config=disk_config,
+            files=files, flavor=flavor, image=image, meta=meta,
+            key_name=key_name, loadbalancers=loadbalancers,
+            min_entities=min_entities, max_entities=max_entities,
+            name=name, networks=networks, server_name=server_name,
+            state=state)
+
+
+# import module snippets
+from ansible.module_utils.basic import *
+from ansible.module_utils.rax import *
+
+# invoke the module
+main()

library/cloud/rax_scaling_policy

@@ -0,0 +1,298 @@
+#!/usr/bin/python
+# 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/>.
+
+# This is a DOCUMENTATION stub specific to this module, it extends
+# a documentation fragment located in ansible.utils.module_docs_fragments
+DOCUMENTATION = '''
+---
+module: rax_scaling_policy
+short_description: Manipulate Rackspace Cloud Autoscale Scaling Policy
+description:
+    - Manipulate Rackspace Cloud Autoscale Scaling Policy
+version_added: 1.7
+options:
+  at:
+    description:
+      - The UTC time when this policy will be executed. The time must be
+        formatted according to C(yyyy-MM-dd'T'HH:mm:ss.SSS) such as
+        C(2013-05-19T08:07:08Z)
+  change:
+    description:
+      - The change, either as a number of servers or as a percentage, to make
+        in the scaling group. If this is a percentage, you must set
+        I(is_percent) to C(true) also.
+  cron:
+    description:
+      - The time when the policy will be executed, as a cron entry. For
+        example, if this is parameter is set to C(1 0 * * *)
+  cooldown:
+    description:
+      - The period of time, in seconds, that must pass before any scaling can
+        occur after the previous scaling. Must be an integer between 0 and
+        86400 (24 hrs).
+  desired_capacity:
+    description:
+      - The desired server capacity of the scaling the group; that is, how
+        many servers should be in the scaling group.
+  is_percent:
+    description:
+      - Whether the value in I(change) is a percent value
+    default: false
+  name:
+    description:
+      - Name to give the policy
+    required: true
+  policy_type:
+    description:
+      - The type of policy that will be executed for the current release.
+    choices:
+      - webhook
+      - schedule
+    required: true
+  scaling_group:
+    description:
+      - Name of the scaling group that this policy will be added to
+    required: true
+  state:
+    description:
+      - Indicate desired state of the resource
+    choices:
+      - present
+      - absent
+    default: present
+author: Matt Martz
+extends_documentation_fragment: rackspace
+'''
+
+EXAMPLES = '''
+---
+- hosts: localhost
+  gather_facts: false
+  connection: local
+  tasks:
+    - rax_scaling_policy:
+        credentials: ~/.raxpub
+        region: ORD
+        at: '2013-05-19T08:07:08Z'
+        change: 25
+        cooldown: 300
+        is_percent: true
+        name: ASG Test Policy - at
+        policy_type: schedule
+        scaling_group: ASG Test
+      register: asps_at
+
+    - rax_scaling_policy:
+        credentials: ~/.raxpub
+        region: ORD
+        cron: '1 0 * * *'
+        change: 25
+        cooldown: 300
+        is_percent: true
+        name: ASG Test Policy - cron
+        policy_type: schedule
+        scaling_group: ASG Test
+      register: asp_cron
+
+    - rax_scaling_policy:
+        credentials: ~/.raxpub
+        region: ORD
+        cooldown: 300
+        desired_capacity: 5
+        name: ASG Test Policy - webhook
+        policy_type: webhook
+        scaling_group: ASG Test
+      register: asp_webhook
+'''
+
+from uuid import UUID
+
+try:
+    import pyrax
+    HAS_PYRAX = True
+except ImportError:
+    HAS_PYRAX = False
+
+NON_CALLABLES = (basestring, bool, dict, int, list, type(None))
+PUBLIC_NET_ID = "00000000-0000-0000-0000-000000000000"
+SERVICE_NET_ID = "11111111-1111-1111-1111-111111111111"
+
+
+def to_dict(obj):
+    instance = {}
+    for key in dir(obj):
+        value = getattr(obj, key)
+        if (isinstance(value, NON_CALLABLES) and not key.startswith('_')):
+            instance[key] = value
+    return instance
+
+
+def rax_asp(module, at=None, change=0, cron=None, cooldown=300,
+            desired_capacity=0, is_percent=False, name=None,
+            policy_type=None, scaling_group=None, state='present'):
+    changed = False
+
+    au = pyrax.autoscale
+    if not au:
+        module.fail_json(msg='Failed to instantiate client. This '
+                             'typically indicates an invalid region or an '
+                             'incorrectly capitalized region name.')
+
+    try:
+        UUID(scaling_group)
+    except ValueError:
+        try:
+            sg = au.find(name=scaling_group)
+        except Exception, e:
+            module.fail_json(msg='%s' % e.message)
+    else:
+        try:
+            sg = au.get(scaling_group)
+        except Exception, e:
+            module.fail_json(msg='%s' % e.message)
+
+    if state == 'present':
+        policies = filter(lambda p: name == p.name, sg.list_policies())
+        if len(policies) > 1:
+            module.fail_json(msg='No unique policy match found by name')
+        if at:
+            args = dict(at=at)
+        elif cron:
+            args = dict(cron=cron)
+        else:
+            args = None
+
+        if not policies:
+            try:
+                policy = sg.add_policy(name, policy_type=policy_type,
+                                       cooldown=cooldown, change=change,
+                                       is_percent=is_percent,
+                                       desired_capacity=desired_capacity,
+                                       args=args)
+                changed = True
+            except Exception, e:
+                module.fail_json(msg='%s' % e.message)
+
+        else:
+            policy = policies[0]
+            kwargs = {}
+            if policy_type != policy.type:
+                kwargs['policy_type'] = policy_type
+
+            if cooldown != policy.cooldown:
+                kwargs['cooldown'] = cooldown
+
+            if hasattr(policy, 'change') and change != policy.change:
+                kwargs['change'] = change
+
+            if hasattr(policy, 'changePercent') and is_percent is False:
+                kwargs['change'] = change
+                kwargs['is_percent'] = False
+            elif hasattr(policy, 'change') and is_percent is True:
+                kwargs['change'] = change
+                kwargs['is_percent'] = True
+
+            if hasattr(policy, 'desiredCapacity') and change:
+                kwargs['change'] = change
+            elif ((hasattr(policy, 'change') or
+                    hasattr(policy, 'changePercent')) and desired_capacity):
+                kwargs['desired_capacity'] = desired_capacity
+
+            if hasattr(policy, 'args') and args != policy.args:
+                kwargs['args'] = args
+
+            if kwargs:
+                policy.update(**kwargs)
+                changed = True
+
+        policy.get()
+
+        module.exit_json(changed=changed, autoscale_policy=to_dict(policy))
+
+    else:
+        try:
+            policies = filter(lambda p: name == p.name, sg.list_policies())
+            if len(policies) > 1:
+                module.fail_json(msg='No unique policy match found by name')
+            elif not policies:
+                policy = {}
+            else:
+                policy.delete()
+                changed = True
+        except Exception, e:
+            module.fail_json(msg='%s' % e.message)
+
+        module.exit_json(changed=changed, autoscale_policy=to_dict(policy))
+
+
+def main():
+    argument_spec = rax_argument_spec()
+    argument_spec.update(
+        dict(
+            at=dict(),
+            change=dict(type='int'),
+            cron=dict(),
+            cooldown=dict(type='int', default=300),
+            desired_capacity=dict(type='int'),
+            is_percent=dict(type='bool', default=False),
+            name=dict(required=True),
+            policy_type=dict(required=True, choices=['webhook', 'schedule']),
+            scaling_group=dict(required=True),
+            state=dict(default='present', choices=['present', 'absent']),
+        )
+    )
+
+    module = AnsibleModule(
+        argument_spec=argument_spec,
+        required_together=rax_required_together(),
+        mutually_exclusive=[
+            ['cron', 'at'],
+            ['change', 'desired_capacity'],
+        ]
+    )
+
+    if not HAS_PYRAX:
+        module.fail_json(msg='pyrax is required for this module')
+
+    at = module.params.get('at')
+    change = module.params.get('change')
+    cron = module.params.get('cron')
+    cooldown = module.params.get('cooldown')
+    desired_capacity = module.params.get('desired_capacity')
+    is_percent = module.params.get('is_percent')
+    name = module.params.get('name')
+    policy_type = module.params.get('policy_type')
+    scaling_group = module.params.get('scaling_group')
+    state = module.params.get('state')
+
+    if (at or cron) and policy_type == 'webhook':
+        module.fail_json(msg='policy_type=schedule is required for a time '
+                             'based policy')
+
+    setup_rax_module(module, pyrax)
+
+    rax_asp(module, at=at, change=change, cron=cron, cooldown=cooldown,
+            desired_capacity=desired_capacity, is_percent=is_percent,
+            name=name, policy_type=policy_type, scaling_group=scaling_group,
+            state=state)
+
+
+# import module snippets
+from ansible.module_utils.basic import *
+from ansible.module_utils.rax import *
+
+# invoke the module
+main()