| # Copyright (c) 2009-2010 Reza Lotun http://reza.lotun.name/ |
| # Copyright (c) 2011 Jann Kleen |
| # |
| # Permission is hereby granted, free of charge, to any person obtaining a |
| # copy of this software and associated documentation files (the |
| # "Software"), to deal in the Software without restriction, including |
| # without limitation the rights to use, copy, modify, merge, publish, dis- |
| # tribute, sublicense, and/or sell copies of the Software, and to permit |
| # persons to whom the Software is furnished to do so, subject to the fol- |
| # lowing conditions: |
| # |
| # The above copyright notice and this permission notice shall be included |
| # in all copies or substantial portions of the Software. |
| # |
| # THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS |
| # OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABIL- |
| # ITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT |
| # SHALL THE AUTHOR BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, |
| # WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, |
| # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS |
| # IN THE SOFTWARE. |
| |
| from boto.resultset import ResultSet |
| from boto.ec2.elb.listelement import ListElement |
| |
| class Alarm(object): |
| def __init__(self, connection=None): |
| self.connection = connection |
| self.name = None |
| self.alarm_arn = None |
| |
| def __repr__(self): |
| return 'Alarm:%s' % self.name |
| |
| def startElement(self, name, attrs, connection): |
| return None |
| |
| def endElement(self, name, value, connection): |
| if name == 'AlarmName': |
| self.name = value |
| elif name == 'AlarmARN': |
| self.alarm_arn = value |
| else: |
| setattr(self, name, value) |
| |
| |
| class AdjustmentType(object): |
| def __init__(self, connection=None): |
| self.connection = connection |
| self.adjustment_types = ListElement([]) |
| |
| def __repr__(self): |
| return 'AdjustmentType:%s' % self.adjustment_types |
| |
| def startElement(self, name, attrs, connection): |
| if name == 'AdjustmentType': |
| return self.adjustment_types |
| |
| def endElement(self, name, value, connection): |
| return |
| |
| |
| class MetricCollectionTypes(object): |
| class BaseType(object): |
| arg = '' |
| def __init__(self, connection): |
| self.connection = connection |
| self.val = None |
| def __repr__(self): |
| return '%s:%s' % (self.arg, self.val) |
| def startElement(self, name, attrs, connection): |
| return |
| def endElement(self, name, value, connection): |
| if name == self.arg: |
| self.val = value |
| class Metric(BaseType): |
| arg = 'Metric' |
| class Granularity(BaseType): |
| arg = 'Granularity' |
| |
| def __init__(self, connection=None): |
| self.connection = connection |
| self.metrics = [] |
| self.granularities = [] |
| |
| def __repr__(self): |
| return 'MetricCollectionTypes:<%s, %s>' % (self.metrics, self.granularities) |
| |
| def startElement(self, name, attrs, connection): |
| if name == 'Granularities': |
| self.granularities = ResultSet([('member', self.Granularity)]) |
| return self.granularities |
| elif name == 'Metrics': |
| self.metrics = ResultSet([('member', self.Metric)]) |
| return self.metrics |
| |
| def endElement(self, name, value, connection): |
| return |
| |
| |
| class ScalingPolicy(object): |
| def __init__(self, connection=None, **kwargs): |
| """ |
| Scaling Policy |
| |
| :type name: str |
| :param name: Name of scaling policy. |
| |
| :type adjustment_type: str |
| :param adjustment_type: Specifies the type of adjustment. Valid values are `ChangeInCapacity`, `ExactCapacity` and `PercentChangeInCapacity`. |
| |
| :type as_name: str or int |
| :param as_name: Name or ARN of the Auto Scaling Group. |
| |
| :type scaling_adjustment: int |
| :param scaling_adjustment: Value of adjustment (type specified in `adjustment_type`). |
| |
| :type cooldown: int |
| :param cooldown: Time (in seconds) before Alarm related Scaling Activities can start after the previous Scaling Activity ends. |
| |
| """ |
| self.name = kwargs.get('name', None) |
| self.adjustment_type = kwargs.get('adjustment_type', None) |
| self.as_name = kwargs.get('as_name', None) |
| self.scaling_adjustment = kwargs.get('scaling_adjustment', None) |
| self.cooldown = kwargs.get('cooldown', None) |
| self.connection = connection |
| |
| def __repr__(self): |
| return 'ScalingPolicy(%s group:%s adjustment:%s)' % (self.name, |
| self.as_name, |
| self.adjustment_type) |
| |
| def startElement(self, name, attrs, connection): |
| if name == 'Alarms': |
| self.alarms = ResultSet([('member', Alarm)]) |
| return self.alarms |
| |
| def endElement(self, name, value, connection): |
| if name == 'PolicyName': |
| self.name = value |
| elif name == 'AutoScalingGroupName': |
| self.as_name = value |
| elif name == 'PolicyARN': |
| self.policy_arn = value |
| elif name == 'ScalingAdjustment': |
| self.scaling_adjustment = int(value) |
| elif name == 'Cooldown': |
| self.cooldown = int(value) |
| elif name == 'AdjustmentType': |
| self.adjustment_type = value |
| |
| def delete(self): |
| return self.connection.delete_policy(self.name, self.as_name) |
| |