53 lines
1.6 KiB
Python
53 lines
1.6 KiB
Python
import os
|
|
from prometheus_client import Gauge
|
|
from prometheus_async import aio
|
|
import asyncio
|
|
import pyfabapi.fabapi
|
|
|
|
def convert2int(enum):
|
|
if enum == "free":
|
|
return 0.0
|
|
if enum == "inUse":
|
|
return 1.0
|
|
if enum == "toCheck":
|
|
return 2.0
|
|
if enum == "blocked":
|
|
return 3.0
|
|
if enum == "disabled":
|
|
return 4.0
|
|
if enum == "reserved":
|
|
return 5.0
|
|
if enum == "totakeover":
|
|
return 6.0
|
|
|
|
|
|
async def main():
|
|
polling_interval_seconds = int(os.getenv("POLLING_INTERVAL_SECONDS", "5"))
|
|
exporter_port = int(os.getenv("EXPORTER_PORT", "9000"))
|
|
|
|
bffh_host = str(os.getenv("BFFH_HOST", "localhost"))
|
|
bffh_port = int(os.getenv("BFFH_PORT", "59661"))
|
|
bffh_user = str(os.getenv("BFFH_USER", "Admin1"))
|
|
bffh_password = str(os.getenv("BFFH_PASSWORD", "secret"))
|
|
|
|
await aio.web.start_http_server(addr="0.0.0.0", port=exporter_port)
|
|
|
|
metriclist = {}
|
|
session = await pyfabapi.fabapi.connect(bffh_host, bffh_port, bffh_user, bffh_password)
|
|
|
|
machine_state = Gauge("bffh_machine_state", "Machine State", ["machine_id", "machine_name", "category"])
|
|
|
|
while True:
|
|
machine_list = await session.machineSystem.info.getMachineList().a_wait()
|
|
machine_list = machine_list.machine_list
|
|
|
|
for machine in machine_list:
|
|
machine_state.labels(machine.id, machine.name, machine.category).set(convert2int(machine.state))
|
|
|
|
await asyncio.sleep(polling_interval_seconds)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
loop = asyncio.new_event_loop()
|
|
asyncio.set_event_loop(loop)
|
|
loop.run_until_complete(main()) |