Commit 2c82dc58 authored by Giorgos Korfiatis's avatar Giorgos Korfiatis
Browse files

Modify resource limit from the command line

parent 287c45c2
# Copyright 2013 GRNET S.A. All rights reserved.
#
# Redistribution and use in source and binary forms, with or
# without modification, are permitted provided that the following
# conditions are met:
#
# 1. Redistributions of source code must retain the above
# copyright notice, this list of conditions and the following
# disclaimer.
#
# 2. Redistributions in binary form must reproduce the above
# copyright notice, this list of conditions and the following
# disclaimer in the documentation and/or other materials
# provided with the distribution.
#
# THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS
# OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
# PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR
# CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
# USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
# AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
# ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
# POSSIBILITY OF SUCH DAMAGE.
#
# The views and conclusions contained in the software and
# documentation are those of the authors and should not be
# interpreted as representing official policies, either expressed
# or implied, of GRNET S.A.
from optparse import make_option
from django.core.management.base import BaseCommand, CommandError
from astakos.im.models import Resource
from astakos.im.resources import update_resource
class Command(BaseCommand):
args = "<resource name>"
help = "Modify a resource (currently only change the default base quota)"
option_list = BaseCommand.option_list + (
make_option('--limit',
dest='limit',
help="Change default base quota"),
)
def handle(self, *args, **options):
if len(args) < 1:
raise CommandError("Please provide a resource name.")
resource_name = args[0]
limit = options['limit']
if limit is None:
raise CommandError("Use --limit to change default base quota.")
try:
limit = int(limit)
except:
raise CommandError("Limit should be an integer.")
try:
update_resource(resource_name, limit)
except Resource.DoesNotExist:
raise CommandError("Resource %s does not exist." % resource_name)
......@@ -71,3 +71,15 @@ def add_resource(service, resource, uplimit):
qh_add_resource_limit(name, diff)
else:
qh_sync_new_resource(name, uplimit)
def update_resource(name, uplimit):
r = Resource.objects.get_for_update(name=name)
old_uplimit = r.uplimit
r.uplimit = uplimit
r.save()
logger.info("Updated resource %s with limit %s." % (name, uplimit))
diff = uplimit - old_uplimit
if diff != 0:
qh_add_resource_limit(name, diff)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment