62 lines
1.7 KiB
Python
62 lines
1.7 KiB
Python
from flask import Flask, request
|
|
from reconnect import reboot
|
|
from os import environ
|
|
from subprocess import call
|
|
|
|
app = Flask(__name__)
|
|
|
|
# Get router username and password from environment variables
|
|
username = environ.get('ROUTER_USERNAME')
|
|
password = environ.get('ROUTER_PASSWORD')
|
|
|
|
# Get router IP and path from environment variables if they exist
|
|
url = environ.get('ROUTER_IP') or '192.168.1.1'
|
|
path = environ.get('ROUTER_PATH') or '/ws'
|
|
|
|
@app.route('/', methods=['POST'])
|
|
def result():
|
|
print(request.json)
|
|
return 'Received request, printing to console!'
|
|
|
|
@app.route('/safe', methods=['POST'])
|
|
def safe():
|
|
# Try to ping the router 100 times
|
|
for i in range(50):
|
|
if call(['ping', '-c', '1', url]) == 0:
|
|
break
|
|
else:
|
|
return 'Error: Unable to ping and reboot router automatically!'
|
|
|
|
# Try to nslookup 5 different websites
|
|
websites = ['www.google.com', 'www.facebook.com', 'www.twitter.com', 'www.reddit.com', 'www.amazon.com']
|
|
success = 0
|
|
|
|
for website in websites:
|
|
if call(['nslookup', website]) == 0:
|
|
success += 1
|
|
|
|
if success > 3:
|
|
return 'Error: DNS seems to be working, debug manually!'
|
|
|
|
# Reboot router
|
|
status = reboot(url, path, username, password).text
|
|
|
|
if '{"status":true}' not in status:
|
|
return 'Error: Something went wrong while rebooting router!'
|
|
|
|
return 'Success: Rebooted router!'
|
|
|
|
@app.route('/force', methods=['POST'])
|
|
def force():
|
|
status = reboot(url, path, username, password).text
|
|
|
|
if '{"status":true}' not in status:
|
|
return 'Error: Something went wrong while rebooting router!'
|
|
|
|
return 'Success: Rebooted router!'
|
|
|
|
@app.route('/grafana', methods=['POST'])
|
|
def grafana():
|
|
if request.json['state'] == 'alerting':
|
|
safe()
|