68 lines
2 KiB
Python
Executable file
68 lines
2 KiB
Python
Executable file
#!/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)
|