爬取12306火車票信息

# -*- 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)