Merge pull request boto#887 from graphaelli/add-PutNotificationConfig…

…uration

add PutNotificationConfiguration support

boto/ec2/autoscale/__init__.py

@@ -630,6 +630,29 @@ def execute_policy(self, policy_name, as_group=None, honor_cooldown=None):
             params['HonorCooldown'] = honor_cooldown
         return self.get_status('ExecutePolicy', params)
 
+    def put_notification_configuration(self, autoscale_group, topic, notification_types):
+        """
+        Configures an Auto Scaling group to send notifications when specified events take place.
+
+        :type as_group: str or :class:`boto.ec2.autoscale.group.AutoScalingGroup` object
+        :param as_group: The Auto Scaling group to put notification configuration on.
+
+        :type topic: str
+        :param topic: The Amazon Resource Name (ARN) of the Amazon Simple Notification Service (SNS) topic.
+
+        :type notification_types: list
+        :param notification_types: The type of events that will trigger the notification.
+        """
+
+        name = autoscale_group
+        if isinstance(autoscale_group, AutoScalingGroup):
+            name = autoscale_group.name
+
+        params = {'AutoScalingGroupName': name,
+                  'TopicARN': topic}
+        self.build_list_params(params, notification_types, 'NotificationTypes')
+        return self.get_status('PutNotificationConfiguration', params)
+
     def set_instance_health(self, instance_id, health_status,
                             should_respect_grace_period=True):
         """

boto/ec2/autoscale/group.py

@@ -270,6 +270,12 @@ def get_activities(self, activity_ids=None, max_records=50):
         return self.connection.get_all_activities(self, activity_ids,
                                                   max_records)
 
+    def put_notification_configuration(self, topic, notification_types):
+        """
+        Configures an Auto Scaling group to send notifications when specified events take place.
+        """
+        return self.connection.put_notification_configuration(self, topic, notification_types)
+
     def suspend_processes(self, scaling_processes=None):
         """
         Suspends Auto Scaling processes for an Auto Scaling group.