import requests
# 導(dǎo)入漂亮的表格輸出模塊 ---> 第三方模塊 需要 在cmd里面進(jìn)行 pip install prettytable
import prettytable as pt
# 導(dǎo)入json
import json
# 導(dǎo)入selenium模塊 ---> 第三方模塊 需要 在cmd里面進(jìn)行 pip install selenium==3.141.0
from selenium import webdriver
# 導(dǎo)入賬號密碼
from password import account, Password
# 導(dǎo)入selenium控制鍵盤
from selenium.webdriver.common.keys import Keys
# 創(chuàng)建瀏覽器 <打開瀏覽器>  瀏覽器驅(qū)動放在python安裝目錄里面就不需要添加路徑
 
 
def get_ticket(num, from_station, to_station, date_key):
    driver = webdriver.Chrome()
    # 輸入網(wǎng)址
    driver.get('https://kyfw.12306.cn/otn/resources/login.html')
    # 通過元素定位, 找到賬號輸入框 輸入賬號 send_keys() 發(fā)送或者輸入內(nèi)容
    driver.find_element_by_css_selector('#J-userName').send_keys(account)  # 通過 css 選擇器查找元素
    # 通過元素定位, 找到密碼輸入框 輸入密碼 send_keys() 發(fā)送或者輸入內(nèi)容
    driver.find_element_by_css_selector('#J-password').send_keys(Password)  # 通過 css 選擇器查找元素
    # 通過元素定位, 找到登陸按鈕, 點擊登陸
    driver.find_element_by_css_selector('#J-login').click()
    # 隱式等待
    driver.implicitly_wait(10)
    # 通過元素定位, 點擊彈窗確定按鈕 .表示class # 表示ID
    driver.find_element_by_css_selector('.modal-ft .btn').click()
    # 通過元素定位, 點擊車票預(yù)定
    driver.find_element_by_css_selector('#link_for_ticket').click()
    # 通過元素定位, 點擊鍵盤回車
    driver.find_element_by_css_selector('#fromStationText').send_keys(Keys.ENTER)
    # 通過元素定位, 清空輸入框內(nèi)容
    driver.find_element_by_css_selector('#fromStationText').clear()
    # 通過元素定位, 點擊一下輸入框
    driver.find_element_by_css_selector('#fromStationText').click()
    # 通過元素定位, 輸入出發(fā)城市
    driver.find_element_by_css_selector('#fromStationText').send_keys(from_station)
    # 通過元素定位, 點擊鍵盤回車
    driver.find_element_by_css_selector('#fromStationText').send_keys(Keys.ENTER)
    driver.find_element_by_css_selector('#toStationText').clear()
    driver.find_element_by_css_selector('#toStationText').click()
    driver.find_element_by_css_selector('#toStationText').send_keys(to_station)
    driver.find_element_by_css_selector('#toStationText').send_keys(Keys.ENTER)
    date = driver.find_element_by_css_selector('.inp-w #train_date')
    date.clear()
    date.send_keys(date_key)
    # 通過元素定位, 點擊查詢按鈕
    driver.find_element_by_css_selector('#query_ticket').click()
    if num == 1:
        driver.find_element_by_css_selector(f'#queryLeftTable tr:nth-child({num}) a.btn72').click()
    elif num != 1:
        driver.find_element_by_css_selector(f'#queryLeftTable tr:nth-child({num + 1}) a.btn72').click()
        # #normalPassenger_0
    driver.find_element_by_css_selector('#normalPassenger_1').click()
 
    driver.find_element_by_css_selector('#submitOrder_id').click()
    driver.find_element_by_css_selector('#qr_submit_id').click()
    driver.find_element_by_css_selector('#qr_submit_id').click()
 
 
f = open('city.json', encoding='utf-8')
text = f.read()  # 讀取文件
json_data = json.loads(text)
from_city = input('請輸入你要出發(fā)城市: ')
to_city = input('請輸入你要到達(dá)城市: ')
date = input('請輸入你要出發(fā)的時間: ')
from_station = json_data[from_city]
to_station = json_data[to_city]
 
"""
發(fā)送請求, 模擬瀏覽器對于url地址發(fā)送請求
    快捷批量替換方法:
        選中內(nèi)容, ctrl + R 輸入正則表達(dá)式命令點擊全部替換即可
        (.*?): (.*)
        '$1': '$2',
"""
# 確定請求網(wǎng)址 ---> 如果對于長鏈接, 我們是可以分段寫 ? 后面都是屬于請求參數(shù)
url = 'https://kyfw.12306.cn/otn/leftTicket/query'
# 請求參數(shù) ---> 字典數(shù)據(jù)類型, 構(gòu)建成完整鍵值對, 鍵值對與鍵值對之間要用逗號隔開
data = {
    'leftTicketDTO.train_date': date,
    'leftTicketDTO.from_station': from_station,
    'leftTicketDTO.to_station': to_station,
    'purpose_codes': 'ADULT',
}
# 請求頭 ---> 偽裝模擬瀏覽器的, 把python代碼偽裝成瀏覽器發(fā)送請求
headers = {
    # User-Agent: 用戶代理, 表示瀏覽器基本身份標(biāo)識
    '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ā)送請求 ---> 返回數(shù)據(jù) <Response [200]> 響應(yīng)對象 200狀態(tài)碼表示請求成功
response = requests.get(url=url, params=data, headers=headers)
 
# 獲取數(shù)據(jù) response.json() 獲取響應(yīng)對象json字典數(shù)據(jù)
# 解析數(shù)據(jù) ---> 根據(jù)數(shù)據(jù)類型以及你想要獲取數(shù)據(jù)內(nèi)容, 選擇最合適提取方式 根據(jù)冒號左邊的內(nèi)容, 提取冒號右邊內(nèi)容
result = response.json()['data']['result']
 
# 實例化對象
tb = pt.PrettyTable()
tb.field_names = [
    '序號',
    '車次',
    '出發(fā)時間',
    '到達(dá)時間',
    '耗時',
    '特等座',
    '一等',
    '二等',
    '軟臥',
    '硬臥',
    '硬座',
    '無座',
]
page = 0
lis = []
# for循環(huán) 就把列表里面元素一個一個提取出來
for index in result:
    # split() 字符串分割方法 --> 返回是列表
    content_list = index.split('|')
    #  根據(jù)列表索引位置提取內(nèi)容
    num = content_list[3]  # 車次
    start_time = content_list[8]  # 出發(fā)時間
    end_time = content_list[9]  # 到達(dá)時間
    use_time = content_list[10]  # 耗時
    topGrade = content_list[25]  # 特等座
    if topGrade:
        pass
    else:
        topGrade = content_list[32]
    first_class = content_list[31]  # 一等
    second_class = content_list[30]  # 二等
    hard_sleeper = content_list[28]  # 硬臥
    hard_seat = content_list[29]  # 硬座
    no_seat = content_list[26]  # 無座
    soft_sleeper = content_list[23]  # 軟臥
    dit = {
        '車次': num,
        '出發(fā)時間': start_time,
        '到達(dá)時間': end_time,
        '耗時': use_time,
        '特等座': topGrade,
        '一等': first_class,
        '二等': second_class,
        '軟臥': soft_sleeper,
        '硬臥': hard_sleeper,
        '硬座': hard_seat,
        '無座': no_seat,
    }
    lis.append(dit)
    tb.add_row([
        page,
        num,
        start_time,
        end_time,
        use_time,
        topGrade,
        first_class,
        second_class,
        soft_sleeper,
        hard_sleeper,
        hard_seat,
        no_seat,
    ])
    page += 1
    # 更加方便直觀查看 索引位置 ---> 列表可以通過索引位置取值
    # num = 0
    # for i in content_list:
    #     # i 列表里面元素, num 對應(yīng)列表索引位置
    #     print(i, num)
    #     num += 1
    # break
print(tb)
 
 
num = input('請輸入你想要購買車票序號: ')
 
get_ticket(num=int(num), from_station=from_city, to_station=to_city, date_key=date)