python之12306自动查票

时间:2024-04-26 00:05:09
python之12306自动查票

一、导读

本篇文章所采用的技术仅用于学习、研究,任何其他用途请自行承担后果。

12306自动查票使用到的python库主要是splinter,同时也涉及到查票的城市编码,具体的城市编码请在网络上搜索,基本格式如下:

北京北:VAP

北京东:BOP

北京: BJP

北京南:VNP

北京西:BXP

实现的功能包括:
(1)自动打开Google浏览器,进入12306登录页面
(2)命令行手动输入账号、密码、出发、目的地,时间等相关信息,登录验证图片需要手动选择。
(3)自动填充输入,完成查询和页面跳转。

后续可以优化的功能:
(1)从配置文件中读取账号、密码、出发、目的地,时间等相关信息。
(2)从配置文件中获取城市编码
(3)登录验证图片可以使用第三方识别自动选择

二、初识Splinter

1.简介

Splinter 是用 Python 开发的一个开源web自动化测试的工具集。 它可以帮你自动化浏览器的行为,比如浏览 URLs 并和页面进行交互。

Splinter是现有浏览器之上抽象层自动化工具(如 SeleniumPhantomJS 和 zope.testbrowser )。它具有 高级API ,这使得它很容易去编写Web应用程序的自动化测试。

例如, 用Splinter填写一个表单项:

browser.fill('username', 'janedoe')

在Selenium中, 等效代码会是:

elem =browser.find_element.by_name('username')

elem.send_keys('janedoe')

2.安装

命令行下执行以下命令: sudo pip install splinter

3.快速上手

(1)导入Browser类

from splinter.browser import Browser

(2)创建一个实例

指定driver为chrome浏览器,如果你不为 Browser 指定 driver, 那么会默认使用 firefox。

browser = Browser(driver_name='chrome',executable_path='xxx')

其中executable_path为对应浏览器driver的本地目录。

(3)访问百度搜索页面

使用 browser.visit 方法可访问任意网站:

browser.visit('http://baidu.com')

(4)输入搜索关键词

页面加载完毕后,可以在输入框填充字段,大过年的我们就搜索一下2018年新年祝福吧:

browser.fill('wd', '2018年新年祝福')

(5)点击搜索按钮

Splinter 可以通过按钮的css, xpath, id, tag 或 name来识别,百度搜素按钮使用以下来操作:

button = browser.find_by_xpath('//input[@type="submit"]').click()

(6)匹配结果

使用is_text_present查看匹配结果:

if browser.is_text_present('春节'):

    print("找到了")

else:

    print("没找到")

(7)关闭浏览器

结束测试后,我们需要使用 browser.quit 关闭浏览器:

browser.quit()

完整代码如下:

# -*- coding: utf-8 -*-

#导入Browser类

from splinter.browser import Browser

#创建一个实例, 指定driver为chrome浏览器,如果你不为 Browser 指定 driver, 那么会默认使用 firefox。

browser = Browser(driver_name='chrome',executable_path='/Users/xxx/Downloads/chromedriver')

browser.visit('http://baidu.com')

browser.fill('wd', '2018年新年祝福')

button = browser.find_by_xpath('//input[@type="submit"]').click()

if browser.is_text_present('春节'):

    print("找到了")

else:

    print("没找到")

browser.quit()

三、12306自动查票
1、流程分析
(1)执行python脚本后,能够自动打开浏览器,进入12306登录页面。因此需要加载浏览器驱动并打开登录页面。

(2)命令行提示用户输入用户名、密码,并等待用户手动在浏览器选择验证码完成登录。

(3)命令行提示用户输入出发地、目的地以及出发时间。

(4)根据输入查询车次信息

2、示例代码
代码均有详细注释,根据上面的流程分析,我们简化为三步。
第一步是加载基本信息,包括浏览器、url等。第二步是输入个人信息登录。第三步是输入查询条件查询车次信息。

# -*- coding: utf-8 -*-

from splinter.browser import Browser
from time import sleep

class TicketsUtil(object):

    def __init__(self):
        self.loadBasicInfo()

    def loadBasicInfo(self):
        # 登录的url
        self.loginUrl = 'https://kyfw.12306.cn/otn/login/init'
        #登录成功后的url
        self.myUrl = 'https://kyfw.12306.cn/otn/index/initMy12306'
        #余票查询页面
        self.ticketUrl = 'https://kyfw.12306.cn/otn/leftTicket/init'
        # 初始化驱动
        self.driver=Browser(driver_name='chrome',executable_path='/Users/xxx/Downloads/chromedriver')
        # 初始化浏览器窗口大小
        self.driver.driver.set_window_size(1400, 1000)

    def login(self):
        print("开始登录...")
        # 登录
        self.driver.visit(self.loginUrl)

        self.username = input("\n请输入用户名,输入按回车...")
        #合法性判断
        while True:
            if self.username == '':
                self.username = input("\n请输入用户名,输入按回车...")
            else:
                break

        self.password = input("\n请输入密码,输入按回车...")
        #合法性判断
        while True:
            if self.password == '':
                self.password = input("\n请输入密码,输入按回车...")
            else:
                break

        # 自动填充用户名
        self.driver.fill("loginUserDTO.user_name", self.username)
        # 自动填充密码
        self.driver.fill("userDTO.password", self.password)

        print(u"等待验证码,自行输入...")

        # 验证码需要自行输入,程序自旋等待,直到验证码通过,点击登录
        while True:
            if self.driver.url != self.myUrl:
                sleep(1)
            else:
                break

        print(u"登录成功...")

    def query(self):
        self.source = input("\n请输入出发地(格式为:北京,BJP),输入按回车...")
        #合法性判断
        while True:
            if self.source == '':
                self.source = input("\n请输入出发地(格式为:北京,BJP),输入按回车...")
            else:
                break

        self.destination = input("\n请输入目的地(格式为:深圳,SZQ),输入按回车...")
        while True:
            if self.destination == '':
                self.destination = input("\n请输入目的地(格式为:深圳,SZQ),输入按回车...")
            else:
                break

        self.date = input("\n请输入出发日期(格式为:2018-02-14),输入按回车...")
        while True:
            if self.date == '':
                self.date = input("\n请输入出发日期,输入按回车...")
            else:
                break
        #转换输入的出发地成"武汉,WHN",再进行编码
        self.source = self.source.encode('unicode_escape').decode("utf-8").replace("\\u", "%u").replace(",", "%2c")
        #转换输入的目的地
        self.destination = self.destination.encode('unicode_escape').decode("utf-8").replace("\\u", "%u").replace(",", "%2c")

        # 加载出发地
        self.driver.cookies.add({"_jc_save_fromStation": self.source})
        # 加载目的地
        self.driver.cookies.add({"_jc_save_toStation": self.destination})
        # 加载出发日
        self.driver.cookies.add({"_jc_save_fromDate": self.date})

        # 带着查询条件,重新加载页面
        self.driver.reload()
        # 查询余票
        self.driver.find_by_text(u"查询").click()
        sleep(0.1)
        # 防止超时再次查询余票
        self.driver.find_by_text(u"查询").click()

        print('查询成功')

    """入口函数"""
    def start(self):
        self.loadBasicInfo()

        # 登录,自动填充用户名、密码,自旋等待输入验证码,输入完验证码,点登录后,访问 tick_url(余票查询页面)
        self.login()

        # 登录成功,访问余票查询页面
        self.driver.visit(self.ticketUrl)

        self.query()

if __name__ == '__main__':
    print(u"===========自动查票开启===========")
    ticketsUtil = TicketsUtil()
    ticketsUtil.start()