Re-merging code for generating system-usages to get around bzr merge braindeadness.

1 files changed, 127 insertions, 0 deletions

diff --git a/bin/instance-usage-audit b/bin/instance-usage-audit
new file mode 100755
index 000000000..1124cf550
--- /dev/null
+++ b/bin/instance-usage-audit
@@ -0,0 +1,127 @@
+#!/usr/bin/env python
+# vim: tabstop=4 shiftwidth=4 softtabstop=4
+
+# Copyright (c) 2011 Openstack, LLC.
+# All Rights Reserved.
+#
+#    Licensed under the Apache License, Version 2.0 (the "License"); you may
+#    not use this file except in compliance with the License. You may obtain
+#    a copy of the License at
+#
+#         http://www.apache.org/licenses/LICENSE-2.0
+#
+#    Unless required by applicable law or agreed to in writing, software
+#    distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+#    WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
+#    License for the specific language governing permissions and limitations
+#    under the License.
+
+"""Cron script to generate usage notifications for instances neither created
+   nor destroyed in a given time period. 
+   
+   Together with the notifications generated by compute on instance
+   create/delete/resize, over that ime period, this allows an external 
+   system consuming usage notification feeds to calculate instance usage 
+   for each tenant.
+
+   Time periods are specified like so:
+   <number>[mdy]   
+
+   1m = previous month. If the script is run April 1, it will generate usages
+        for March 1 thry March 31.
+   3m = 3 previous months.
+   90d = previous 90 days. 
+   1y = previous year. If run on Jan 1, it generates usages for 
+        Jan 1 thru Dec 31 of the previous year.      
+"""
+
+import datetime
+import gettext
+import os
+import sys
+import time
+
+# If ../nova/__init__.py exists, add ../ to Python search path, so that
+# it will override what happens to be installed in /usr/(local/)lib/python...
+POSSIBLE_TOPDIR = os.path.normpath(os.path.join(os.path.abspath(sys.argv[0]),
+                                   os.pardir,
+                                   os.pardir))
+if os.path.exists(os.path.join(POSSIBLE_TOPDIR, 'nova', '__init__.py')):
+    sys.path.insert(0, POSSIBLE_TOPDIR)
+
+gettext.install('nova', unicode=1)
+
+
+from nova import context
+from nova import db
+from nova import exception
+from nova import flags
+from nova import log as logging
+from nova import utils
+
+from nova.notifier import api as notifier_api
+
+FLAGS = flags.FLAGS
+flags.DEFINE_string('instance_usage_audit_period', '1m',
+                    'time period to generate instance usages for.')
+
+def time_period(period):
+    today = datetime.date.today()
+    unit = period[-1]
+    if unit not in 'mdy':
+        raise ValueError('Time period must be m, d, or y')
+    n = int(period[:-1])
+    if unit == 'm':
+       year = today.year - (n//12)
+       n = n % 12 
+       if n >= today.month:
+           year -= 1 
+           month = 12 + (today.month - n)
+       else:
+           month = today.month - n 
+       begin = datetime.datetime(day=1, month=month, year=year)
+       end = datetime.datetime(day=1, month=today.month, year=today.year)
+
+    elif unit == 'y':
+       begin = datetime.datetime(day=1, month=1, year=today.year - n)
+       end = datetime.datetime(day=1, month=1, year=today.year)
+      
+    elif unit == 'd':
+       b = today - datetime.timedelta(days=n)
+       begin = datetime.datetime(day=b.day, month=b.month, year=b.year)
+       end = datetime.datetime(day=today.day, 
+                               month=today.month, 
+                               year=today.year)
+ 
+    return (begin, end)
+
+if __name__ == '__main__':
+    utils.default_flagfile()
+    flags.FLAGS(sys.argv)
+    logging.setup()
+    begin, end = time_period(FLAGS.instance_usage_audit_period)
+    print "Creating usages for %s until %s" % (str(begin), str(end))
+    instances = db.instance_get_active_by_window(context.get_admin_context(),
+                                                 begin,
+                                                 end)
+    print "%s instances" % len(instances)
+    for instance_ref in instances:
+        usage_info = dict(
+              tenant_id=instance_ref['project_id'],
+              user_id=instance_ref['user_id'],
+              instance_id=instance_ref['id'],
+              instance_type=instance_ref['instance_type']['name'],
+              instance_type_id=instance_ref['instance_type_id'],
+              display_name=instance_ref['display_name'],
+              created_at=str(instance_ref['created_at']),
+              launched_at=str(instance_ref['launched_at']) \
+                          if instance_ref['launched_at'] else '',
+              image_id=instance_ref['image_id'],
+              audit_period_begining=str(begin),
+              audit_period_ending=str(end))
+        notifier_api.notify('compute.%s' % FLAGS.host,
+                            'compute.instance.exists',
+                            notifier_api.INFO,
+                            usage_info)
+
+