Antoine Van Elstraete
10fad0d9f3
- New temperature_oid config parameter - Temperature published as integer via MQTT - Auto-creates Home Assistant temperature sensor via MQTT Discovery Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
81 lines
2.7 KiB
Python
81 lines
2.7 KiB
Python
import asyncio
|
|
import time
|
|
|
|
from pysnmp.hlapi.v1arch.asyncio import (
|
|
Slim,
|
|
ObjectIdentity,
|
|
ObjectType,
|
|
)
|
|
|
|
from .base import BaseChecker, CheckResult
|
|
|
|
|
|
class SnmpChecker(BaseChecker):
|
|
def check(self) -> CheckResult:
|
|
return asyncio.run(self._async_check())
|
|
|
|
async def _async_check(self) -> CheckResult:
|
|
host = self.config["host"]
|
|
port = self.config.get("port", 161)
|
|
community = self.config.get("community", "public")
|
|
oid = self.config.get("oid", "1.3.6.1.2.1.1.1.0") # sysDescr
|
|
temperature_oid = self.config.get("temperature_oid")
|
|
timeout_val = self.config.get("timeout", 5)
|
|
|
|
start = time.time()
|
|
try:
|
|
with Slim() as slim:
|
|
# Build list of OIDs to query
|
|
oids = [ObjectType(ObjectIdentity(oid))]
|
|
if temperature_oid:
|
|
oids.append(ObjectType(ObjectIdentity(temperature_oid)))
|
|
|
|
error_indication, error_status, error_index, var_binds = await slim.get(
|
|
community,
|
|
host,
|
|
port,
|
|
*oids,
|
|
timeout=timeout_val,
|
|
retries=1
|
|
)
|
|
|
|
response_time = (time.time() - start) * 1000 # ms
|
|
|
|
if error_indication:
|
|
return CheckResult(
|
|
success=False,
|
|
message=f"SNMP error: {error_indication}",
|
|
response_time=None
|
|
)
|
|
elif error_status:
|
|
return CheckResult(
|
|
success=False,
|
|
message=f"SNMP error: {error_status.prettyPrint()}",
|
|
response_time=None
|
|
)
|
|
else:
|
|
details = {str(oid): str(val) for oid, val in var_binds}
|
|
|
|
# Extract temperature if configured
|
|
if temperature_oid:
|
|
for oid_key, val in var_binds:
|
|
if str(oid_key) == temperature_oid:
|
|
try:
|
|
details["temperature"] = int(val)
|
|
except (ValueError, TypeError):
|
|
pass # Ignore if not a valid integer
|
|
break
|
|
|
|
return CheckResult(
|
|
success=True,
|
|
message="SNMP response OK",
|
|
response_time=response_time,
|
|
details=details
|
|
)
|
|
except Exception as e:
|
|
return CheckResult(
|
|
success=False,
|
|
message=f"SNMP error: {e}",
|
|
response_time=None
|
|
)
|