python爬蟲十五:爬取12306火車票資訊
阿新 • • 發佈:2019-01-08
轉: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)
力不從心