#ticket.py
import openpyxl
from selenium import webdriver
from selenium.webdriver.support.ui import WebDriverWait #显示等待
from selenium.webdriver.support import expected_conditions as ec #等待的条件
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import Select
from selenium.common.exceptions import NoSuchElementException,ElementNotVisibleException
driver=webdriver.Chrome() #创建浏览器对象
class TrainSpider(object):
#定义类属性
login_url='https://kyfw.12306.cn/otn/resources/login.html' #登录的页面
profile_url='https://kyfw.12306.cn/otn/view/index.html' #个人中心的网址
left_ticket='https://kyfw.12306.cn/otn/leftTicket/init?linktypeid=dc' #余票查询页面
confirm_url='https://kyfw.12306.cn/otn/confirmPassenger/initDc' #确认乘车人和座席
#定义init初始化方法
def __init__(self,from_station,to_station,train_date,trains,passenger_name_lst):
self.from_station=from_station
self.to_station=to_station
self.train_date=train_date
self.station_code=self.init_station_code() #selft.station_code结果为dict
self.trains=trains
self.passenger_name_lst=passenger_name_lst
self.selected_no=None
self.selected_seat=None
#登录的方法
def login(self):
# 打开登录的页面
driver.get(self.login_url)
WebDriverWait(driver,1000).until(
ec.url_to_be(self.profile_url) #等待直到URL成为个人中心页面
)
print('登录成功')
#查询余票
def search_ticket(self):
# 打开查询余票的网站
driver.get(self.left_ticket)
#找到出发站到达站的 隐藏的HTML标签
from_station_input=driver.find_element_by_id('fromStation')
to_station_input=driver.find_element_by_id('toStation')
#找到出发时间的的input标签
train_date_input=driver.find_element_by_id('train_date')
#根据key获取value
from_station_code=self.station_code[self.from_station] #根据出发地找到出发地的代号
to_station_code=self.station_code[self.to_station] #根据目的地找到目的地的代码
#执行js代码
driver.execute_script('arguments[0].value="%s"' % from_station_code,from_station_input)
driver.execute_script('arguments[0].value="%s"' % to_station_code,to_station_input)
driver.execute_script('arguments[0].value="%s"' % self.train_date, train_date_input)
#执行点击查询按钮，执行查询操作
query_ticket_tag=driver.find_element_by_id('query_ticket')
query_ticket_tag.click()
#解析车次，显示等待，等待tbody的出现
WebDriverWait(driver,1000).until(
ec.presence_of_element_located((By.XPATH,'//tbody[@id="queryLeftTable"]/tr'))
)
#筛选出有数据的tr，去掉属性为datatran的tr
trains=driver.find_elements_by_xpath('//tbody[@id="queryLeftTable"]/tr[not(@datatran)]')
is_flag=False #标记是否有余票 ，没有余票为False，有余票为True
while True:
# 分别遍历每个车次
for train in trains:
# print(train.text)
infos = train.text.replace('\n', ' ').split(' ')
# print(infos)
train_no = infos[0] # 列表中索引为0的为车次
if train_no in self.trains:
# 根据key获取值 席别是一个列表
seat_types = self.trains[train_no]
for seat_type in seat_types: # 遍历席位的列表
if seat_type == 'O': # 说明是二等座
count = infos[9] # 索引为9的是二等座
if count.isdigit() or count == '有':
is_flag = True
break # 退出了遍历席位的循环
elif seat_type == 'M': # 说明是一等座
count = infos[8] # 索引为8的是一等座
if count.isdigit() or count == '有':
is_flag = True
break
# 判断是否有余票
if is_flag:
self.selected_no = train_no
# 有票就可以执行单击预订
order_btn = train.find_element_by_xpath('.//a[@class="btn72"]')
order_btn.click()
return # 退出的是遍历车次的循环
def confirm(self):
#等待来到确认乘车人界面
WebDriverWait(driver,1000).until(
ec.url_to_be(self.confirm_url)
)
#等待乘车人标签显示出来
WebDriverWait(driver,1000).until(
ec.presence_of_element_located((By.XPATH,'//ul[@id="normal_passenger_id"]/li/label'))
)
#获取所有的乘车人
passengers=driver.find_elements_by_xpath('//ul[@id="normal_passenger_id"]/li/label')
for passenger in passengers: #分别遍历每一位乘车人 (label标签)
name=passenger.text #获取乘车人的姓名
if name in self.passenger_name_lst:
passenger.click() #label标签的单击
#确认席位
seat_select=Select(driver.find_element_by_id('seatType_1'))
seat_types=self.trains[self.selected_no] #根据key获取值 self.trains是要抢票的车次的字典，self.selected_no 要抢票的车次的key
for seat_type in seat_types:
try:
seat_select.select_by_value(seat_type)
self.selected_seat=seat_type #记录有票的座席
except NoSuchElementException:
continue
else:
break
WebDriverWait(driver,1000).until(
ec.element_to_be_clickable((By.ID,'submitOrder_id')) #等待到a标签可以单击的时候
)
#提交订单
submit_btn=driver.find_element_by_id('submitOrder_id')
submit_btn.click()
#显示等待，等待到模式对话框窗口出现
WebDriverWait(driver,1000).until(
ec.presence_of_element_located((By.CLASS_NAME,'dhtmlx_window_active'))
)
#等待按钮可以点击
WebDriverWait(driver,1000).until(
ec.element_to_be_clickable((By.ID,'qr_submit_id'))
)
submit_btn=driver.find_element_by_id('qr_submit_id')
while submit_btn:
try:
submit_btn.click()
submit_btn = driver.find_element_by_id('qr_submit_id')
except ElementNotVisibleException:
break
print(f'恭喜{self.selected_no}的{self.selected_seat}抢票成功!!!')
#负责调用其它方法（组织其它的代码）
def run(self):
# 1.登录
self.login()
# 2.余票查询
self.search_ticket()
# 3.确认乘车人和订单
self.confirm()
def init_station_code(self):
wb=openpyxl.load_workbook('车站代码.xlsx')
ws=wb.active #使用活动表
lst=[] #存储所有车站名称及代号
for row in ws.rows: #遍历所有行
sub_lst=[] #用于存储每行中的车站名称，车站代号
for cell in row: #遍历一行中的单元格
sub_lst.append(cell.value)
lst.append(sub_lst)
#print(dict(lst)) #将列表转成字典类型
return dict(lst)
#启动爬虫程序
def start(): # O 表示的是二等座,M 代表是一等座
spider=TrainSpider('长春','北京','2020-11-28',{'D30':['O','M']},['杨淑娟'])
spider.run()
#spider.init_station_code()
if __name__ == '__main__':
start()
#demo.py
import requests
import re
import openpyxl
def get_station():
url='https://kyfw.12306.cn/otn/resources/js/framework/station_name.js?station_version=1.9151'
headers={'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/80.0.3987.100 Safari/537.36'}
resp=requests.get(url,headers=headers)
resp.encoding='utf-8'
#print(resp.text)
stations=re.findall('([\u4e00-\u9fa5]+)\|([A-Z]+)',resp.text)
#print(stations)
return stations #stations是一个列表
def save(lst):
wb=openpyxl.Workbook() #创建工作薄对象
ws=wb.active #使用活动表
for item in lst:
ws.append(item) #每执行一次append将向sheet中添加一行
wb.save('车站代码.xlsx')
if __name__ == '__main__':
lst=get_station()
save(lst)