Speed Wi-Fi 接続端末からデータ通信量を確認する

from loguru import logger

from selenium import webdriver
from selenium.common.exceptions import TimeoutException
from selenium.webdriver.chrome.options import Options
from selenium.webdriver.common.by import By
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.support.ui import WebDriverWait


class BrowserOperator(object):
    def __init__(self, headless=False):
        options = Options()
        options.headless = headless
        self.driver = webdriver.Chrome(options=options)

    def __del__(self):
        self.driver.close()

    def wait(self, f, until_not=False):
        wt = 10
        try:
            if until_not:
                return WebDriverWait(self.driver, wt).until_not(f)
            return WebDriverWait(self.driver, wt).until(f)
        except TimeoutException:
            logger.error("[timeout]")
            return False

    def get(self, url):
        logger.debug(f"[get] {url}")
        self.driver.get(url)


class SpeedwifiNext(BrowserOperator):
    BASE = "http://speedwifi-next.home/html/"

    def login(self, user: str, pswd: str) -> None:
        self.get(self.BASE + "login.htm")

        self.driver.find_element_by_id("user_type").send_keys(user)
        self.driver.find_element_by_id("input_password").send_keys(pswd)
        self.driver.find_element_by_id("login").click()

        status = self.wait(EC.url_to_be(self.BASE + "status.htm"))
        if status:
            logger.debug("login success")
        else:
            raise Exception("login failed\nPlease check your User and Pass!")

    def statistics(self) -> None:
        self.get(self.BASE + "statistics.htm")
        self.wait(
            EC.text_to_be_present_in_element((By.ID, "label_usedData"), "0 KB"),
            until_not=True,
        )

        def f(x) -> str:
            return self.driver.find_element_by_id(x).get_attribute("innerText")

        print("1ヶ月間：", f("label_usedData"))
        print("前日までの3日間：", f("label_usedData_yesterday"))
        print("本日までの3日間：", f("label_usedData_today"))


if __name__ == "__main__":
    user = ""
    pswd = ""

    sn = SpeedwifiNext(headless=True)
    sn.login(user, pswd)
    sn.statistics()