98 lines
2.9 KiB
Python
Executable File
98 lines
2.9 KiB
Python
Executable File
#!/usr/bin/env -S uv run --script
|
|
# /// script
|
|
# dependencies = [
|
|
# "playwright>=1.45.0",
|
|
# ]
|
|
# ///
|
|
|
|
from __future__ import annotations
|
|
|
|
import argparse
|
|
import sys
|
|
import time
|
|
|
|
from playwright.sync_api import Playwright, TimeoutError as PlaywrightTimeoutError, sync_playwright
|
|
|
|
ROUTER_URL = "http://192.168.1.1"
|
|
ROUTER_USERNAME = "abdus"
|
|
ROUTER_PASSWORD = "xAsametk50"
|
|
|
|
|
|
def reboot_router(playwright: Playwright, headed: bool, timeout_ms: int) -> None:
|
|
browser = playwright.chromium.launch(headless=not headed)
|
|
context = browser.new_context(ignore_https_errors=True)
|
|
page = context.new_page()
|
|
page.set_default_timeout(timeout_ms)
|
|
|
|
try:
|
|
page.goto(f"{ROUTER_URL}/login", wait_until="domcontentloaded")
|
|
|
|
username_box = page.get_by_role("textbox", name="User Name")
|
|
password_box = page.get_by_role("textbox", name="Password")
|
|
|
|
username_box.click()
|
|
username_box.press("ControlOrMeta+a")
|
|
username_box.press("Backspace")
|
|
username_box.type(ROUTER_USERNAME, delay=80)
|
|
|
|
password_box.click()
|
|
password_box.press("ControlOrMeta+a")
|
|
password_box.press("Backspace")
|
|
password_box.type(ROUTER_PASSWORD, delay=140)
|
|
page.get_by_role("textbox", name="Password").press("Enter")
|
|
|
|
page.wait_for_load_state("domcontentloaded")
|
|
|
|
try:
|
|
page.get_by_text("The username or password is not correct", exact=False).wait_for(
|
|
state="visible", timeout=1500
|
|
)
|
|
raise RuntimeError("Router rejected configured credentials")
|
|
except PlaywrightTimeoutError:
|
|
pass
|
|
|
|
menu_trigger = page.locator("#h_menu_list").first
|
|
if menu_trigger.count() and menu_trigger.is_visible():
|
|
menu_trigger.click()
|
|
|
|
restart_icon = page.locator("#navbar_reboot .icon-menu-restart, #navbar_reboot").first
|
|
restart_icon.wait_for(state="visible")
|
|
|
|
with page.expect_response(lambda r: "/cgi-bin/Reboot" in r.url, timeout=timeout_ms):
|
|
restart_icon.click()
|
|
page.get_by_role("button", name="OK").click()
|
|
|
|
time.sleep(5)
|
|
|
|
print("Router reboot request sent.")
|
|
finally:
|
|
context.close()
|
|
browser.close()
|
|
|
|
|
|
def parse_args() -> argparse.Namespace:
|
|
parser = argparse.ArgumentParser(description="Log in and reboot router")
|
|
parser.add_argument("--headed", action="store_true", help="Run with visible browser")
|
|
parser.add_argument(
|
|
"--timeout",
|
|
type=int,
|
|
default=60000,
|
|
help="Timeout in milliseconds (default: 60000)",
|
|
)
|
|
return parser.parse_args()
|
|
|
|
|
|
def main() -> int:
|
|
args = parse_args()
|
|
try:
|
|
with sync_playwright() as playwright:
|
|
reboot_router(playwright, headed=args.headed, timeout_ms=args.timeout)
|
|
return 0
|
|
except Exception as exc:
|
|
print(f"Error: {exc}", file=sys.stderr)
|
|
return 1
|
|
|
|
|
|
if __name__ == "__main__":
|
|
raise SystemExit(main())
|