mirror of
https://github.com/xcat2/confluent.git
synced 2026-09-05 20:47:57 +00:00
Markus Hilger
5d7fadf615
nodesensors and nodeconfig printed an error and exited 0, so `nodesensors n1 && next-step` ran next-step after the read it guarded had already failed. nodesensors had three of these: the per-node branch never set the exit code, the top-level branch beside it read `exitcode |= exitcode`, and a normal return from main() fell off the end of the file. nodeconfig accumulates with |= all through its read path except the last line, which assigned, so a failed bmc configuration read was discarded by the system read after it. The fixed branches now match how nodehealth and client.py spell the same thing.
271 lines
10 KiB
Python
Executable File
271 lines
10 KiB
Python
Executable File
#!/usr/bin/python2
|
|
# vim: tabstop=4 shiftwidth=4 softtabstop=4
|
|
|
|
# Copyright 2015-2017 Lenovo
|
|
#
|
|
# 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.
|
|
|
|
import csv
|
|
import datetime
|
|
import optparse
|
|
import os
|
|
import signal
|
|
import sys
|
|
import time
|
|
|
|
try:
|
|
signal.signal(signal.SIGPIPE, signal.SIG_DFL)
|
|
except AttributeError:
|
|
pass
|
|
path = os.path.dirname(os.path.realpath(__file__))
|
|
path = os.path.realpath(os.path.join(path, '..', 'lib', 'python'))
|
|
if path.startswith('/opt'):
|
|
sys.path.append(path)
|
|
|
|
|
|
class hybridcsv(csv.excel):
|
|
lineterminator = '\n'
|
|
|
|
|
|
def floatformat(num):
|
|
fm = u'{:.5f}'.format(num).rstrip('0')
|
|
if fm[-1:] == u'.':
|
|
return fm + u'0'
|
|
return fm
|
|
|
|
csv.register_dialect('hybrid', hybridcsv)
|
|
|
|
import confluent.client as client
|
|
|
|
sensorcollections = {
|
|
'all': 'sensors/hardware/all/all',
|
|
'energy': 'sensors/hardware/energy/all',
|
|
'temperature': 'sensors/hardware/temperature/all',
|
|
'temp': 'sensors/hardware/temperature/all',
|
|
'power': 'sensors/hardware/power/all',
|
|
'fans': 'sensors/hardware/fans/all',
|
|
'fanspeed': 'sensors/hardware/fans/all',
|
|
}
|
|
|
|
|
|
argparser = optparse.OptionParser(
|
|
usage="Usage: %prog [options] noderange ([sensor(s)])")
|
|
argparser.add_option('-i', '--interval', type='float',
|
|
help='Interval to do repeated samples over')
|
|
argparser.add_option('-n', '--numreadings', type='int',
|
|
help='Number of readings to gather')
|
|
argparser.add_option('-c', '--csv', action='store_true',
|
|
help='Output in CSV format')
|
|
argparser.add_option('-s', '--skipnumberless', action='store_true',
|
|
help='Do not show non-numeric sensors')
|
|
(options, args) = argparser.parse_args()
|
|
repeatmode = False
|
|
if options.interval:
|
|
repeatmode = True
|
|
if options.numreadings:
|
|
repeatmode = options.numreadings
|
|
if options.interval is None:
|
|
options.interval = 1
|
|
try:
|
|
noderange = args[0]
|
|
client.check_globbing(noderange)
|
|
except IndexError:
|
|
argparser.print_help()
|
|
sys.exit(1)
|
|
sensors = []
|
|
for sensorgroup in args[1:]:
|
|
for sensor in sensorgroup.split(','):
|
|
sensor = sensor.replace(' ', '_').lower()
|
|
if '/' not in sensor:
|
|
if sensor in sensorcollections:
|
|
sensors.append(sensorcollections[sensor])
|
|
else:
|
|
sensors.append('sensors/hardware/all/' + sensor)
|
|
if not sensors:
|
|
sensors = ['sensors/hardware/all/all']
|
|
session = client.Command()
|
|
exitcode = 0
|
|
sensorheaders = {}
|
|
|
|
|
|
def sensorpass(showout=True, appendtime=False):
|
|
global exitcode
|
|
resultdata = {}
|
|
for reqsensor in sensors:
|
|
for reading in session.read(
|
|
'/noderange/' + noderange + '/' + reqsensor):
|
|
if 'error' in reading:
|
|
sys.stderr.write('Error: {0}\n'.format(reading['error']))
|
|
if 'errorcode' in reading:
|
|
exitcode |= reading['errorcode']
|
|
else:
|
|
exitcode |= 1
|
|
if 'databynode' not in reading:
|
|
continue
|
|
reading = reading['databynode']
|
|
for node in reading:
|
|
if node not in resultdata:
|
|
resultdata[node] = {}
|
|
if 'error' in reading[node]:
|
|
sys.stderr.write(
|
|
'{0}: Error: {1}\n'.format(node,
|
|
reading[node]['error']))
|
|
if 'errorcode' in reading[node]:
|
|
exitcode |= reading[node]['errorcode']
|
|
else:
|
|
exitcode |= 1
|
|
if 'sensors' not in reading[node]:
|
|
continue
|
|
for sensedata in reading[node]['sensors']:
|
|
if sensedata.get('value', None) is None and options.skipnumberless:
|
|
continue
|
|
for redundant_state in ('Non-Critical', 'Critical'):
|
|
try:
|
|
if sensedata.get('states', False):
|
|
sensedata['states'].remove(redundant_state)
|
|
except ValueError:
|
|
pass
|
|
resultdata[node][sensedata['name']] = sensedata
|
|
sensorname = sensedata['name']
|
|
sensorheaders[sensorname] = sensorname
|
|
if sensedata.get('units', None) not in (None, u''):
|
|
sensorheaders[sensorname] += u' ({0})'.format(
|
|
sensedata['units'])
|
|
if showout:
|
|
if sensedata.get('value', None) is None:
|
|
showval = ''
|
|
elif isinstance(sensedata['value'], float):
|
|
showval = u' {0} '.format(floatformat(sensedata['value']))
|
|
else:
|
|
showval = u' {0} '.format(sensedata.get('value', ''))
|
|
# A discrete sensor has no reading for a unit to apply
|
|
# to, and printing the unit on its own says nothing
|
|
if (showval
|
|
and sensedata.get('units', None) not in (None, u'')):
|
|
showval += sensedata['units']
|
|
if sensedata.get('health', 'ok') != 'ok':
|
|
datadescription = [sensedata['health']]
|
|
else:
|
|
datadescription = []
|
|
if sensedata.get('states', False):
|
|
datadescription.extend(sensedata['states'])
|
|
if datadescription:
|
|
if showval == '':
|
|
showval += u' {0}'.format(
|
|
','.join(datadescription))
|
|
else:
|
|
showval += u' ({0})'.format(
|
|
','.join(datadescription))
|
|
if appendtime:
|
|
showval += ' @' + time.strftime(
|
|
'%Y-%m-%dT%H:%M:%S')
|
|
printval = u'{0}: {1}:{2}'.format(
|
|
node, sensedata['name'], showval)
|
|
if not isinstance(printval, str):
|
|
printval = printval.encode('utf-8')
|
|
print(printval)
|
|
sys.stdout.flush()
|
|
return resultdata
|
|
|
|
|
|
def format_csv(csvwriter, orderedsensors, resdata, showtime=True):
|
|
for nodekey in resdata:
|
|
if showtime:
|
|
if isinstance(showtime, int):
|
|
rowdata = [time.strftime('%Y-%m-%dT%H:%M:%S'), nodekey]
|
|
else:
|
|
rowdata = [time.strftime('%Y-%m-%dT%H:%M:%S.') +
|
|
str(datetime.datetime.now().microsecond//1000),
|
|
nodekey]
|
|
else:
|
|
rowdata = [nodekey]
|
|
for sensorkey in orderedsensors:
|
|
try:
|
|
datum = resdata[nodekey][sensorkey]['value']
|
|
if datum is None:
|
|
if resdata[nodekey][sensorkey]['health'] != 'ok':
|
|
datum = resdata[nodekey][sensorkey]['health']
|
|
else:
|
|
datum = ''
|
|
if resdata[nodekey][sensorkey]['states']:
|
|
healthstates = ','.join(
|
|
resdata[nodekey][sensorkey]['states'])
|
|
if datum != '':
|
|
datum = ','.join([datum, healthstates])
|
|
else:
|
|
datum = healthstates
|
|
if isinstance(datum, float):
|
|
datum = floatformat(datum)
|
|
rowdata.append(datum)
|
|
except KeyError:
|
|
rowdata.append('N/A')
|
|
csvwriter.writerow(rowdata)
|
|
sys.stdout.flush()
|
|
|
|
|
|
def main():
|
|
linebyline = True
|
|
headernames = []
|
|
orderedsensors = []
|
|
csvwriter = None
|
|
if options.interval or options.csv:
|
|
resdata = sensorpass(False)
|
|
for name in sensorheaders:
|
|
orderedsensors.append(name)
|
|
orderedsensors.sort()
|
|
for name in orderedsensors:
|
|
headername = sensorheaders[name]
|
|
if (not isinstance(headername, str) and
|
|
not isinstance(headername, bytes)):
|
|
headername = headername.encode('utf-8')
|
|
headernames.append(headername)
|
|
if options.csv:
|
|
linebyline = False
|
|
csvwriter = csv.writer(sys.stdout, dialect='hybrid')
|
|
if options.interval:
|
|
csvwriter.writerow(['time', 'node'] + headernames)
|
|
else:
|
|
csvwriter.writerow(['node'] + headernames)
|
|
if options.interval:
|
|
# first do a pass to swallow up probable causes of uneven timing
|
|
# for example if some have sdrs fetch and others not, this should
|
|
# get a common baseline going
|
|
while True:
|
|
nextstart = os.times()[4] + options.interval
|
|
resdata = sensorpass(linebyline, True)
|
|
if options.csv:
|
|
format_csv(csvwriter, orderedsensors, resdata,
|
|
showtime=options.interval)
|
|
if options.numreadings:
|
|
options.numreadings -= 1
|
|
if options.numreadings <= 0:
|
|
sys.exit(exitcode)
|
|
sleeptime = nextstart - os.times()[4]
|
|
if sleeptime > 0:
|
|
time.sleep(sleeptime)
|
|
else:
|
|
if options.csv:
|
|
format_csv(csvwriter, orderedsensors, resdata, showtime=False)
|
|
else:
|
|
sensorpass(True)
|
|
|
|
|
|
try:
|
|
main()
|
|
except KeyboardInterrupt:
|
|
print('')
|
|
sys.exit(0)
|
|
# Falling off the end exited 0 whatever sensorpass recorded, so only the
|
|
# --interval path ever reported a failed read.
|
|
sys.exit(exitcode)
|