munin-contrib/plugins/keystone/keystone_stats

124 lines
3.0 KiB
Python
Executable File

#!/usr/bin/env python3
"""
=head1 NAME
keystone_stats - monitor status of Keystone
=head1 CONFIGURATION
Needs following minimal configuration in plugin-conf.d/keystone:
[keystone_*]
user keystone
=head1 AUTHORS
Copyright 2012 - Mehdi Abaakouk <sileht@sileht.net>
=head1 MAGIC MARKERS
#%# capabilities=autoconf
#%# family=auto
=cut
"""
import sys
import traceback
try:
from keystone.common import utils
from keystone import config
from keystone import exception
from keystone import identity
except ImportError:
successful_import = False
else:
successful_import = True
stats = ['users', 'tenants']
def print_config():
print('graph_title Keystone Stats')
print('graph_vlabel count')
print('graph_args --base 1000 --lower-limit 0')
print('graph_category auth')
print('graph_scale no')
print('graph_info This graph shows stats about keystone: ' + (', ').join(stats))
for field in stats:
print('%s_enabled.label enabled %s' % (field, field))
print('%s_enabled.draw LINE2' % field)
print('%s_enabled.info %s enabled' % (field, field))
print('%s_total.label total %s' % (field, field))
print('%s_total.draw LINE2' % field)
print('%s_total.info %s total' % (field, field))
def get_status():
enabled = {}
total = {}
for k in stats:
enabled[k] = 0
total[k] = 0
identity_api = identity.Manager()
for user in identity_api.list_users(None):
total['users'] += 1
if user['enabled']:
enabled['users'] += 1
# Ldap and pam driver don't support get_all_tenants()
# kvs and sql implement get_tenants() instead of get_all_tenants()
# Whoo: None of backend implements the correct function
tenants = []
for api_func in ['get_all_tenants', 'get_tenants']:
try:
tenants = getattr(identity_api, api_func)(None)
except (exception.NotImplemented, NotImplementedError):
pass
for tenant in tenants:
total['tenants'] += 1
if tenant['enabled']:
enabled['tenants'] += 1
return {'enabled': enabled, 'total': total}
def print_values():
stats = get_status()
for state in stats.keys():
for (field, value) in stats[state].items():
print("%s_%s.value %s" % (field, state, value))
def load_conf():
config.CONF(config_files=[utils.find_config('keystone.conf')])
if __name__ == '__main__':
if len(sys.argv) > 1:
if sys.argv[1] == "config":
print_config()
elif sys.argv[1] == "autoconf":
if not successful_import:
print('no (failed import keystone module)')
sys.exit(0)
try:
load_conf()
identity.Manager()
except Exception:
print('no (failed to connect keystone backend: %s' % traceback.format_exc())
sys.exit(0)
print('yes')
elif successful_import:
load_conf()
print_values()