1. 程式人生 > >python爬蟲十五:爬取12306火車票資訊

python爬蟲十五:爬取12306火車票資訊

轉:https://zhuanlan.zhihu.com/p/26701898 

# -*- coding: utf-8 -*-
'''
獲取12306城市名和城市程式碼的資料
檔名: parse_station.py
'''
import requests
import re
import json

#關閉https證書驗證警告
requests.packages.urllib3.disable_warnings()

def getStation():
        # 12306的城市名和城市程式碼js檔案url
        url = 'https://kyfw.12306.cn/otn/resources/js/framework/station_name.js?station_version=1.9018'
        r = requests.get(url,verify=False)
        pattern = u'([\u4e00-\u9fa5]+)\|([A-Z]+)'
        result = re.findall(pattern,r.text)
        station = dict(result)#{'北京北': 'VAP', '北京東': 'BOP', '北京': 'BJP',
        #print(station)
        return station

'''
查詢兩站之間的火車票資訊

輸入引數: <date> <from> <to>

12306 api:
'https://kyfw.12306.cn/otn/leftTicket/query?leftTicketDTO.train_date=2017-07-18&leftTicketDTO.from_station=NJH&leftTicketDTO.to_station=SZH&purpose_codes=ADULT'

'''





#生成查詢的url
def get_query_url(text):
    # 城市名程式碼查詢字典
    # key:城市名 value:城市程式碼


    try:
        date = '2018-06-01'
        from_station_name = '上海'
        to_station_name = '北京'
        from_station = text[from_station_name]
        to_station = text[to_station_name]
    except:
        date, from_station, to_station = '--', '--', '--'
        # 將城市名轉換為城市程式碼

    # api url 構造
    url = (
        'https://kyfw.12306.cn/otn/leftTicket/query?'
        'leftTicketDTO.train_date={}&'
        'leftTicketDTO.from_station={}&'
        'leftTicketDTO.to_station={}&'
        'purpose_codes=ADULT'
    ).format(date, from_station, to_station)
    print(url)

    return url




#獲取資訊
def query_train_info(url,text):
    '''
    查詢火車票資訊:
    返回 資訊查詢列表
    '''

    info_list = []
    try:
        r = requests.get(url, verify=False)

        # 獲取返回的json資料裡的data欄位的result結果
        raw_trains = r.json()['data']['result']

        for raw_train in raw_trains:
            # 迴圈遍歷每輛列車的資訊
            data_list = raw_train.split('|')

            # 車次號碼
            train_no = data_list[3]
            # 出發站
            from_station_code = data_list[6]
            from_station_name = text['上海']
            # 終點站
            to_station_code = data_list[7]
            to_station_name = text['北京']
            # 出發時間
            start_time = data_list[8]
            # 到達時間
            arrive_time = data_list[9]
            # 總耗時
            time_fucked_up = data_list[10]
            # 一等座
            first_class_seat = data_list[31] or '--'
            # 二等座
            second_class_seat = data_list[30]or '--'
            # 軟臥
            soft_sleep = data_list[23]or '--'
            # 硬臥
            hard_sleep = data_list[28]or '--'
            # 硬座
            hard_seat = data_list[29]or '--'
            # 無座
            no_seat = data_list[26]or '--'

            # 列印查詢結果
            info = ('車次:{}\n出發站:{}\n目的地:{}\n出發時間:{}\n到達時間:{}\n消耗時間:{}\n座位情況:\n 一等座:「{}」 \n二等座:「{}」\n軟臥:「{}」\n硬臥:「{}」\n硬座:「{}」\n無座:「{}」\n\n'.format(
                train_no, from_station_name, to_station_name, start_time, arrive_time, time_fucked_up, first_class_seat,
                second_class_seat, soft_sleep, hard_sleep, hard_seat, no_seat))


            print(info)
            info_list.append(info)

        return info_list
    except:
        return ' 輸出資訊有誤,請重新輸入'


text=getStation();
url=get_query_url(text)

query_train_info(url,text)

力不從心