# coding=utf-8

__author__ = 'flyingfishyx <[email protected]>'

import requests
import sys
import json

reload(sys)
sys.setdefaultencoding('utf-8')


class Express100(object):

    company_url = "http://www.kuaidi100.com/autonumber/autoComNum"
    trace_url = "http://www.kuaidi100.com/query"

    @classmethod
 

    def get_json_data(cls, url, payload):
        r = requests.get(url=url, params=payload)
        return r.json()

    @classmethod
    def get_company_info(cls, express_code):

        """
        {
            comCode: "",
            num: "3351419285305",
            auto: [
                {
                    comCode: "shentong",
                    id: "",
                    noCount: 13852,
                    noPre: "33514",
                    startTime: ""
                }
            ]
        }
        """
 


        payload = {'text': express_code}
        data = cls.get_json_data(cls.company_url, payload)
        return data

    @classmethod
    def get_express_info(cls, express_code):
        """
        {
            message: "ok",
            nu: "3351419285305",
            ischeck: "0",
            condition: "00",
            com: "shentong",
            status: "200",
            state: "0",
            data: [
                {
                    time: "2018-01-21 22:19:45",
                    ftime: "2018-01-21 22:19:45",
                    context: "淄博市 山東淄博公司-已發往-遼寧盤錦中轉部",
                    location: ""
                },
            ]
        }
        """
 


        company_info = cls.get_company_info(express_code)

        company_code = ""

        if company_info.get("auto", ""):
            company_code = company_info.get("auto", "")[0].get("comCode", "")

        payload = {'type': company_code, 'postid': express_code, 'id': 1}

        data = cls.get_json_data(cls.trace_url, payload)

        data.update(company_info)

        return data


if __name__ == "__main__":
    while True:
        code = input("請輸入快遞單號：")
        res = Express100.get_express_info(str(code).strip())
        print json.dumps(res, ensure_ascii=False, sort_keys=True, indent=4)