77 lines
1.8 KiB
Python
77 lines
1.8 KiB
Python
#!/usr/bin/env python
|
|
|
|
import os
|
|
import requests
|
|
import yaml
|
|
import argparse
|
|
|
|
from novaclient.client import Client as nova_auth
|
|
from keystoneclient.v3.client import Client as keystone_auth
|
|
|
|
def get_credentials(site_id):
|
|
home = os.environ['HOME']
|
|
with open('{}/clouds.yaml'.format(home),'r') as stream:
|
|
creds = yaml.safe_load(stream)['clouds'][site_id]['auth']
|
|
|
|
creds['version'] = '2'
|
|
return creds
|
|
|
|
def get_auth_token(creds):
|
|
keystone_client = keystone_auth(**creds)
|
|
|
|
return keystone_client.auth_token
|
|
|
|
def get_servers(creds):
|
|
nova_client = nova_auth(**creds)
|
|
|
|
print(nova_client.servers.list())
|
|
|
|
def get_cores(creds):
|
|
url="http://placement.example.com:8778"
|
|
|
|
token = get_auth_token(creds)
|
|
|
|
headers = {}
|
|
headers['content-type'] = 'application/json'
|
|
headers['x-auth-token'] = token
|
|
|
|
request = "/resource_providers"
|
|
|
|
r = requests.get("{}/{}".format(url,request),headers=headers)
|
|
|
|
for res in r.json()['resource_providers']:
|
|
uuid = res['uuid']
|
|
hostname = res['name']
|
|
|
|
request = "/resource_providers/{}/inventories".format(uuid)
|
|
|
|
r = requests.get("{}/{}".format(url,request),headers=headers)
|
|
|
|
inventory = r.json()['inventories']
|
|
|
|
cores=0
|
|
|
|
if 'VCPU' in inventory:
|
|
cores+=inventory['VCPU']['total']
|
|
if 'PCPU' in inventory:
|
|
cores+=inventory['PCPU']['total']
|
|
|
|
|
|
print("{}: {}".format(hostname,cores))
|
|
|
|
def _parse_args():
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument('-s', '--site', metavar="<site-id>",
|
|
help="Site to run against", dest="site_id",
|
|
required=True)
|
|
return parser.parse_args()
|
|
|
|
def main(args):
|
|
creds = get_credentials(args.site_id)
|
|
|
|
get_servers(creds)
|
|
get_cores(creds)
|
|
|
|
if __name__ == '__main__':
|
|
main(_parse_args())
|