PyTelekomPass/PyPass.py

#!/usr/bin/env python3

# Python Telekom Data Usage Logger
# Copyright (c) 2018 Nis Wechselberg <enbewe@enbewe.de>
#
# This work is licensed under MIT license. You should have
# received a copy of the MIT license legalcode along with this
# work. If not, see <https://opensource.org/licenses/MIT>.

# pylint: disable=C0103

"""
Python Telekom Data Usage Logger

Small script to request data from the portal pass.telekom.de and
submit it to InfluxDB
"""

import argparse
import requests
from influxdb import InfluxDBClient

REQUEST_API_URL = 'http://pass.telekom.de/api/service/generic/v1/status'
REQUEST_HEADERS = {'User-Agent': 'Mozilla/4.0'}

parser = argparse.ArgumentParser()
parser.add_argument("hostname", help="Hostname of this machine")
parser.add_argument("influxHost", help="Hostname/IP of the InfluxDB machine")
parser.add_argument("database", help="Name of the database to write to")
args = parser.parse_args()

# Send http request to telekom page and extract json response
response = requests.get(REQUEST_API_URL, headers=REQUEST_HEADERS)
respJson = response.json()

# Prepare timedata for database
if respJson['usedPercentage'] < 100:
    measurement = [
        {
            "measurement": "pass_telekom_de",
            "tags": {
                "host": args.hostname
            },
            "fields": {
                "initialVolume": respJson['initialVolume'],
                "usedVolume": respJson['usedVolume'],
                "remainingSeconds": respJson['remainingSeconds'],
                "usedAtMillis": respJson['usedAt']
            }
        }
    ]
else:
    measurement = [
        {
            "measurement": "pass_telekom_de",
            "tags": {
                "host": args.hostname
            },
            "fields": {
                "remainingSeconds": respJson['remainingSeconds'],
                "usedAtMillis": respJson['usedAt']
            }
        }
    ]

# Send data to InfluxDB
influx = InfluxDBClient(host=args.influxHost, database=args.database)
influx.write_points(measurement)