67aaf5f5ab
Some few useful bug fixes such as replacement of password without "\n", and issues with "flavor.name" vs "instance_type".
98 lines
3.3 KiB
Python
98 lines
3.3 KiB
Python
#!/usr/bin/env python
|
|
|
|
import os, sys
|
|
|
|
try:
|
|
from artifice import interface
|
|
except ImportError:
|
|
loc, fn = os.path.split(__file__)
|
|
print loc
|
|
here = os.path.abspath(os.path.join(loc +"/../"))
|
|
sys.path.insert(0, here)
|
|
# # Are we potentially in a virtualenv? Add that in.
|
|
# if os.path.exists( os.path.join(here, "lib/python2.7" ) ):
|
|
# sys.path.insert(1, os.path.join(here, "lib/python2.7"))
|
|
from artifice import interface
|
|
|
|
import datetime
|
|
import yaml
|
|
|
|
date_format = "%Y-%m-%dT%H:%M:%S"
|
|
other_date_format = "%Y-%m-%dT%H:%M:%S.%f"
|
|
date_fmt = "%Y-%m-%d"
|
|
|
|
def date_fmt_fnc(val):
|
|
return datetime.datetime.strptime(val, date_fmt)
|
|
|
|
if __name__ == '__main__':
|
|
import argparse
|
|
parser = argparse.ArgumentParser()
|
|
# Takes names to display.
|
|
# none means display them all.
|
|
parser.add_argument("-t", "--tenant", dest="tenants", help='Tenant to display', action="append", default=[])
|
|
|
|
# Add some sections to show data from.
|
|
# Empty is display all
|
|
parser.add_argument("-s", "--section", dest="sections", help="Sections to display", action="append")
|
|
|
|
|
|
# Ranging
|
|
# We want to get stuff from, to.
|
|
|
|
parser.add_argument(
|
|
"--from",
|
|
dest="start",
|
|
help="When to start our range, date format %s",
|
|
type=date_fmt_fnc,
|
|
default=datetime.datetime.now() - datetime.timedelta(days=31)
|
|
)
|
|
parser.add_argument("--to", dest="end", help="When to end our date range. Defaults to yesterday.",
|
|
type=date_fmt_fnc, default=datetime.datetime.now() - datetime.timedelta(days=1) )
|
|
|
|
parser.add_argument("-c", "--config", dest="config", help="Config file", default="/opt/stack/artifice/etc/artifice/conf.yaml")
|
|
|
|
args = parser.parse_args()
|
|
print "\n # ----------------- usage summary for: ----------------- # "
|
|
print "Range: %s -> %s" % (args.start, args.end)
|
|
try:
|
|
conf = yaml.load(open(args.config).read())
|
|
except IOError:
|
|
# Whoops
|
|
print "couldn't load %s " % args.config
|
|
sys.exit(1)
|
|
|
|
fh = open(conf["database"]["password_path"])
|
|
password = fh.read()
|
|
fh.close()
|
|
# Make ourselves a nice interaction object
|
|
conf["database"]["password"] = password
|
|
instance = interface.Artifice(conf)
|
|
tenants = args.tenants
|
|
if not args.tenants:
|
|
# only parse this list of tenants
|
|
tenants = instance.tenants
|
|
|
|
for tenant_name in tenants:
|
|
# artifact = n.tenant(tenant_name).section(section).usage(args.start, args.end)
|
|
# data should now be an artifact-like construct.
|
|
# Data also knows about Samples, where as an Artifact doesn't.
|
|
|
|
# An artifact knows its section
|
|
tenant = instance.tenant(tenant_name)
|
|
# Makes a new invoice up for this tenant.
|
|
invoice = tenant.invoice(args.start, args.end)
|
|
print "\n# ------------------------ Tenant: %s ------------------------ #" % tenant.name
|
|
|
|
|
|
# usage = tenant.usage(start, end)
|
|
usage = tenant.usage(args.start, args.end)
|
|
# A Usage set is the entirety of time for this Tenant.
|
|
# It's not time-limited at all.
|
|
|
|
invoice.bill(usage.vms)
|
|
invoice.bill(usage.volumes)
|
|
# invoice.bill(usage.objects)
|
|
|
|
print "Total invoice value: %s" % invoice.total()
|
|
print "# --------------------- End of Tenant: %s --------------------- #" % tenant.name
|