Python火车票抢票

import requests
# 导入漂亮的表格输出模块 ---> 第三方模块 需要 在cmd里面进行 pip install prettytable
import prettytable as pt
# 导入json
import json
# 导入selenium模块 ---> 第三方模块 需要 在cmd里面进行 pip install selenium==3.141.0
from selenium import webdriver
# 导入账号密码
from password import account, Password
# 导入selenium控制键盘
from selenium.webdriver.common.keys import Keys
# 创建浏览器 <打开浏览器>  浏览器驱动放在python安装目录里面就不需要添加路径
 
 
def get_ticket(num, from_station, to_station, date_key):
    driver = webdriver.Chrome()
    # 输入网址
    driver.get('https://kyfw.12306.cn/otn/resources/login.html')
    # 通过元素定位, 找到账号输入框 输入账号 send_keys() 发送或者输入内容
    driver.find_element_by_css_selector('#J-userName').send_keys(account)  # 通过 css 选择器查找元素
    # 通过元素定位, 找到密码输入框 输入密码 send_keys() 发送或者输入内容
    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()
    # 通过元素定位, 点击车票预定
    driver.find_element_by_css_selector('#link_for_ticket').click()
    # 通过元素定位, 点击键盘回车
    driver.find_element_by_css_selector('#fromStationText').send_keys(Keys.ENTER)
    # 通过元素定位, 清空输入框内容
    driver.find_element_by_css_selector('#fromStationText').clear()
    # 通过元素定位, 点击一下输入框
    driver.find_element_by_css_selector('#fromStationText').click()
    # 通过元素定位, 输入出发城市
    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('请输入你要出发城市: ')
to_city = input('请输入你要到达城市: ')
date = input('请输入你要出发的时间: ')
from_station = json_data[from_city]
to_station = json_data[to_city]
 
"""
发送请求, 模拟浏览器对于url地址发送请求
    快捷批量替换方法:
        选中内容, ctrl + R 输入正则表达式命令点击全部替换即可
        (.*?): (.*)
        '$1': '$2',
"""
# 确定请求网址 ---> 如果对于长链接, 我们是可以分段写 ? 后面都是属于请求参数
url = 'https://kyfw.12306.cn/otn/leftTicket/query'
# 请求参数 ---> 字典数据类型, 构建成完整键值对, 键值对与键值对之间要用逗号隔开
data = {
    'leftTicketDTO.train_date': date,
    'leftTicketDTO.from_station': from_station,
    'leftTicketDTO.to_station': to_station,
    'purpose_codes': 'ADULT',
}
# 请求头 ---> 伪装模拟浏览器的, 把python代码伪装成浏览器发送请求
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',
}
# 发送请求 ---> 返回数据 <Response [200]> 响应对象 200状态码表示请求成功
response = requests.get(url=url, params=data, headers=headers)
 
# 获取数据 response.json() 获取响应对象json字典数据
# 解析数据 ---> 根据数据类型以及你想要获取数据内容, 选择最合适提取方式 根据冒号左边的内容, 提取冒号右边内容
result = response.json()['data']['result']
 
# 实例化对象
tb = pt.PrettyTable()
tb.field_names = [
    '序号',
    '车次',
    '出发时间',
    '到达时间',
    '耗时',
    '特等座',
    '一等',
    '二等',
    '软卧',
    '硬卧',
    '硬座',
    '无座',
]
page = 0
lis = []
# for循环 就把列表里面元素一个一个提取出来
for index in result:
    # split() 字符串分割方法 --> 返回是列表
    content_list = index.split('|')
    #  根据列表索引位置提取内容
    num = content_list[3]  # 车次
    start_time = content_list[8]  # 出发时间
    end_time = content_list[9]  # 到达时间
    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,
        '出发时间': start_time,
        '到达时间': 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 对应列表索引位置
    #     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)