mirror of
https://github.com/xcat2/confluent.git
synced 2025-07-02 10:55:36 +00:00
94 lines
3.5 KiB
Python
94 lines
3.5 KiB
Python
# Copyright 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 confluent.core as core
|
|
import confluent.messages as msg
|
|
import pyghmi.exceptions as pygexc
|
|
import confluent.exceptions as exc
|
|
|
|
def retrieve(nodes, element, configmanager, inputdata):
|
|
emebs = configmanager.get_node_attributes(
|
|
nodes, (u'power.*pdu', u'power.*outlet'))
|
|
if element == ['power', 'inlets']:
|
|
outletnames = set([])
|
|
for node in nodes:
|
|
for attrib in emebs[node]:
|
|
attrib = attrib.replace('power.', '').rsplit('.', 1)
|
|
if len(attrib) > 1:
|
|
outletnames.add('inlet_' + attrib[0])
|
|
else:
|
|
outletnames.add('default')
|
|
if outletnames:
|
|
outletnames.add('all')
|
|
for inlet in outletnames:
|
|
yield msg.ChildCollection(inlet)
|
|
elif len(element) == 3:
|
|
inletname = element[-1]
|
|
outlets = get_outlets(nodes, emebs, inletname)
|
|
for node in outlets:
|
|
for pgroup in outlets[node]:
|
|
pdu = outlets[node][pgroup]['pdu']
|
|
outlet = outlets[node][pgroup]['outlet']
|
|
for rsp in core.handle_path(
|
|
'/nodes/{0}/_pdu/outlets/{1}'.format(pdu, outlet),
|
|
'retrieve', configmanager):
|
|
yield msg.KeyValueData({pgroup: rsp.kvpairs['state']['value']}, node)
|
|
|
|
def get_outlets(nodes, emebs, inletname):
|
|
outlets = {}
|
|
for node in nodes:
|
|
if node not in outlets:
|
|
outlets[node] = {}
|
|
for attrib in emebs[node]:
|
|
v = emebs[node][attrib].get('value', None)
|
|
if not v:
|
|
continue
|
|
attrib = attrib.replace('power.', '').rsplit('.', 1)
|
|
if len(attrib) > 1:
|
|
pgroup = 'inlet_' + attrib[0]
|
|
else:
|
|
pgroup = 'default'
|
|
if inletname == 'all' or pgroup == 'inletname':
|
|
if pgroup not in outlets[node]:
|
|
outlets[node][pgroup] = {}
|
|
outlets[node][pgroup][attrib[-1]] = v
|
|
return outlets
|
|
|
|
|
|
def update(nodes, element, configmanager, inputdata):
|
|
emebs = configmanager.get_node_attributes(
|
|
nodes, (u'power.*pdu', u'power.*outlet'))
|
|
for node in nodes:
|
|
for attrib in emebs[node]:
|
|
print(repr(attrib))
|
|
try:
|
|
em = emebs[node]['enclosure.manager']['value']
|
|
eb = emebs[node]['enclosure.bay']['value']
|
|
except KeyError:
|
|
em = node
|
|
eb = -1
|
|
if not em:
|
|
em = node
|
|
if not eb:
|
|
eb = -1
|
|
try:
|
|
for rsp in core.handle_path(
|
|
'/nodes/{0}/_enclosure/reseat_bay'.format(em),
|
|
'update', configmanager,
|
|
inputdata={'reseat': int(eb)}):
|
|
yield rsp
|
|
except pygexc.UnsupportedFunctionality as uf:
|
|
yield msg.ConfluentNodeError(node, str(uf))
|
|
except exc.TargetEndpointUnreachable as uf:
|
|
yield msg.ConfluentNodeError(node, str(uf))
|