utils/env_prepare/quota_prepare.py

   1 #!/usr/bin/env python
   2 ##############################################################################
   3 # Copyright (c) 2016 Huawei Technologies Co.,Ltd and others.
   4 #
   5 # All rights reserved. This program and the accompanying materials
   6 # are made available under the terms of the Apache License, Version 2.0
   7 # which accompanies this distribution, and is available at
   8 # http://www.apache.org/licenses/LICENSE-2.0
   9 ##############################################################################
  10
  11
  12 import os
  13 import commands
  14 import utils.logger as log
  15 import utils.infra_setup.heat.manager as client_manager
  16
  17 LOG = log.Logger(__name__).getLogger()
  18
  19 neutron_quota = {"subnet": -1,
  20                  "network": -1,
  21                  "floatingip": -1,
  22                  "subnetpool": -1,
  23                  "router": -1,
  24                  "port": -1}
  25
  26 nova_quota = {"ram": -1,
  27               "cores": -1,
  28               "instances": -1,
  29               "key_pairs": -1,
  30               "fixed_ips": -1,
  31               "floating_ips": -1,
  32               "server_groups": -1,
  33               "injected_files": -1,
  34               "metadata_items": -1,
  35               "security_groups": -1,
  36               "security_group_rules": -1,
  37               "server_group_members": -1,
  38               "injected_file_content_bytes": -1,
  39               "injected_file_path_bytes": -1}
  40
  41
  42 def quota_env_prepare():
  43     tenant_name = os.getenv("OS_TENANT_NAME")
  44     cmd = ("openstack project list | grep " +
  45            tenant_name +
  46            " | awk '{print $2}'")
  47
  48     result = commands.getstatusoutput(cmd)
  49     if result[0] == 0:
  50         LOG.info(result[1])
  51     else:
  52         LOG.error("can't get openstack project id")
  53         return 1
  54
  55     openstack_id = result[1]
  56
  57     nova_client = client_manager._get_nova_client()
  58     neutron_client = client_manager._get_neutron_client()
  59
  60     nova_q = nova_client.quotas.get(openstack_id).to_dict()
  61     neutron_q = neutron_client.show_quota(openstack_id)
  62     LOG.info(tenant_name + "tenant nova and neutron quota(previous) :")
  63     LOG.info(nova_q)
  64     LOG.info(neutron_q)
  65
  66     nova_client.quotas.update(openstack_id, **nova_quota)
  67     neutron_client.update_quota(openstack_id,
  68                                 {'quota': neutron_quota})
  69     LOG.info("Quota has been changed!")
  70
  71     nova_q = nova_client.quotas.get(openstack_id).to_dict()
  72     neutron_q = neutron_client.show_quota(openstack_id)
  73     LOG.info(tenant_name + "tenant nova and neutron quota(now) :")
  74     LOG.info(nova_q)
  75     LOG.info(neutron_q)
  76     return 0