Untitled

from selenium import webdriver
from selenium.webdriver.remote.webdriver import WebDriver
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import Select, WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.common.keys import Keys
import time

def set_up():
    driver = webdriver.Chrome()
    driver.get("https://congchuc.quangninh.gov.vn/sso/Login.aspx")
    driver.maximize_window()
    return driver

def login(driver):
    input_user = driver.find_element(By.ID,'IDToken1')
    input_pass = driver.find_element(By.ID,'IDToken2')
    input_user.send_keys("nguyenviethoang2")
    input_pass.send_keys("hoangnhung12", Keys.ENTER)

def check_all(driver):
    wait = WebDriverWait(driver, 10)
    driver.get('https://congchuc.quangninh.gov.vn/Default.aspx?tabid=1126')
    vanbanquamang = driver.find_element(By.XPATH,"//span[text()='Văn bản qua mạng']")
    vanbanquamang.click()
    time.sleep(2)
    btn_check_all = wait.until(EC.presence_of_element_located((By.XPATH,'/html/body/form/div[3]/div[1]/div[3]/div/div/div/div/div/div/div/div[1]/div/div[4]/div[2]/div[3]/div/div/div/table/thead/tr[2]/th[1]/input')))
    btn_check_all.click()
    btn_vaoso = driver.find_element(By.XPATH,'/html/body/form/div[3]/div[1]/div[3]/div/div/div/div/div/div/div/div[1]/div/div[4]/div[2]/div[3]/div/div/div/table/thead/tr[1]/td/div/span[3]/label')
    btn_vaoso.click()
    btn_ok = wait.until(EC.presence_of_element_located((By.XPATH,'//span[@class="rwInnerSpan" and text()="OK"]')))
    btn_ok.click()
    time.sleep(1)
    vanbandaxuly = driver.find_element(By.XPATH,"//span[text()='Đã xử lý']")
    vanbandaxuly.click()
    time.sleep(2)
    # btn_check_all_1 = wait.until(EC.presence_of_element_located((By.XPATH,'/html/body/form/div[4]/div[1]/div[3]/div/div/div/div/div/div/div/div[1]/div/div[4]/div[2]/div[2]/div/div[2]/div/table/thead/tr[2]/th[1]/input')))
    # btn_check_all_1.click()
    btn_download = driver.find_elements(By.XPATH,'//input[@title="Tải tất cả file"]')
    print(len(btn_download))
    
def main():
    driver = set_up()
    login(driver)
    check_all(driver)
    driver.quit()

if __name__ == "__main__":
    main()