YuruC3
10e6ae2347
Using modules and imports makes it much more modular and managable. There are no 600 row files now
71 lines
2.7 KiB
Python
71 lines
2.7 KiB
Python
import influxdb_client, os, asyncio
|
|
from influxdb_client import InfluxDBClient, Point, WritePrecision
|
|
from influxdb_client.client.write_api import SYNCHRONOUS, ASYNCHRONOUS, WriteOptions
|
|
from typing import Annotated, Final
|
|
from models.idracModel import snmpPyIDRACData
|
|
|
|
# FluxQL ENV----------------------------------------
|
|
USEINFLUX: Final[int] = int(os.getenv("USEINFLUX", 1))
|
|
if USEINFLUX:
|
|
INFLXDBTOKEN: Final[str] = os.getenv("INFLXDBTOKEN", "123" )
|
|
INFLUXBCKT: Final[str] = os.getenv("INFLUXBCKT", "SNMPyth")
|
|
INFLUXORG: Final[str] = os.getenv("INFLUXORG", "staging")
|
|
INFLXDBURL: Final[str] = os.getenv("INFLXDBURL", "http://localhost:8086")
|
|
INFXLUXDB_MEASUEREMENT: Final[str] = os.getenv("INFXLUXDB_MEASUEREMENT", "SNMPyth-containrr")
|
|
INFLX_SEPARATE_POINTS: Final[float] = float(os.getenv("INFLX_SEPARATE_POINTS", 0.1))
|
|
|
|
# Prepare-------------------------------------------
|
|
# INFLUX
|
|
if USEINFLUX:
|
|
fluxdb_client = influxdb_client.InfluxDBClient(url=INFLXDBURL, token=INFLXDBTOKEN, org=INFLUXORG)
|
|
write_fluxdb_api = fluxdb_client.write_api(write_options=ASYNCHRONOUS)
|
|
query_fluxdb_api = fluxdb_client.query_api()
|
|
else:
|
|
fluxdb_client = write_fluxdb_api = query_fluxdb_api = None
|
|
|
|
async def fluxIDRACWriter(
|
|
inpDict: dict
|
|
) -> int:
|
|
|
|
"""
|
|
Insert temperature data into InfluxDB
|
|
:inputQueue: Asyncio Queue that has what is needed to be sent
|
|
"""
|
|
# inputQueue have multiple such Dicts
|
|
# {
|
|
# "source": "IDRAC",
|
|
# "value": returnObj,
|
|
# "type": "snmpPyIDRACData"
|
|
# }
|
|
|
|
if write_fluxdb_api is None:
|
|
return 2
|
|
|
|
# Prep InfluxDB data
|
|
inflxdb_Data_To_Send = (
|
|
influxdb_client.Point(INFXLUXDB_MEASUEREMENT)
|
|
.tag("SOURCE", inpDict["source"])
|
|
.tag("TYPE", inpDict["type"])
|
|
.tag("HOSTNAME", inpDict["value"].hostname)
|
|
.field("PowerDrawPSU1", inpDict["value"].powerDrawPSU1)
|
|
.field("PowerDrawPSU2", inpDict["value"].powerDrawPSU2)
|
|
.field("TotalBoardPower", inpDict["value"].powerDrawBoard)
|
|
.field("VoltagePSU1", inpDict["value"].voltagePSU1)
|
|
.field("VoltagePSU2", inpDict["value"].voltagePSU2)
|
|
.field("InletTemperature", inpDict["value"].inletTemp)
|
|
.field("ExhaustTemperature", inpDict["value"].exhaustTemp)
|
|
.field("TemperatureCPU1", inpDict["value"].cpu1Temp)
|
|
.field("TemperatureCPU2", inpDict["value"].cpu2Temp)
|
|
.field("UptimeInSeconds", inpDict["value"].uptimeS)
|
|
.field("UptimeInHours", inpDict["value"].uptimeH)
|
|
.field("UptimeInDays", inpDict["value"].uptimeD)
|
|
)
|
|
|
|
try:
|
|
write_fluxdb_api.write(bucket=INFLUXBCKT, org=INFLUXORG, record=inflxdb_Data_To_Send)
|
|
except Exception as e:
|
|
print(e)
|
|
return 1
|
|
|
|
return 0
|
|
# return {"STATUS": "succesfully inserted to InfluxDB"} |