# 導(dǎo)入數(shù)據(jù)請(qǐng)求模塊 --> 第三方模塊, 需要安裝 pip install requests
import requests
# 導(dǎo)入正則模塊 --> 內(nèi)置模塊, 不需要安裝
import re
# 導(dǎo)入自動(dòng)化模塊 --> 第三方模塊, 需要安裝 pip install selenium==3.141.0 <指定版本安裝>
from selenium import webdriver

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

    打亂了 --> 題目 + 答案內(nèi)容 都保存下來(lái), 然后進(jìn)行對(duì)比
    """
    # 構(gòu)建答案頁(yè)面url --> f'{}' 字符串格式化方法 --> format
    answer_url = f'https://tiba.jsyks.com/Post/{answer_id}.htm'
    # 偽裝模擬
    headers = {
        # User-Agent 表示瀏覽器基本身份標(biāo)識(shí)
        '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ā)送請(qǐng)求
    response = requests.get(url=answer_url, headers=headers)
    """
    2. 獲取數(shù)據(jù), 獲取頁(yè)面網(wǎng)頁(yè)源代碼
    3. 解析數(shù)據(jù), 提取我們想要答案內(nèi)容
        正則表達(dá)式: 會(huì) 1 不會(huì) 0
        re.findall('匹配的數(shù)據(jù)', '什么地方匹配') --> re模塊里面findall方法: 找到所有我們想要的數(shù)據(jù)內(nèi)容

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

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

    len() 統(tǒng)計(jì)元素個(gè)數(shù)
        answer --> A 元素只有一個(gè)  <單選題>
        answer --> ABD 元素三個(gè)  <多選題>
    """
    if len(answer) == 1:
        driver.find_element_by_css_selector(f'#btn{answer}').click()
        # 點(diǎn)擊下一題
        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()
        # 點(diǎn)擊確定選好了的按鈕
        driver.find_element_by_css_selector('#ExamOptDa input').click()
        # 點(diǎn)擊下一題
        driver.find_element_by_css_selector('#btn_PN span').click()