無線ルータAtermを再起動するPythonスクリプト

#!/usr/bin/env python3

# reboot_aterm.py (Ver.20220118)

import sys
import time
from selenium import webdriver
from selenium.common.exceptions import TimeoutException
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.common.by import By
from selenium.webdriver.common.alert import Alert

username = 'admin'
password = 'Atermの管理者パスワード'
url = 'http://'+username+':'+password+'@192.168.10.1/index.cgi/reboot_main.html'

timeout = 10
options = webdriver.ChromeOptions()
options.add_argument('--headless')
options.add_argument('--disable-gpu')
driver = webdriver.Chrome(options=options)

def wait_element(attribute, target):
	try:
		if attribute == 'ID':
			element_present = EC.presence_of_element_located((By.ID, target))
		elif attribute == 'CLASS_NAME':
			element_present = EC.presence_of_element_located((By.CLASS_NAME, target))
		elif attribute == 'XPATH':
			element_present = EC.presence_of_element_located((By.XPATH, target))
		elif attribute == 'CSS_SELECTOR':
			element_present = EC.presence_of_element_located((By.CSS_SELECTOR, target))
		WebDriverWait(driver, timeout).until(element_present)
	except TimeoutException:
		print('ERROR: Timed out waiting for page to load. %s = %s' % (attribute, target), file=sys.stderr)
		driver.quit()
		sys.exit(1)
	else:
		return True

driver.get(url)
wait_element('CSS_SELECTOR', '.btn.blue') # 「再起動」ボタン
driver.find_element_by_css_selector('.btn.blue').click()
time.sleep(2)
Alert(driver).accept()
time.sleep(5)
driver.quit()