#coding:UTF-8
# 快遞100 API
# 郵箱: [email protected]

import requests
import urllib

# 函式: 承運公司名到文字
def GetComName(comCode):
    if comCode=='shentong':
        return '申通快遞'
    elif comCode=='zhontong':
        return '中通快遞'
    elif comCode=='ems':
        return 'EMS'
    elif comCode=='huitongkuaidi'
 
:
        return '匯通快運'
    else:
        return comCode

# 函式: 取狀態文字
def GetStateText(num):
    if num==0:
        return '運輸中'
    elif num==1:
        return '攬件'
    elif num==2:
        return '疑難'
    elif num==3:
        return '已簽收'
    elif num==4:
        return '退回簽收'
    elif num==5:
        return
 
 '派送中'
    elif num==6:
        return '退回中'

p = {}
p['text'] = input("請輸入快遞運單編號: ")  #比如: 227728570825
autoComNum = requests.get("http://www.kuaidi100.com/autonumber/autoComNum", params=p)
com = autoComNum.json()


if com['auto'] == []:
    print("這是一個錯誤的運單編號!")

else:
    print("\n---------------- 承運公司 ------------------\n"
 
)
    i=0
    for this in com['auto']:
        i = i + 1
        print( str(i) + ". " + GetComName(this['comCode']) + "\n")


    num = input("承運公司序號: ")
    print("\n---------------- 正在查詢, 請稍等... ------------------\n")
    data = {}
    data['type'] = com['auto'][int(num)-1]['comCode']
    data['postid'] =  p['text']
    data['valicode'] = ''
    data['id'] = 1
    data['temp'] = '0.14881871496191512'
    query = requests.get("http://www.kuaidi100.com/query", params=data)
    res = query.json()

    print("\n運單編號 --> " + res['nu'])
    print("\n承運公司 --> " + GetComName(res['com']))
    print("\n當前狀態 --> " + GetStateText(int(res['state'])))
    print("\n---------------- 跟蹤資訊 ------------------\n")
    for this in res['data']:
        print(this['time'] + "\t" + this['context'] + "\n")