# 導入數據請求模塊 --> 第三方模塊, 需要安裝 pip install requests
import requests
# 導入正則模塊 --> 內置模塊, 不需要安裝
import re
# 導入自動化模塊 --> 第三方模塊, 需要安裝 pip install selenium==3.141.0 <指定版本安裝>
from selenium import webdriver

"""
1. 打開瀏覽器
    配置瀏覽器驅動:
        大版本一樣, 小版本最相近即可
"""
driver = webdriver.Chrome(r'D:\download\anaconda\chromedriver.exe')
# 2. 訪問網站
driver.get('https://www.jsyks.com/kms-sxlx')
# 設置全屏
driver.maximize_window()
# 讀取答案文本內容, 獲取答案ID <讀取數據>
f = open('答案ID.txt', encoding='utf-8')
# 字符串分割方法: split() --> 字符串數據變成列表
answer_id_list = f.read().split(',')
# for循環(huán)遍歷, 把列表里面元素一個一個提取出來
for answer_id in answer_id_list:
    """
    1. 發(fā)送請求, 模擬瀏覽器對于url地址發(fā)送請求
        請求地址: 答案頁面url
        - 安裝模塊:
            I. 點擊pycharm終端 輸入命令 pip install requests
            II. win + R 輸入cmd 輸入命令 pip install requests
        - 模擬瀏覽器發(fā)送請求
            好處: 防止被反爬
            headers 請求頭 --> 可以復制
        - <Response [200]> 表示響應對象
            Response: 中文意思響應/回復/回答
            <>: 表示對象
            200: 狀態(tài)碼 表示請求成功 --> 相當于你打電話, 打通了有嘟嘟嘟聲音
            404: 鏈接不對 --> 相當于你打電話, 你所撥打的電話是空號

    打亂了 --> 題目 + 答案內容 都保存下來, 然后進行對比
    """
    # 構建答案頁面url --> f'{}' 字符串格式化方法 --> format
    answer_url = f'https://tiba.jsyks.com/Post/{answer_id}.htm'
    # 偽裝模擬
    headers = {
        # User-Agent 表示瀏覽器基本身份標識
        'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/101.0.0.0 Safari/537.36'
    }
    # 發(fā)送請求
    response = requests.get(url=answer_url, headers=headers)
    """
    2. 獲取數據, 獲取頁面網頁源代碼
    3. 解析數據, 提取我們想要答案內容
        正則表達式: 會 1 不會 0
        re.findall('匹配的數據', '什么地方匹配') --> re模塊里面findall方法: 找到所有我們想要的數據內容

        (.*?) --> ()精確匹配 .匹配任意字符(除了\n換行符) *匹配前一個字符0或者N個 ? 非貪婪匹配模式

    網站不同, 數據結構不同 --> 基本上一個代碼對一個網站
    """
    answer = re.findall('答案是：(.*?)。', response.text)[0]
    print(answer)
    if answer == '對':
        answer = 'Y'
    elif answer == '錯':
        answer = 'N'
    """
    選擇正確答案進行點擊操作:
        先定位元素, 然后再進行操作
    find_element_by_css_selector --> 通過css選擇器查找元素

    len() 統(tǒng)計元素個數
        answer --> A 元素只有一個  <單選題>
        answer --> ABD 元素三個  <多選題>
    """
    if len(answer) == 1:
        driver.find_element_by_css_selector(f'#btn{answer}').click()
        # 點擊下一題
        driver.find_element_by_css_selector('#btn_PN span').click()
    elif len(answer) > 1:
        # len(answer) --> 3 num -> 0 / 1 / 2
        for num in range(len(answer)):
            driver.find_element_by_css_selector(f'#btn{answer[num]}').click()
        # 點擊確定選好了的按鈕
        driver.find_element_by_css_selector('#ExamOptDa input').click()
        # 點擊下一題
        driver.find_element_by_css_selector('#btn_PN span').click()