Exemple #1
0
def main():
    import docopt
    from model import Customer

    opt = docopt.docopt(__doc__)

    tenant_id = opt["--tenant"]
    customer_id = opt["--customer"]
    if customer_id:
        try:
            customer_id = int(customer_id)
            customer = Customer.get_by_id(customer_id)
        except ValueError:
            customer = Customer.get_by_email(customer_id)
        if not customer:
            logbook.warning("Customer {} not found", customer_id)
            return
        else:
            tenant_id = customer.tenant.tenant_id
    start = opt["--start"]
    end = opt["--end"]
    now = arrow.utcnow().date()

    start = arrow.get(start) if start else now - timedelta(minutes=15)
    end = arrow.get(end) if end else now

    metric = opt["--metric"]
    service = opt["--service"]
    if service:
        service_mapping = {data["service"]: name for name, data in conf.fitter.collection.meter_mappings.items()}
        if service not in service_mapping:
            logbook.warning("Service {} not found", service)
            return
        metric = service_mapping[service]

    samples = openstack.get_tenant_usage(tenant_id, metric, start, end, limit=int(opt["--limit"]))
    logbook.info("Found {} samples", len(samples))
    for s in samples:
        pprint(s._info)