12306火车票查询爬虫（基于selenium）

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
import time
import datetime
from selenium.webdriver import ActionChains
from selenium import webdriver

#形参为出发城市，目的地，月份，日。
def train_tickets(scity,ecity,month,day):
    #打开浏览器
    browser = webdriver.Chrome()
    url = ‘https://kyfw.12306.cn/otn/leftTicket/init‘
    browser.get(url)
    #设置隐形等待
    browser.implicitly_wait(15)
    #选择单程
    browser.find_element_by_xpath(‘//label[@for="dc"]‘).click()
    time.sleep(1)
    #定位出发城市和目的地城市的输入框
    start_city = browser.find_element_by_xpath(‘//input[@id="fromStationText"]‘)
    end_city = browser.find_element_by_xpath(‘//input[@id="toStationText"]‘)
    #点击并写入城市。
    ActionChains(browser).click(start_city).send_keys(scity).perform()
    try:
        list_0 = browser.find_elements_by_xpath(‘//div[@id="panel_cities"]//div/span[1]‘)
        for vb in list_0:
            if vb.text == scity:
                time.sleep(3)
                vb.click()
    except:
        pass
    time.sleep(3)

    ActionChains(browser).click(end_city).send_keys(ecity).perform()
    try:
        list_01 = browser.find_elements_by_xpath(‘//div[@id="panel_cities"]//div/span[1]‘)
        for vb2 in list_01:
            if vb2.text == ecity:
                time.sleep(3)
                vb2.click()
    except:
        pass
    time.sleep(3)
    #选定日期
    if month == ‘当月‘:
        browser.find_element_by_xpath(‘//input[@readonly="readonly"]‘).click()
        time.sleep(2)
        qa = browser.find_elements_by_xpath(‘//div[@class="cal-cm"][1]//div[@class="cell"]‘)[int(day)-1]
        ActionChains(browser).click(qa).perform()
        time.sleep(3)
    elif month == ‘下月‘:
        browser.find_element_by_xpath(‘//input[@readonly="readonly"]‘).click()
        time.sleep(2)
        qb = browser.find_elements_by_xpath(‘//div[@class="cal-cm"][2]//div[@class="cell"]‘)[int(day)-1]
        ActionChains(browser).click(qb).perform()
        time.sleep(2)
        #点击查询
    browser.find_element_by_xpath(‘//a[@id="query_ticket"]‘).click()
    time.sleep(3)
    #车次，出发站，到达站点，出发时间，到达时间，耗时。
    che_ci = []
    start_c = []
    end_c = []
    start_t = []
    end_t = []
    time_2 = []
    #获取列车车次，并写入列表。
    che_ci_list = browser.find_elements_by_xpath(‘//tbody[@id="queryLeftTable"]//tr[@style="display:none;"]‘)
    for w in che_ci_list:
        aa = w.get_attribute(‘datatran‘)
        che_ci.append(aa)
    #获取出发站点。
    start_c_list = browser.find_elements_by_xpath(‘//strong[@class="start-s"]‘)
    for w2 in start_c_list:
        start_c.append(w2.text)
    #获取目的地站点
    end_c_list = browser.find_elements_by_xpath(‘//strong[@class="end-s"]‘)
    for w3 in end_c_list:
        end_c.append(w3.text)
    #获取出发时间
    start_t_list = browser.find_elements_by_xpath(‘//strong[@class="start-t"]‘)
    for w4 in start_t_list:
        start_t.append(w4.text)
    #获取到达时间
    end_t_list = browser.find_elements_by_xpath(‘//strong[@class="color999"]‘)
    for w5 in end_t_list:
        end_t.append(w5.text)
    #获取耗时
    time_2_list = browser.find_elements_by_xpath(‘//div[@class="ls"]‘)
    for w6 in time_2_list:
        time_2.append(w6.text)
    #关闭浏览器。
    time.sleep(3)
    browser.quit()
    #返回值
    return che_ci,start_c,end_c,start_t,end_t,time_2

#qaz = train_tickets(scity=‘北京‘,ecity=‘上海‘,month=‘当月‘,day=‘28‘)
#print(qaz)
#创建一个打印所有信息的函数。
def print_info(tuple):
    for aaa,bbb,ccc,ddd,eee,fff in zip(tuple[0],tuple[1],tuple[2],tuple[3],tuple[4],tuple[5]):
        print(‘车次： %s  出发站点：%s  到达站点：%s  出发时间：%s  到达时间：%s  总耗时：%s  ‘ % (aaa,bbb,ccc,ddd,eee,fff))
    return True
#把函数封装。
def search_train_tickets():
    time_1 = datetime.datetime.now()
    scity = input(‘请输入你的出发城市站点或城市：‘)
    ecity = input(‘请输入你的目的地城市站点或城市：‘)
    month = input(‘请输入出发月份（余票只能查询即日起30天的票，输入‘当月’或‘下月’）:‘)
    day = input(‘请输入你几号出发：‘)
    num_1 = train_tickets(scity=scity,ecity=ecity,month=month,day=day)
    print_info(num_1)
    time_2 = datetime.datetime.now()
    cha = (time_2 - time_1).seconds
    print(‘\n\n此次运行耗时%s秒。‘ % cha)

#运行程序
search_train_tickets()

import time
import datetime
from selenium.webdriver import ActionChains
from selenium import webdriver

def train_tickets(scity,ecity,month,day):
    #打开浏览器
    browser = webdriver.Chrome()
    url = ‘https://kyfw.12306.cn/otn/leftTicket/init‘
    browser.get(url)
 
    #设置隐形等待
    browser.implicitly_wait(15)

#选择单程
    browser.find_element_by_xpath(‘//label[@for="dc"]‘).click()
    time.sleep(1)

#运行程序
search_train_tickets()