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'])