check_speedport

#!/usr/bin/env python3
import argparse
import requests
from influxdb import InfluxDBClient
from influxdb.exceptions import InfluxDBServerError, InfluxDBClientError

url = 'http://[ROUTER_IP]/data/Status.json'
headers = {'content-type': 'application/json', 'Accept-Charset': 'UTF-8', 'Accept-Language': 'en'}

parser = argparse.ArgumentParser(description='Read Speedport router data.')
parser.add_argument('-v', '--verbose', action='store_true')
parser.add_argument('-w', '--write', action='store_true')
args = parser.parse_args()

try:
    r = requests.get(url, headers=headers)
except requests.exceptions.RequestException as e:
    exit(2)

data = {}
if r.content.decode('utf-8').__contains__("login.html"):
    print('Router is not supported')
    exit(4)
if r.status_code == requests.codes.ok:
    j = r.json()
    for l in j:
        if args.verbose:
            print(l['varid'], ':', l['varvalue'])
        data[l['varid']] = l['varvalue']

    # send to influxDB
    if args.write:
        d = [{
            "measurement": "speedport_data",
            "fields": data
        }]
        try:
            client = InfluxDBClient(host='127.0.0.1', port=8086, database='speedport')
            client.write_points(d)
        except (InfluxDBClientError, InfluxDBServerError):
            print("Error writing to InfluxDB")
            exit(3)

    if data['dsl_link_status'] == 'online':
        exit(0)

exit(1)