Support path to tagging resources during start and removing tags during stop

update plugin yaml for base ec2 type with tags

Author: earthmant

CHANGELOG.txt

@@ -1,3 +1,4 @@
+2.7.0: Add Tag property/runtime property/operation input for EC2 resource node templates (without the previously only available Tagging node type): Customer Gateway, Volumes, Elastic IPs, Elastic Network Interfaces, Instances, Internet Gateways, Keypairs, NAT Gateways, Network ACLs, Route Tables, Security Groups, Subnets, VPCs, VPC Peerings, and VPN Gateways.
 2.6.0:
   - Add Modify VPC Attribute
   - Add Modify Subnet Attribute

cloudify_awssdk/common/decorators.py

@@ -97,9 +97,9 @@ def wrapper_outer(function):
         def wrapper_inner(**kwargs):
             '''Inner, worker function'''
             ctx = kwargs['ctx']
+            _, _, _, operation_name = ctx.operation.name.split('.')
             props = ctx.node.properties
             runtime_instance_properties = ctx.instance.runtime_properties
-
             # Override the resource ID if needed
             resource_id = kwargs.get(EXT_RES_ID)
             if resource_id and not \
@@ -178,11 +178,11 @@ def wrapper_inner(**kwargs):
                     kwargs['resource_config'] =\
                         runtime_instance_properties['resource_config']
                     resource_config = kwargs['resource_config']
-
+            resource_id = utils.get_resource_id(
+                node=ctx.node,
+                instance=ctx.instance)
             # Check if using external
             if ctx.node.properties.get('use_external_resource', False):
-                resource_id = utils.get_resource_id(
-                    node=ctx.node, instance=ctx.instance)
                 ctx.logger.info('%s ID# "%s" is user-provided.'
                                 % (resource_type, resource_id))
                 if not kwargs.get('force_operation', False):
@@ -192,7 +192,6 @@ def wrapper_inner(**kwargs):
                     # Set "resource_config" and "EXT_RES_ID"
                     ctx.instance.runtime_properties[
                         'resource_config'] = resource_config
-                    _, _, _, operation_name = ctx.operation.name.split('.')
                     ctx.instance.runtime_properties[EXT_RES_ID] = resource_id
                     if operation_name not in ['delete', 'create'] and \
                             not kwargs['iface'].verify_resource_exists():
@@ -361,3 +360,42 @@ def wrapper(**kwargs):
 
         return func(**kwargs)
     return wrapper
+
+
+def tag_resources(fn):
+    def wrapper(**kwargs):
+        result = fn(**kwargs)
+        ctx = kwargs.get('ctx')
+        iface = kwargs.get('iface')
+        resource_id = utils.get_resource_id(
+            node=ctx.node,
+            instance=ctx.instance)
+        tags = utils.get_tags_list(
+            ctx.node.properties.get('Tags'),
+            ctx.instance.runtime_properties.get('Tags'),
+            kwargs.get('Tags'))
+        if iface and tags and resource_id:
+            iface.tag({
+                'Tags': tags,
+                'Resources': [resource_id]})
+        return result
+    return wrapper
+
+
+def untag_resources(fn):
+    def wrapper(**kwargs):
+        ctx = kwargs.get('ctx')
+        iface = kwargs.get('iface')
+        resource_id = utils.get_resource_id(
+            node=ctx.node,
+            instance=ctx.instance)
+        tags = utils.get_tags_list(
+            ctx.node.properties.get('Tags'),
+            ctx.instance.runtime_properties.get('Tags'),
+            kwargs.get('Tags'))
+        if iface and tags and resource_id:
+            iface.untag({
+                'Tags': tags,
+                'Resources': [resource_id]})
+        return fn(**kwargs)
+    return wrapper

cloudify_awssdk/common/tests/test_base.py

@@ -100,7 +100,7 @@ def get_mock_ctx(self,
                      ctx_operation_name=None):
 
         operation_ctx = {
-            'retry_number': 0
+            'retry_number': 0, 'name': 'cloudify.interfaces.lifecycle.'
         } if not ctx_operation_name else {
             'retry_number': 0, 'name': ctx_operation_name
         }

cloudify_awssdk/common/utils.py

@@ -444,3 +444,14 @@ def generate_swift_access_config(auth_url, username, password):
     # client
     token = resp.headers['X-Auth-Token']
     return endpoint_url, token
+
+
+def get_tags_list(node_prop, runtime_prop, input_prop):
+    tags_list = []
+    if isinstance(node_prop, list):
+        tags_list = node_prop
+    if isinstance(runtime_prop, list):
+        tags_list = list(set(tags_list + runtime_prop))
+    if isinstance(input_prop, list):
+        tags_list = list(set(tags_list + input_prop))
+    return tags_list

cloudify_awssdk/ec2/__init__.py

@@ -50,3 +50,17 @@ def create(self, params):
     def delete(self, params=None):
         """Deletes a resource"""
         raise NotImplementedError()
+
+    def tag(self, params):
+        """Creates a resource"""
+        self.logger.info('Tagging %s.' % params)
+        res = self.client.create_tags(**params)
+        self.logger.debug('Response: %s' % res)
+        return res
+
+    def untag(self, params):
+        """Creates a resource"""
+        self.logger.info('Untagging %s.' % params)
+        res = self.client.delete_tags(**params)
+        self.logger.debug('Response: %s' % res)
+        return res

cloudify_awssdk/ec2/resources/customer_gateway.py

@@ -88,6 +88,7 @@ def prepare(ctx, resource_config, **_):
 @decorators.aws_resource(EC2CustomerGateway, RESOURCE_TYPE)
 @decorators.wait_for_status(status_good=['available'],
                             status_pending=['pending'])
+@decorators.tag_resources
 def create(ctx, iface, resource_config, **_):
     """Creates an AWS EC2 Customer Gateway"""
 
@@ -117,6 +118,7 @@ def create(ctx, iface, resource_config, **_):
                          ignore_properties=True)
 @decorators.wait_for_delete(status_deleted=['deleted'],
                             status_pending=['deleting'])
+@decorators.untag_resources
 def delete(iface, resource_config, **_):
     """Deletes an AWS EC2 Customer Gateway"""
 

cloudify_awssdk/ec2/resources/ebs.py

@@ -152,6 +152,7 @@ def prepare(ctx, resource_config, **_):
 
 @decorators.aws_resource(EC2Volume, RESOURCE_TYPE_VOLUME)
 @decorators.wait_for_status(status_good=[AVAILABLE], status_pending=[CREATING])
+@decorators.tag_resources
 def create(ctx, iface, resource_config, **_):
     """
     Creates an AWS EC2 EBS Volume
@@ -184,6 +185,7 @@ def create(ctx, iface, resource_config, **_):
 
 @decorators.aws_resource(EC2Volume, RESOURCE_TYPE_VOLUME,
                          ignore_properties=True)
+@decorators.untag_resources
 def delete(ctx, iface, resource_config, **_):
     """
     Deletes an AWS EC2 EBS Volume

cloudify_awssdk/ec2/resources/elasticip.py

@@ -105,6 +105,7 @@ def prepare(ctx, resource_config, **_):
 
 
 @decorators.aws_resource(EC2ElasticIP, RESOURCE_TYPE)
+@decorators.tag_resources
 def create(ctx, iface, resource_config, **_):
     """Creates an AWS EC2 ElasticIP"""
 
@@ -125,6 +126,7 @@ def create(ctx, iface, resource_config, **_):
 
 @decorators.aws_resource(EC2ElasticIP, RESOURCE_TYPE,
                          ignore_properties=True)
+@decorators.untag_resources
 def delete(ctx, iface, resource_config, **_):
     """Deletes an AWS EC2 ElasticIP"""
 

cloudify_awssdk/ec2/resources/eni.py

@@ -125,6 +125,7 @@ def prepare(ctx, resource_config, **_):
 
 @decorators.aws_resource(EC2NetworkInterface, RESOURCE_TYPE)
 @decorators.wait_for_status(status_good=['available'])
+@decorators.tag_resources
 def create(ctx, iface, resource_config, **_):
     """Creates an AWS EC2 NetworkInterface"""
 
@@ -178,6 +179,7 @@ def create(ctx, iface, resource_config, **_):
 
 @decorators.aws_resource(EC2NetworkInterface, RESOURCE_TYPE,
                          ignore_properties=True)
+@decorators.untag_resources
 def delete(ctx, iface, resource_config, **_):
     """Deletes an AWS EC2 NetworkInterface"""
 

cloudify_awssdk/ec2/resources/instances.py

@@ -168,6 +168,7 @@ def prepare(ctx, iface, resource_config, **_):
 @decorators.wait_for_status(
     status_good=[RUNNING, PENDING],
     fail_on_missing=False)
+@decorators.tag_resources
 def create(ctx, iface, resource_config, **_):
     '''Creates AWS EC2 Instances'''
 
@@ -350,6 +351,7 @@ def stop(ctx, iface, resource_config, **_):
 @decorators.wait_for_delete(
     status_deleted=[TERMINATED],
     status_pending=[PENDING, STOPPING, SHUTTING_DOWN])
+@decorators.untag_resources
 def delete(iface, resource_config, **_):
     '''Deletes AWS EC2 Instances'''