Compare commits

..

No commits in common. "5df2efac05e9fd6c748d806685f52ad922744bf4" and "9f24acca6bbf5a1ab17416dcf1c2f5cce217eed6" have entirely different histories.

2 changed files with 6 additions and 85 deletions

View file

@ -19,18 +19,12 @@ service nagios-nrpe-server reload
## Usage ## Usage
``` ```
usage: check_esphome_devices [-h] [-d] [-H HOST] [-r RETRY_COUNT] [-D RETRY_DELAY] [-x EXCLUDE] usage: check_esphome_devices [-h] [-d] [-H HOST]
optional arguments: optional arguments:
-h, --help show this help message and exit -h, --help show this help message and exit
-d, --debug -d, --debug
-H HOST, --host HOST ESPHome dashboard URL (default: http://127.0.0.1:6052) -H HOST, --host HOST ESPHome dashboard URL (default: http://127.0.0.1:6052)
-r RETRY_COUNT, --retry RETRY_COUNT
Number of retry to retreive device status (default: 4)
-D RETRY_DELAY, --delay RETRY_DELAY
Delay in second between two retry to retreive device status (default: 1s)
-x EXCLUDE, --exclude EXCLUDE
Regex exclude pattern(
``` ```
## Copyright ## Copyright

View file

@ -21,9 +21,7 @@ Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
import argparse import argparse
import sys import sys
import time
import logging import logging
import re
import requests import requests
@ -37,8 +35,6 @@ STATUS = {
} }
DEFAULT_HOST = 'http://127.0.0.1:6052' DEFAULT_HOST = 'http://127.0.0.1:6052'
DEFAULT_RETRY_COUNT = 4
DEFAULT_RETRY_DELAY = 1
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument( parser.add_argument(
@ -57,42 +53,6 @@ parser.add_argument(
default=DEFAULT_HOST default=DEFAULT_HOST
) )
parser.add_argument(
'-r', '--retry',
action="store",
dest="retry_count",
help=(
'Number of retry to retreive device status '
f'(default: {DEFAULT_RETRY_COUNT})'),
type=int,
default=DEFAULT_RETRY_COUNT
)
parser.add_argument(
'-D', '--delay',
action="store",
dest="retry_delay",
help=(
'Delay in second between two retry to retreive device status '
f'(default: {DEFAULT_RETRY_DELAY}s)'),
type=int,
default=DEFAULT_RETRY_DELAY
)
def exclude_pattern(value):
""" Check and compile exclusion pattern parameter """
return re.compile(value)
parser.add_argument(
'-x', '--exclude',
action="append",
dest="exclude",
help='Regex exclude pattern(s)',
type=exclude_pattern,
default=[]
)
options = parser.parse_args() options = parser.parse_args()
logging.basicConfig( logging.basicConfig(
@ -102,17 +62,6 @@ logging.basicConfig(
if options.host[-1] == '/': if options.host[-1] == '/':
options.host = options.host[-1] options.host = options.host[-1]
def is_excluded(name):
""" Check if device is excluded """
for pattern in options.exclude:
if pattern.search(name):
logging.debug('Device %s is excluded', name)
return True
logging.debug('Device %s is not excluded', name)
return False
r = requests.get(f'{options.host}/devices') r = requests.get(f'{options.host}/devices')
devices_data = r.json() devices_data = r.json()
logging.debug('Devices data: %s (%s)', devices_data, type(devices_data)) logging.debug('Devices data: %s (%s)', devices_data, type(devices_data))
@ -121,29 +70,10 @@ if not devices_data:
print('UNKNOWN - Fail to retreive devices using ESPHome Dashboard API') print('UNKNOWN - Fail to retreive devices using ESPHome Dashboard API')
sys.exit(STATUS['UNKNOWN']) sys.exit(STATUS['UNKNOWN'])
COUNT = 0
while COUNT < options.retry_count:
r = requests.get(f'{options.host}/ping') r = requests.get(f'{options.host}/ping')
COUNT += 1
ping_data = r.json() ping_data = r.json()
logging.debug('Ping data: %s (%s)', ping_data, type(ping_data)) logging.debug('Ping data: %s (%s)', ping_data, type(ping_data))
if ping_data:
UNREACHABLE = False
for dev in ping_data:
if (
not ping_data[dev]
and not is_excluded(dev.replace('.yaml', ''))
):
UNREACHABLE = True
break
if not UNREACHABLE:
break
logging.debug('Wait %d seconds before retry...', options.retry_delay)
time.sleep(options.retry_delay)
if not ping_data: if not ping_data:
print( print(
'UNKNOWN - Fail to retreive devices status ' 'UNKNOWN - Fail to retreive devices status '
@ -158,8 +88,6 @@ devices = {}
for dev in devices_data['configured']: for dev in devices_data['configured']:
devices[dev['name']] = dev devices[dev['name']] = dev
logging.debug('Device %s: %s', dev['name'], dev) logging.debug('Device %s: %s', dev['name'], dev)
if is_excluded(dev['name']):
continue
if dev['deployed_version'] != dev['current_version']: if dev['deployed_version'] != dev['current_version']:
UPDATE_AVAILABLE += 1 UPDATE_AVAILABLE += 1
errors.append( errors.append(
@ -192,9 +120,8 @@ else:
print("\nDevices:\n" + "\n".join([ print("\nDevices:\n" + "\n".join([
( (
f'- {name} (version = ' f'- {name} (version = {dev["deployed_version"]}, '
f'{dev["deployed_version"] if dev["deployed_version"] else "unknown"}' f'address = {dev["address"]})'
f', address = {dev["address"] if dev["address"] else "unknown"})'
) )
for name, dev in devices.items() for name, dev in devices.items()
])) ]))