python 用selenuim判斷頁面是否全部加載完成,並且加上最大時長,超過時長報錯

STR_READY_STATE = ''
time_start = time.time()
while STR_READY_STATE != 'complete':
     time.sleep(0.001)
     STR_READY_STATE = driver.execute_script('return document.readyState')
     time_end = time.time()
     if int(time_end - time_start) > 60:
        raise MyException("hpl 頁面打開 超過60s")

 

 

---

 

異常類

class MyException(Exception):

    def __init__(self, *args):
        self.args = args