Compare commits
No commits in common. "5df2efac05e9fd6c748d806685f52ad922744bf4" and "9f24acca6bbf5a1ab17416dcf1c2f5cce217eed6" have entirely different histories.
5df2efac05
...
9f24acca6b
2 changed files with 6 additions and 85 deletions
|
@ -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
|
||||||
|
|
|
@ -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()
|
||||||
]))
|
]))
|
||||||
|
|
Loading…
Reference in a new issue