b2cfa3d465
Implements: https://blueprints.launchpad.net/reddwarf/+spec/rate-limits Change-Id: I525bbdb7e58c100a3fbd29493a06e3aee9417cc9
55 lines
1.6 KiB
Python
55 lines
1.6 KiB
Python
# Copyright (c) 2013 OpenStack, LLC.
|
|
# All Rights Reserved.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
from reddwarfclient import base
|
|
import exceptions
|
|
|
|
RESPONSE_KEY = "limits"
|
|
ABSOLUTE = "absolute"
|
|
RATE = "rate"
|
|
LIMIT = 'limit'
|
|
|
|
|
|
class Limits(base.ManagerWithFind):
|
|
"""
|
|
Manages :class `Limit` resources
|
|
"""
|
|
resource_class = base.Resource
|
|
|
|
def index(self):
|
|
"""
|
|
Retrieve the limits
|
|
"""
|
|
URL = "/limits"
|
|
resp, body = self.api.client.get(URL)
|
|
|
|
if resp is None or resp.status != 200:
|
|
raise exceptions.from_response(resp, body)
|
|
|
|
if not body:
|
|
raise Exception("Call to " + URL + " did not return a body.")
|
|
|
|
absolute_rates = self._get_absolute_rates(body)
|
|
rates = self._get_rates(body)
|
|
return absolute_rates + rates
|
|
|
|
def _get_rates(self, body):
|
|
rates = body[RESPONSE_KEY][RATE][0].get(LIMIT, {})
|
|
return [self.resource_class(self, res) for res in rates]
|
|
|
|
def _get_absolute_rates(self, body):
|
|
absolute_rates = body[RESPONSE_KEY].get(ABSOLUTE, {})
|
|
return [self.resource_class(self, absolute_rates)]
|