# designate.quota.impl_storage

# Source code for designate.quota.impl_storage

    # Copyright 2013 Hewlett-Packard Development Company, L.P.
    #
    # Author: Kiall Mac Innes <kiall@hpe.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.
    from oslo_log import log as logging

    from designate import exceptions
    from designate import objects
    from designate.quota import base
    from designate import storage
    from designate.storage import transaction

    LOG = logging.getLogger(__name__)

    class StorageQuota(base.Quota):
        __plugin_name__ = 'storage'

        def __init__(self):
            super().__init__()
            self.storage = storage.get_storage()

        def _get_quotas(self, context, tenant_id):
            """Internal Get Quotas used by get_quotas"""
            quotas = self.storage.find_quotas(context, {
                'tenant_id': tenant_id,
            })

            return {q['resource']: q['hard_limit'] for q in quotas}

        def get_quota(self, context, tenant_id, resource):
            """Get Quota"""
            context = context.deepcopy()
            context.all_tenants = True

            quota = self.storage.find_quota(context, {
                'tenant_id': tenant_id,
                'resource': resource,
            })

            return {
                resource: quota['hard_limit']
            }

        @transaction
        def set_quota(self, context, tenant_id, resource, hard_limit):
            """Set Quota"""
            context = context.deepcopy()
            context.all_tenants = True

            if resource not in list(self.get_default_quotas(context).keys()):
                raise exceptions.QuotaResourceUnknown(
                    f'{resource} is not a valid quota resource'
                )

            try:
                self._create_quota(context, tenant_id, resource, hard_limit)
            except exceptions.Duplicate:
                quota = self.storage.find_quota(context, {
                    'tenant_id': tenant_id,
                    'resource': resource,
                })
                self._update_quota(context, quota, hard_limit)

            return {resource: hard_limit}

        def _create_quota(self, context, project_id, resource, hard_limit):
            quota = objects.Quota(
                tenant_id=project_id, resource=resource, hard_limit=hard_limit
            )
            self.storage.create_quota(context, quota)

        def _update_quota(self, context, quota, hard_limit):
            quota.hard_limit = hard_limit
            self.storage.update_quota(context, quota)

        @transaction
        def reset_quotas(self, context, tenant_id):
            """Reset Quotas"""
            context = context.deepcopy()
            context.all_tenants = True

            quotas = self.storage.find_quotas(context, {
                'tenant_id': tenant_id,
            })

            for quota in quotas:
                self.storage.delete_quota(context, quota['id'])

this page last updated: 2026-08-19 02:25:39

Except where otherwise noted, this document is licensed under Creative Commons Attribution 3.0 License. See all OpenStack Legal Documents.

found an error? report a bug