102 lines
3.8 KiB
Python
Executable File
102 lines
3.8 KiB
Python
Executable File
#!/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 volumes existing during
|
|
the audit period.
|
|
|
|
Together with the notifications generated by volumes
|
|
create/delete/resize, over that time period, this allows an external
|
|
system consuming usage notification feeds to calculate volume usage
|
|
for each tenant.
|
|
|
|
Time periods are specified as 'hour', 'month', 'day' or 'year'
|
|
|
|
hour = previous hour. If run at 9:07am, will generate usage for 8-9am.
|
|
month = previous month. If the script is run April 1, it will generate
|
|
usages for March 1 through March 31.
|
|
day = previous day. if run on July 4th, it generates usages for July 3rd.
|
|
year = previous year. If run on Jan 1, it generates usages for
|
|
Jan 1 through Dec 31 of the previous year.
|
|
"""
|
|
|
|
import os
|
|
import sys
|
|
import traceback
|
|
|
|
# If ../cinder/__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, 'cinder', '__init__.py')):
|
|
sys.path.insert(0, POSSIBLE_TOPDIR)
|
|
|
|
from cinder.openstack.common import gettextutils
|
|
gettextutils.install('cinder')
|
|
|
|
from cinder import context
|
|
from cinder import db
|
|
from cinder import flags
|
|
from cinder.openstack.common import log as logging
|
|
from cinder.openstack.common import rpc
|
|
from cinder import utils
|
|
import cinder.volume.utils
|
|
|
|
|
|
FLAGS = flags.FLAGS
|
|
|
|
if __name__ == '__main__':
|
|
admin_context = context.get_admin_context()
|
|
flags.parse_args(sys.argv)
|
|
logging.setup("cinder")
|
|
begin, end = utils.last_completed_audit_period()
|
|
print _("Starting volume usage audit")
|
|
msg = _("Creating usages for %(begin_period)s until %(end_period)s")
|
|
print (msg % {"begin_period": str(begin), "end_period": str(end)})
|
|
|
|
extra_info = {
|
|
'audit_period_beginning': str(begin),
|
|
'audit_period_ending': str(end),
|
|
}
|
|
|
|
volumes = db.volume_get_active_by_window(admin_context,
|
|
begin,
|
|
end)
|
|
print _("Found %d volumes") % len(volumes)
|
|
for volume_ref in volumes:
|
|
try:
|
|
cinder.volume.utils.notify_usage_exists(
|
|
admin_context, volume_ref)
|
|
except Exception, e:
|
|
print traceback.format_exc(e)
|
|
|
|
snapshots = db.snapshot_get_active_by_window(admin_context,
|
|
begin,
|
|
end)
|
|
print _("Found %d snapshots") % len(snapshots)
|
|
for snapshot_ref in snapshots:
|
|
try:
|
|
cinder.volume.utils.notify_about_snapshot_usage(admin_context,
|
|
snapshot_ref,
|
|
'exists',
|
|
extra_info)
|
|
except Exception, e:
|
|
print traceback.fromat_exc(e)
|
|
|
|
print _("Volume usage audit completed")
|