12306/inter/GetPassengerDTOs.py

# coding=utf-8
from config.TicketEnmu import ticket
from myException.PassengerUserException import PassengerUserException
import wrapcache

try:
    xrange          # Python 2
except NameError:
    xrange = range  # Python 3


class getPassengerDTOs:
    """
    获取乘客信息
    :return:
    """
    def __init__(self, session, ticket_peoples, set_type, is_more_ticket_num):
        """
        :param session: 登录实例
        :param ticket_peoples: 乘客
        :param set_type: 坐席
        """
        self.session = session
        self.ticket_peoples = ticket_peoples
        self.is_more_ticket_num = is_more_ticket_num
        self.set_type = set_type.encode("utf8")

    def sendGetPassengerDTOs(self):
        getPassengerDTOsResult = self.session.httpClint.send(self.session.urls["get_passengerDTOs"], {})
        if getPassengerDTOsResult.get("data", False) and getPassengerDTOsResult["data"].get("normal_passengers", False):
            normal_passengers = getPassengerDTOsResult['data']['normal_passengers']
            _normal_passenger = [normal_passengers[i] for i in range(len(normal_passengers)) if
                                 normal_passengers[i]["passenger_name"] in self.ticket_peoples]
            return _normal_passenger if _normal_passenger else [normal_passengers[0]]  # 如果配置乘车人没有在账号，则默认返回第一个用户
        else:
            if getPassengerDTOsResult.get("data", False) and getPassengerDTOsResult['data'].get("exMsg", False):
                print(getPassengerDTOsResult['data'].get("exMsg", False))
            elif getPassengerDTOsResult.get('messages', False):
                print(getPassengerDTOsResult.get('messages', False))
            else:
                raise PassengerUserException(ticket.DTO_NOT_FOUND)

    def getPassengerTicketStr(self, set_type):
        """
        获取getPassengerTicketStr 提交对应的代号码
        :param str: 坐席
        :return:
        """
        passengerTicketStr = {
            '一等座': 'M',
            '特等座': 'P',
            '二等座': 'O',
            '商务座': 9,
            '硬座': 1,
            '无座': 1,
            '软座': 2,
            '软卧': 4,
            '硬卧': 3,
        }
        return str(passengerTicketStr[set_type.replace(' ', '')])

    def getPassengerTicketStrListAndOldPassengerStr(self):
        """
        获取提交车次人内容格式
        passengerTicketStr	O,0,1,文贤平,1,43052419950223XXXX,15618715583,N_O,0,1,梁敏,1,43052719920118XXXX,,N
        oldPassengerStr	文贤平,1,43052719920118XXXX,1_梁敏,1,43052719920118XXXX,1
        ps: 如果is_more_ticket打开了的话，那就是读取联系人列表里面前符合车次数量的前几个联系人
        :return:
        """
        passengerTicketStrList = []
        oldPassengerStr = []
        if wrapcache.get("user_info"):  # 如果缓存中有联系人方式，则读取缓存中的联系人
            user_info = wrapcache.get("user_info")
            print(u"缓存中找到联系人信息: {0}".format(user_info))
        else:
            user_info = self.sendGetPassengerDTOs()
            wrapcache.set("user_info", user_info, timeout=9999999)
        set_type = self.getPassengerTicketStr(self.set_type)
        if not user_info:
            raise PassengerUserException(ticket.DTO_NOT_IN_LIST)
        if len(user_info) < self.is_more_ticket_num:  # 如果乘车人填错了导致没有这个乘车人的话，可能乘车人数会小于自动乘车人
            self.is_more_ticket_num = len(user_info)
        if self.is_more_ticket_num is 1:
            passengerTicketStrList.append(
                '0,' + user_info[0]['passenger_type'] + "," + user_info[0][
                    "passenger_name"] + "," +
                user_info[0]['passenger_id_type_code'] + "," + user_info[0]['passenger_id_no'] + "," +
                user_info[0]['mobile_no'] + ',N')
            oldPassengerStr.append(
                user_info[0]['passenger_name'] + "," + user_info[0]['passenger_id_type_code'] + "," +
                user_info[0]['passenger_id_no'] + "," + user_info[0]['passenger_type'] + '_')
        else:
            for i in xrange(self.is_more_ticket_num):
                passengerTicketStrList.append(
                    '0,' + user_info[i]['passenger_type'] + "," + user_info[i][
                        "passenger_name"] + "," + user_info[i]['passenger_id_type_code'] + "," + user_info[i][
                        'passenger_id_no'] + "," + user_info[i]['mobile_no'] + ',N_' + set_type)
                oldPassengerStr.append(
                    user_info[i]['passenger_name'] + "," + user_info[i]['passenger_id_type_code'] + "," +
                    user_info[i]['passenger_id_no'] + "," + user_info[i]['passenger_type'] + '_')
        return {
            "passengerTicketStrList": set_type + "," + ",".join(passengerTicketStrList),
            "oldPassengerStr": "".join(oldPassengerStr),
            "code": ticket.SUCCESS_CODE,
            "set_type": set_type,
            "status": True,
            "user_info": user_info,
        }