78 lines
2.6 KiB
Python
78 lines
2.6 KiB
Python
import os
|
|
from time import sleep
|
|
from datetime import datetime
|
|
from subprocess import Popen, DEVNULL
|
|
|
|
from selenium import webdriver
|
|
from selenium.webdriver.common.by import By
|
|
from selenium.webdriver.common.keys import Keys
|
|
|
|
|
|
class Router:
|
|
def __init__(self, ip, password):
|
|
self.ip = ip
|
|
self.password = password
|
|
|
|
def _ping(self, ip, count=3, timeout=3):
|
|
process = Popen(['ping', '-c', str(count), '-w', str(timeout), ip], stdout=DEVNULL, stderr=DEVNULL)
|
|
exit_code = process.wait()
|
|
return exit_code == 0
|
|
|
|
def check_lan_connectivity(self):
|
|
return self._ping(self.ip)
|
|
|
|
def check_internet_connectivity(self):
|
|
return self._ping('8.8.8.8')
|
|
|
|
def reboot(self):
|
|
options = webdriver.FirefoxOptions()
|
|
options.headless = True
|
|
|
|
with webdriver.Firefox(options=options) as driver:
|
|
driver.get(f'http://{self.ip}/webpages/login.html')
|
|
sleep(10)
|
|
|
|
password_input = driver.find_elements(By.CSS_SELECTOR, 'input.l[type=password]')[2]
|
|
password_input.send_keys(self.password)
|
|
sleep(1)
|
|
password_input.send_keys(Keys.ENTER)
|
|
sleep(10)
|
|
|
|
driver.find_element(By.CSS_SELECTOR, 'a#top-control-reboot').click()
|
|
sleep(1)
|
|
driver.find_element(By.CSS_SELECTOR, '#reboot_confirm_msg button.btn-msg-ok').click()
|
|
sleep(1)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
router = Router(os.environ['ROUTER_IP'], os.environ['ROUTER_PASSWORD'])
|
|
|
|
consecutive_internet_downtime = 0
|
|
while True:
|
|
sleep(60)
|
|
|
|
if not router.check_internet_connectivity():
|
|
# Internet is down
|
|
if router.check_lan_connectivity():
|
|
# LAN is up
|
|
if consecutive_internet_downtime >= 2:
|
|
with open('/logs/connectivity.log', 'a') as log:
|
|
log.write(f'{datetime.now()} - Internet down, rebooting router\n')
|
|
|
|
consecutive_internet_downtime = 0
|
|
router.reboot()
|
|
sleep(120)
|
|
else:
|
|
consecutive_internet_downtime += 1
|
|
with open('/logs/connectivity.log', 'a') as log:
|
|
log.write(f'{datetime.now()} - Internet down\n')
|
|
|
|
else:
|
|
# LAN is down
|
|
with open('/logs/connectivity.log', 'a') as log:
|
|
log.write(f'{datetime.now()} - Internet and LAN down\n')
|
|
|
|
consecutive_internet_downtime = 0
|
|
else:
|
|
consecutive_internet_downtime = 0
|