Source code for designateclient.v2.cli.quotas

# Copyright 2014 Hewlett-Packard Development Company, L.P.
#
# Author: Endre Karlson <endre.karlson@hp.com>
#
# 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.
import itertools
import logging

from cliff import command
from cliff import show

from designateclient.v2.cli import common

LOG = logging.getLogger(__name__)


DNS_QUOTAS = {
    "api_export_size": "api-export-size",
    "recordset_records": "recordset-records",
    "zone_records": "zone-records",
    "zone_recordsets": "zone-recordsets",
    "zones": "zones"
}


[docs] class ListQuotasCommand(show.ShowOne): """List quotas""" # columns = ['resource', 'hard_limit']
[docs] def get_parser(self, prog_name): parser = super().get_parser(prog_name) common.add_all_common_options(parser) parser.add_argument( '--project-id', help='Project ID Default: current project') return parser
[docs] def take_action(self, parsed_args): client = self.app.client_manager.dns common.set_all_common_headers(client, parsed_args) proj_id = client.session.get_project_id() if parsed_args.project_id and parsed_args.project_id != proj_id: proj_id = parsed_args.project_id common.set_all_projects(client, True) data = client.quotas.list(proj_id) return zip(*sorted(data.items()))
[docs] class SetQuotasCommand(show.ShowOne): """Set quotas""" def _build_options_list(self): return itertools.chain(DNS_QUOTAS.items())
[docs] def get_parser(self, prog_name): parser = super().get_parser(prog_name) common.add_all_common_options(parser) parser.add_argument('--project-id', help='Project ID') for k, v in self._build_options_list(): parser.add_argument( f'--{v}', metavar=f'<{v}>', dest=k, type=int, help=f'New value for the {v} quota', ) return parser
[docs] def take_action(self, parsed_args): client = self.app.client_manager.dns common.set_all_common_headers(client, parsed_args) quotas = {} for k, v in DNS_QUOTAS.items(): value = getattr(parsed_args, k, None) if value is not None: quotas[k] = value proj_id = parsed_args.project_id or client.session.get_project_id() if parsed_args.project_id != client.session.get_project_id(): common.set_all_projects(client, True) updated = client.quotas.update(proj_id, quotas) return zip(*sorted(updated.items()))
[docs] class ResetQuotasCommand(command.Command): """Reset quotas"""
[docs] def get_parser(self, prog_name): parser = super().get_parser(prog_name) common.add_all_common_options(parser) parser.add_argument('--project-id', help='Project ID') return parser
[docs] def take_action(self, parsed_args): client = self.app.client_manager.dns common.set_all_common_headers(client, parsed_args) proj_id = parsed_args.project_id or client.session.get_project_id() if parsed_args.project_id != client.session.get_project_id(): common.set_all_projects(client, True) client.quotas.reset(proj_id) LOG.info('Quota for project %s was reset', parsed_args.project_id)