151 lines
5.9 KiB
Python
151 lines
5.9 KiB
Python
# Copyright 2016 Rackspace
|
|
#
|
|
# 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 oslo_config import cfg
|
|
import pecan
|
|
from wsme import types as wtypes
|
|
from wsmeext import pecan as wsme_pecan
|
|
|
|
from octavia.api.v2.controllers import base
|
|
from octavia.api.v2.types import quotas as quota_types
|
|
from octavia.common import constants
|
|
from octavia.common import exceptions
|
|
|
|
CONF = cfg.CONF
|
|
CONF.import_group('quotas', 'octavia.common.config')
|
|
|
|
|
|
class QuotasController(base.BaseController):
|
|
|
|
def __init__(self):
|
|
super(QuotasController, self).__init__()
|
|
|
|
@wsme_pecan.wsexpose(quota_types.QuotaResponse, wtypes.text)
|
|
def get(self, project_id):
|
|
"""Get a single project's quota details."""
|
|
context = pecan.request.context.get('octavia_context')
|
|
|
|
# Check that the user is authorized to show this quota
|
|
action = '{rbac_obj}{action}'.format(
|
|
rbac_obj=constants.RBAC_QUOTA, action='get_one')
|
|
target = {'project_id': project_id}
|
|
context.policy.authorize(action, target)
|
|
|
|
db_quotas = self._get_db_quotas(context.session, project_id)
|
|
return self._convert_db_to_type(db_quotas, quota_types.QuotaResponse)
|
|
|
|
@wsme_pecan.wsexpose(quota_types.QuotaAllResponse,
|
|
ignore_extra_args=True)
|
|
def get_all(self, project_id=None):
|
|
"""List all non-default quotas."""
|
|
pcontext = pecan.request.context
|
|
context = pcontext.get('octavia_context')
|
|
|
|
# Check that the user is authorized to list quotas under all projects
|
|
action = '{rbac_obj}{action}'.format(
|
|
rbac_obj=constants.RBAC_QUOTA, action='get_all-global')
|
|
target = {'project_id': project_id}
|
|
if not context.policy.authorize(action, target, do_raise=False):
|
|
# Not a global observer or admin
|
|
if project_id is None:
|
|
project_id = context.project_id
|
|
|
|
# Check if user is authorized to list quota under this project
|
|
action = '{rbac_obj}{action}'.format(
|
|
rbac_obj=constants.RBAC_QUOTA, action='get_all')
|
|
target = {'project_id': project_id}
|
|
context.policy.authorize(action, target)
|
|
|
|
if project_id is None:
|
|
query_filter = {}
|
|
else:
|
|
query_filter = {'project_id': project_id}
|
|
|
|
db_quotas, links = self.repositories.quotas.get_all(
|
|
context.session,
|
|
pagination_helper=pcontext.get(constants.PAGINATION_HELPER),
|
|
**query_filter)
|
|
quotas = quota_types.QuotaAllResponse.from_data_model(db_quotas)
|
|
quotas.quotas_links = links
|
|
return quotas
|
|
|
|
@wsme_pecan.wsexpose(quota_types.QuotaResponse, wtypes.text,
|
|
body=quota_types.QuotaPUT, status_code=202)
|
|
def put(self, project_id, quotas):
|
|
"""Update any or all quotas for a project."""
|
|
context = pecan.request.context.get('octavia_context')
|
|
|
|
if not project_id:
|
|
raise exceptions.MissingAPIProjectID()
|
|
|
|
# Check that the user is authorized to update this quota
|
|
action = '{rbac_obj}{action}'.format(
|
|
rbac_obj=constants.RBAC_QUOTA, action='put')
|
|
target = {'project_id': project_id}
|
|
context.policy.authorize(action, target)
|
|
|
|
quotas_dict = quotas.to_dict()
|
|
self.repositories.quotas.update(context.session, project_id,
|
|
**quotas_dict)
|
|
db_quotas = self._get_db_quotas(context.session, project_id)
|
|
return self._convert_db_to_type(db_quotas, quota_types.QuotaResponse)
|
|
|
|
@wsme_pecan.wsexpose(None, wtypes.text, status_code=202)
|
|
def delete(self, project_id):
|
|
"""Reset a project's quotas to the default values."""
|
|
context = pecan.request.context.get('octavia_context')
|
|
|
|
if not project_id:
|
|
raise exceptions.MissingAPIProjectID()
|
|
|
|
# Check that the user is authorized to delete this quota
|
|
action = '{rbac_obj}{action}'.format(
|
|
rbac_obj=constants.RBAC_QUOTA, action='delete')
|
|
target = {'project_id': project_id}
|
|
context.policy.authorize(action, target)
|
|
|
|
self.repositories.quotas.delete(context.session, project_id)
|
|
db_quotas = self._get_db_quotas(context.session, project_id)
|
|
return self._convert_db_to_type(db_quotas, quota_types.QuotaResponse)
|
|
|
|
@pecan.expose()
|
|
def _lookup(self, project_id, *remainder):
|
|
"""Overridden pecan _lookup method for routing default endpoint."""
|
|
if project_id and len(remainder) and remainder[0] == 'default':
|
|
return QuotasDefaultController(project_id), ''
|
|
|
|
|
|
class QuotasDefaultController(base.BaseController):
|
|
|
|
def __init__(self, project_id):
|
|
super(QuotasDefaultController, self).__init__()
|
|
self.project_id = project_id
|
|
|
|
@wsme_pecan.wsexpose(quota_types.QuotaResponse, wtypes.text)
|
|
def get(self):
|
|
"""Get a project's default quota details."""
|
|
context = pecan.request.context.get('octavia_context')
|
|
|
|
if not self.project_id:
|
|
raise exceptions.MissingAPIProjectID()
|
|
|
|
# Check that the user is authorized to see quota defaults
|
|
action = '{rbac_obj}{action}'.format(
|
|
rbac_obj=constants.RBAC_QUOTA, action='get_defaults')
|
|
target = {'project_id': self.project_id}
|
|
context.policy.authorize(action, target)
|
|
|
|
quotas = self._get_default_quotas(self.project_id)
|
|
return self._convert_db_to_type(quotas, quota_types.QuotaResponse)
|