94 lines
3.3 KiB
Python
94 lines
3.3 KiB
Python
# Author: François Rossigneux <francois.rossigneux@inria.fr>
|
|
#
|
|
# 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 time
|
|
|
|
from pysnmp.entity.rfc3413.oneliner import cmdgen
|
|
|
|
from kwapi.openstack.common import log
|
|
from driver import Driver
|
|
|
|
LOG = log.getLogger(__name__)
|
|
|
|
|
|
class Snmp(Driver):
|
|
"""Driver for SNMP based PDUs."""
|
|
|
|
def __init__(self, probe_ids, **kwargs):
|
|
"""Initializes the SNMP driver.
|
|
|
|
Keyword arguments:
|
|
probe_ids -- list containing the probes IDs
|
|
(a wattmeter monitor sometimes several probes)
|
|
kwargs -- keyword (protocol, user or community, ip, oid) defining the
|
|
SNMP parameters
|
|
Eaton OID is 1.3.6.1.4.1.534.6.6.7.6.5.1.3
|
|
Raritan OID is 1.3.6.1.4.1.13742.4.1.2.2.1.7
|
|
|
|
"""
|
|
Driver.__init__(self, probe_ids, kwargs)
|
|
self.cmd_gen = cmdgen.CommandGenerator()
|
|
|
|
def run(self):
|
|
"""Starts the driver thread."""
|
|
while not self.stop_request_pending():
|
|
watts_list = self.get_watts()
|
|
if watts_list is not None:
|
|
i = 0
|
|
for watts in watts_list:
|
|
measurements = {}
|
|
measurements['w'] = watts
|
|
if self.probe_ids[i]:
|
|
self.send_measurements(self.probe_ids[i], measurements)
|
|
i += 1
|
|
time.sleep(1)
|
|
|
|
def get_watts(self):
|
|
"""Returns the power consumption."""
|
|
protocol = self.kwargs.get('protocol')
|
|
if protocol is '1':
|
|
community_or_user = cmdgen.CommunityData(
|
|
self.kwargs.get('community'),
|
|
mpModel=0)
|
|
elif protocol is '2c':
|
|
community_or_user = cmdgen.CommunityData(
|
|
self.kwargs.get('community'),
|
|
mpModel=1)
|
|
elif protocol is '3':
|
|
community_or_user = cmdgen.UsmUserData(self.kwargs.get('user'))
|
|
errorIndication, errorStatus, errorIndex, varBindTable = \
|
|
self.cmd_gen.bulkCmd(
|
|
community_or_user,
|
|
cmdgen.UdpTransportTarget((self.kwargs.get('ip'), 161)),
|
|
1, 0,
|
|
self.kwargs.get('oid'),
|
|
maxRows=len(self.probe_ids),
|
|
)
|
|
if errorIndication:
|
|
LOG.error(errorIndication)
|
|
return None
|
|
else:
|
|
if errorStatus:
|
|
LOG.error('%s at %s' % (
|
|
errorStatus.prettyPrint(),
|
|
errorIndex and varBindTable[-1][int(errorIndex)-1] or '?'
|
|
))
|
|
return None
|
|
else:
|
|
outlet_list = []
|
|
for varBindTableRow in varBindTable:
|
|
for name, value in varBindTableRow:
|
|
outlet_list.append(int(value))
|
|
return outlet_list
|