linspector/uplink
2
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Added threading to cron and foreground execution mode

Browse source
This commit is contained in:
Johannes Findeisen 2021-12-27 16:55:54 +01:00
commit 6d0ef7b137
3 changed files with 16 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

6
config.example.json
View file

@ -1,12 +1,12 @@
{
"interval": 60,
"database_type": "mariadb",
"database": "uplink",
"database_host": "127.0.0.1",
"database_user": "USER",
"database_password": "PASSWORD",
"cron": false,
"daemon": false,
"log_level": "",
"mode": "cron",
"log_level": "ERROR",
"uplinks": [
{ "provider": "Cable Provider", "ip": "192.168.0.1", "password": "1234" },
{ "provider": "DSL Provider", "ip": "192.168.1.1", "password": "1234" }

20
uplink.py
View file

@ -27,10 +27,11 @@ import json
import sys
import time
from threading import Thread
from uplink.uplink import Uplink
# Version format: MAJOR.FEATURE.FIXES
__version__ = "0.4.1-development"
__version__ = "0.4.2-development"
# TODO: CHECK ALL ERROR HANDLING!!!
# TODO: Implement logging
@ -43,7 +44,8 @@ __version__ = "0.4.1-development"
def parse_args():
parser = argparse.ArgumentParser(
description="uplink is a tool to monitor the link status of AVM FRITZ!Box Cable and DSL based routers.",
description="uplink is a tool to monitor the link status of AVM FRITZ!Box Cable and DSL "
"based routers.",
epilog="uplink is not some program expecting uplinks to work!",
prog="uplink")
@ -59,8 +61,9 @@ def parse_args():
mode.add_argument("-f", "--foreground", default=False, dest="foreground", action="store_true",
help="run looped in foreground (default: false)")
parser.add_argument("-i", "--interval", type=int, help="poll interval in seconds. this overrides config file "
"settings. (default: 60)")
parser.add_argument("-i", "--interval", type=int, help="poll interval in seconds. this "
"overrides config file settings. "
"(default: 60)")
"""
parser.add_argument("-b", "--database", metavar="DATABASE", help="database to use")
@ -112,7 +115,7 @@ if __name__ == "__main__":
config = json.loads(config_data)
except Exception as err:
# TODO: Replace all lines like this with generic Python logging
print(str("Uplink: Configuration Error!" + str(err)))
print(str("uplink: Configuration Error! " + str(err)))
sys.exit(1)
try:
@ -129,14 +132,15 @@ if __name__ == "__main__":
if args.cron:
for i in range(len(config["uplinks"])):
uplink.get_data(config, i)
t = Thread(target=uplink.get_data, args=(config, i))
t.start()
elif args.daemon:
uplink.start()
elif args.foreground:
while True:
for i in range(len(config["uplinks"])):
uplink.get_data(config, i)
# TODO: print data formatted to STDOUT
t = Thread(target=uplink.get_data, args=(config, i))
t.start()
try:
time.sleep(config["interval"])
except KeyboardInterrupt as err:

2
uplink/uplink.py
View file

@ -85,7 +85,7 @@ class Uplink(Daemon):
status = "DOWN"
# TODO: Replace all lines like this with generic Python logging
print(str("[" + str(d) + " " + str(t) + "] " +config["uplinks"][inc]["provider"] + " " + status))
print(str("[" + str(d) + " " + str(t) + "] " + config["uplinks"][inc]["provider"] + " " + status))
# TODO: Fix to long lines and make the SQL statement more readable
sql = 'INSERT INTO log (timestamp, date, time, uptime, internal_ip, external_ip, external_ipv6, is_linked, ' \