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                  "security_group": -1}
  26
  27 nova_quota = {"ram": -1,
  28               "cores": -1,
  29               "instances": -1,
  30               "key_pairs": -1,
  31               "fixed_ips": -1,
  32               "floating_ips": -1,
  33               "server_groups": -1,
  34               "injected_files": -1,
  35               "metadata_items": -1,
  36               "security_groups": -1,
  37               "security_group_rules": -1,
  38               "server_group_members": -1,
  39               "injected_file_content_bytes": -1,
  40               "injected_file_path_bytes": -1}
  41
  42
  43 def quota_env_prepare():
  44     tenant_name = os.getenv("OS_TENANT_NAME")
  45     cmd = ("openstack project list | grep " +
  46            tenant_name +
  47            " | awk '{print $2}'")
  48
  49     result = commands.getstatusoutput(cmd)
  50     if result[0] == 0:
  51         LOG.info(result[1])
  52     else:
  53         LOG.error("can't get openstack project id")
  54         return 1
  55
  56     openstack_id = result[1]
  57
  58     nova_client = client_manager._get_nova_client()
  59     neutron_client = client_manager._get_neutron_client()
  60
  61     nova_q = nova_client.quotas.get(openstack_id).to_dict()
  62     neutron_q = neutron_client.show_quota(openstack_id)
  63     LOG.info(tenant_name + "tenant nova and neutron quota(previous) :")
  64     LOG.info(nova_q)
  65     LOG.info(neutron_q)
  66
  67     nova_client.quotas.update(openstack_id, **nova_quota)
  68     neutron_client.update_quota(openstack_id,
  69                                 {'quota': neutron_quota})
  70     LOG.info("Quota has been changed!")
  71
  72     nova_q = nova_client.quotas.get(openstack_id).to_dict()
  73     neutron_q = neutron_client.show_quota(openstack_id)
  74     LOG.info(tenant_name + "tenant nova and neutron quota(now) :")
  75     LOG.info(nova_q)
  76     LOG.info(neutron_q)
  77     return 0