mirror of
https://github.com/xcat2/confluent.git
synced 2026-09-04 12:07:58 +00:00
Markus Hilger
101 lines
3.9 KiB
Python
101 lines
3.9 KiB
Python
import asyncio
|
|
import confluent.exceptions as exc
|
|
import confluent.networking.srlinux as srlinux
|
|
import confluent.messages as msg
|
|
import confluent.tasks as tasks
|
|
|
|
|
|
async def retrieve_node(node, element, user, pwd, configmanager, inputdata, results):
|
|
try:
|
|
await retrieve_node_backend(node, element, user, pwd, configmanager, inputdata, results)
|
|
except exc.PubkeyInvalid as e:
|
|
await results.put(msg.ConfluentNodeError(node, 'Mismatch detected between target certificate fingerprint '
|
|
'and pubkeys.tls_hardwaremanager attribute'))
|
|
except Exception as e:
|
|
await results.put(e)
|
|
|
|
|
|
def simplify_name(name):
|
|
return name.lower().replace(' ', '_').replace('/', '-').replace(
|
|
'_-_', '-')
|
|
|
|
|
|
async def retrieve_node_backend(node, element, user, pwd, configmanager, inputdata, results):
|
|
cli = srlinux.SRLinuxClient(node, user, pwd, configmanager)
|
|
await cli.login()
|
|
if element == ['power', 'state']: # client initted successfully, must be on
|
|
await results.put(msg.PowerState(node, 'on'))
|
|
elif element == ['health', 'hardware']:
|
|
hinfo = await cli.get_health()
|
|
await results.put(msg.HealthSummary(hinfo.get('health', 'unknown'), name=node))
|
|
await results.put(msg.SensorReadings(hinfo.get('sensors', []), name=node))
|
|
elif element[:3] == ['inventory', 'hardware', 'all']:
|
|
if len(element) == 3:
|
|
await results.put(msg.ChildCollection('all'))
|
|
return
|
|
invinfo = await cli.get_inventory()
|
|
if invinfo:
|
|
await results.put(msg.KeyValueData({'inventory': invinfo}, node))
|
|
elif element[:3] == ['inventory', 'firmware', 'all']:
|
|
if len(element) == 3:
|
|
await results.put(msg.ChildCollection('all'))
|
|
return
|
|
fwinfo = []
|
|
for fwnam, fwdat in (await cli.get_firmware()).items():
|
|
fwinfo.append({fwnam: fwdat})
|
|
if fwinfo:
|
|
await results.put(msg.Firmware(fwinfo, node))
|
|
elif element == ['sensors', 'hardware', 'all']:
|
|
sensors = await cli.get_sensors()
|
|
for sensor in sensors:
|
|
await results.put(msg.ChildCollection(simplify_name(sensor['name'])))
|
|
elif element[:3] == ['sensors', 'hardware', 'all']:
|
|
sensors = await cli.get_sensors()
|
|
for sensor in sensors:
|
|
if element[-1] == 'all' or simplify_name(sensor['name']) == element[-1]:
|
|
await results.put(msg.SensorReadings([sensor], node))
|
|
else:
|
|
await results.put(msg.ConfluentNodeError(node, 'Not supported'))
|
|
|
|
|
|
async def retrieve(nodes, element, configmanager, inputdata):
|
|
results = asyncio.Queue()
|
|
workers = set([])
|
|
creds = configmanager.get_node_attributes(
|
|
nodes, ['secret.hardwaremanagementuser', 'secret.hardwaremanagementpassword'], decrypt=True)
|
|
for node in nodes:
|
|
cred = creds.get(node, {})
|
|
user = cred.get('secret.hardwaremanagementuser', {}).get('value')
|
|
pwd = cred.get('secret.hardwaremanagementpassword', {}).get('value')
|
|
try:
|
|
user = user.decode()
|
|
pwd = pwd.decode()
|
|
except Exception:
|
|
pass
|
|
if not user or not pwd:
|
|
yield msg.ConfluentTargetInvalidCredentials(node)
|
|
continue
|
|
workers.add(tasks.spawn(retrieve_node(node, element, user, pwd, configmanager, inputdata, results)))
|
|
while workers:
|
|
try:
|
|
datum = await asyncio.wait_for(results.get(), timeout=10.0)
|
|
while datum:
|
|
if isinstance(datum, Exception):
|
|
raise datum
|
|
if datum:
|
|
yield datum
|
|
datum = results.get_nowait()
|
|
except asyncio.QueueEmpty:
|
|
pass
|
|
await asyncio.sleep(0.001)
|
|
for t in list(workers):
|
|
if t.done():
|
|
workers.discard(t)
|
|
try:
|
|
while True:
|
|
datum = results.get_nowait()
|
|
if datum:
|
|
yield datum
|
|
except asyncio.QueueEmpty:
|
|
pass
|