반응형

파이썬, 셀레니움 경고창 및 새 탭(창) 닫기

 

글. 수알치 오상문

 

    def close_alert(self, n=3):
        """경고창을 닫는다

        Args:
            n (str): 닫을 개수 (기본 3개)
        """

        for i in range(n):
            try:
                self.driver.switch_to.alert.accept()
            except Exception:
                return
            time.sleep(0.1)

    def close_new_tabs(self):
        """ 기본 탭(창)을 제외한 새탭/팝업창을 모두 닫는다
        """

        tabs = self.driver.window_handles
        while len(tabs) != 1:
            self.driver.switch_to.window(tabs[1])
            self.driver.close()
            tabs = self.driver.window_handles
        self.driver.switch_to.window(tabs[0])

 

반응형

+ Recent posts