from helper import * import pdb from time import sleep def start_session(pccToEmulate="6B3D"): body = \ """

999999 123123 6B3D TEST-6B3D SessionCreateRQ SessionCreateRQ 170390 2012-06-07T10:00:00 2013-06-06T23:59:59 170390 WS793581 6B3D DEFAULT

""" response = requests.post(url, data=body, headers=headers) # print response.content; root = ET.fromstring(response.content) # print response.content secToken = root[0][-1][0].text if pccToEmulate !='6B3D': body = \ """

1212 2323 6B3D TEST-6B3D ContextChangeLLSRQ ContextChangeLLSRQ 1001 2012-06-07T10:00:01 2013-06-06T23:59:59 """ \ + secToken \ + """

""" response = requests.post(url, data=body, headers=headers) print 'Session started ... ' + pccToEmulate return secToken def stop_session(secToken): body = \ """

1212 2323 6B3D TEST-6B3D SessionCloseRQ SessionCloseRQ 1001 2012-06-07T10:00:01 2013-06-06T23:59:59 """ \ + secToken \ + """

""" response = requests.post(url, data=body, headers=headers) print 'Session stopped ...' return response.content def ticket(secToken, doc_no): log() body = \ """

1212 2323 6B3D TEST-6B3D eTicketCouponLLSRQ eTicketCouponLLSRQ 1001 2012-06-07T10:00:01 2013-06-06T23:59:59 """ \ + secToken \ + """

""" response = requests.post(url, data=body, headers=headers) myprint('ticket/' + doc_no, response.content) print 'eTicket Retrieved ...' return response.content def queue_count(queue_number, secToken): log() body = \ """

999999 123123 6B3D TEST-6B3D QueueCountLLSRQ QueueCountLLSRQ 170390 2012-06-07T10:00:00 2013-06-06T23:59:59 """ \ + secToken \ + """

""" response = requests.post(url, data=body, headers=headers) print 'Queue Count Done' myprint('QC/' + str(datetime.datetime.now()), response.content) return response.content def queue_access(queue_number, secToken,pccToEmulate): log() body = \ """

999999 123123 6B3D TEST-6B3D QueueAccessLLSRQ QueueAccessLLSRQ 170390 2012-06-07T10:00:00 2013-06-06T23:59:59 """ \ + secToken \ + """

""" response = requests.post(url, data=body, headers=headers) # print response.content; print 'Queue Access Done' myprint('qa/' + str(datetime.datetime.now()), response.content) return response.content def pnr_get(pnr): log() t = start_session(pccToEmulate) x = get_pnr(pnr, t, 'test') stop_session(t) return x def get_pnr(pnr, secToken, filename): log() body = \ """

999999 123123 6B3D TEST-6B3D TravelItineraryReadRQ TravelItineraryReadRQ 170390 2012-06-07T10:00:00 2013-06-06T23:59:59 """ \ + secToken \ + """

FULL

""" response = requests.post(url, data=body, headers=headers) # print response.content; myprint('tir/' + filename, response.content) print 'Itinerary Read Done' return response.content def get_price(secToken, filename): log() body = \ """

999999 123123 6B3D TEST-6B3D DisplayPriceQuoteLLSRQ DisplayPriceQuoteLLSRQ 170390 2012-06-07T10:00:00 2013-06-06T23:59:59 """ \ + secToken \ + """

""" response = requests.post(url, data=body, headers=headers) # print response.content; print 'PriceQuote Done' myprint('pq/PQ_' + filename, response.content) return response.content def get_airline_details(airline_code, flight_number, dt): log() t = start_session(pccToEmulate) body = \ """

1212 2323 6B3D TEST-6B3D OTA_AirFlifoLLSRQ OTA_AirFlifoLLSRQ 1001 2012-06-07T10:00:01 2013-06-06T23:59:59 """ \ + t \ + """

""" response = requests.post(url, data=body, headers=headers) stop_session(t) r = etree.XML(body) # print etree.tostring(r) # myprint(response.content); root = etree.XML(response.content) x = root[-1][-1] flightinfo = x[-1] scheduledinfo = flightinfo[0] origin = scheduledinfo[0] destination = scheduledinfo[1] o = origin.get('LocationCode') d = destination.get('LocationCode') dep_time = origin[0].get('Scheduled') arr_time = destination[0].get('Scheduled') return [o, d, dep_time, arr_time] current_time = '' def alternate_fare(stt, pnr): log() t = start_session(pccToEmulate) body = \ """

1212 2323 6B3D TEST-6B3D SabreCommandLLSRQ SabreCommandLLSRQ 1001 2012-06-07T10:00:01 2013-06-06T23:59:59 """ \ + t \ + """

""" \ + str(stt) \ + """

""" print 'Ding' response = requests.post(url, data=body, headers=headers) stop_session(t) root = etree.XML(response.content) y = root[-1][-1][0] print etree.tostring(y, pretty_print=True) x = etree.tostring(y, pretty_print=True) parts = x.split('ITINERARY OPTION ') myprint('alt/ALT_' + pnr, x) results = [] for part in parts[1:]: ps = part.split('\n') ps.reverse() fareid = 'Fare ' + str(ps.pop()) result = [] check = 0 while ps != []: segment = ps.pop() infos = segment.split() infos.reverse() fare = '' currency = '' try: checkk = infos.pop() print checkk if checkk == 'TOTAL': t_fare = infos[0] t_currency = infos[1] elif checkk.isdigit(): marketing_airline_code = infos.pop() if len(infos) == 13: op_airline_code = infos.pop() flight_number = infos.pop() rbd = infos.pop() dep_date = m_date(infos.pop()) day_week = infos.pop() origin = infos.pop() destination = infos.pop() dep_time = infos.pop() arr_time = infos.pop() equipment = infos.pop() stopover = infos.pop() fare_basis = '' result.append([ marketing_airline_code, origin, destination, currency, fare, pnr, flight_number, rbd, dep_date, fare_basis, dep_time, arr_time, fareid, ]) else: continue except: # pdb.set_trace() continue try: result[0][4] = t_fare result[0][3] = t_currency results.append(result) except: pass return results # def commit_alt_fare( # tktnumber, # timestamp, # gds_source, # marketing_airline_code, # origin, # destination, # currency, # fare, # pnr, # flight_number, # rbd, # dep_date, # fare_basis, # dep_time, # arr_time, # fareid, # depdatetime # ): # try: # cnx = mysql.connector.connect(user='rt_bot', password='mp.work.', # host='tripnomicsdb.cloudapp.net', # database='Alternatefare') # cursor = cnx.cursor();cnx.autocommit = True # add_gen_row = 'INSERT INTO Fare_Details ' + 'VALUES(' + '%s,' \ # * 21 + '%s)' # gen_info_insert = tuple([ # tktnumber, # timestamp, # gds_source, # marketing_airline_code, # origin, # destination, # currency, # '', # fare, # pnr, # flight_number, # rbd, # dep_date, # fare_basis, # dep_time, # arr_time, # fareid, # '', # '', # '','',depdatetime # ]) # cursor.execute(add_gen_row, gen_info_insert) # cnx.commit() # cursor.close() # cnx.close() # print "Committed" # except Exception,e: # print e # print "ERROR !!!!!! Can't commit alternate fare to database" # return False # return True # def components(xml_body, preferred_airlines,pnr,pccToEmulate): # dts, orgs,dstns, rbds, indicators = [],[],[],[],[] # root = etree.XML(xml_body) # y = root[-1][-1] # ns = y.nsmap # flight_segments = y.findall( # './/tir38:ReservationItems/tir38:Item/tir38:FlightSegment', ns) # result = [] # for flight_segment in flight_segments: # try: # connectionInd = flight_segment.get('ConnectionInd') # except: # connectionInd = 'O' # if connectionInd is None: # connectionInd = 'O' # try: # rbd = flight_segment.get('ResBookDesigCode') # except: # rbd = 'Y' # dep_date = flight_segment.get('DepartureDateTime') # origin = flight_segment.find('.//tir38:OriginLocation', # ns).get('LocationCode') # destination = flight_segment.find( # './/tir38:DestinationLocation', # ns).get('LocationCode') # depdate = datetime.datetime.strptime(dep_date, '%Y-%m-%dT%H:%M') # depdate = depdate.strftime('%Y-%m-%dT%H:%M:%S') # orgs.append(origin) # dstns.append(destination) # dts.append(depdate) # rbds.append(rbd) # indicators.append(connectionInd) # altstring = altfare(dts,[orgs,dstns],rbds,pnr,indicators,pccToEmulate) # return altstring # def class_of_booking(rbd): # cnx = mysql.connector.connect(user='rt_bot', password='mp.work.', # host='tripnomicsdb.cloudapp.net', # database='PNR_Trigger') # cursor = cnx.cursor();cnx.autocommit = True # try: # select_stmt = \ # """ select if(count(*)=0,'Economy',CabinClass) # FROM Masters.Cabin_Class_Mapping where FareClass like CONCAT('%',%(fare_class)s,'%') """ # cursor.execute(select_stmt, {'fare_class': rbd}) # row = cursor.fetchone() # print row # cclass = row[0] # except: # cclass = 'Economy' # cursor.close() # cnx.close() # return cclass # def pref_airlines(customer_code, origin, destination): # return [] # def make_body(dates, dests, rbd, indicators): # dd = [] # indicators.append("O") # oo = [] # dts = [] # x = [] # # ###print pnr+" LOG ", "Destinations",dests # origins = dests[0] # # ###print pnr+" LOG ", "Origins",origins # destinations = dests[1] # mmaster = indicators # origin_indicators = mmaster[:-1] # destination_indicators = mmaster[1:] # for i in range(len(destinations)): # if origin_indicators[i] == "O": # oo.append(origins[i]) # if destination_indicators[i] == "O": # dd.append(destinations[i]) # for i in range(len(destinations)): # if indicators[i] == "O": # # ###print pnr+" LOG ", i, len(dates) # dts.append(dates[i]) # # destinations = dd # dates = dts # # ###print pnr+" LOG ", "RBD" # # ###print pnr+" LOG ", rbd # ###print len(oo), len(dd), len(dates), len(rbd) # for i in range(len(oo)): # x.append([oo[i], dd[i], dates[i], rbd[i]]) # ###print len(x),x # # destinations = dd # dates = dts # # ###print pnr+" LOG ", "RBD" # # ###print pnr+" LOG ", rbd # t = "" # ###print "Legs ",len(x) # for i in range(len(x)): # dt = x[i][2] # o = x[i][0] # d = x[i][1] # booking_class = class_of_booking(x[i][3]) # # ###print pnr+" LOG ", booking_class # if 'ECONOMY' in booking_class.upper(): # rd = "Y" # elif 'BUSINESS' in booking_class.upper(): # rd = "J" # else: # rd = "F" # t = t + """ # """ + dt + """ # # # """+""" # # """ # # ###print pnr+" LOG ", t # return t # def altfare(dt, o, rbd, pnr, indicators,pccToEmulate): # x = """ # # # # # # # # """ + make_body(dt, o, rbd, indicators) + """ # # # 1 # # # # # # # # # # """ # t = start_session(pccToEmulate) # body = """ # # # # 1212 # # # 2323 # # 6B3D # TEST-6B3D # BargainFinderMaxRQ # BargainFinderMaxRQ # # 1001 # 2012-06-07T10:00:01 # 2013-06-06T23:59:59 # # # # """ + \ # t + """ # # # """ + x + """ # """ # # ###print pnr+" LOG ", "#" * 80 # response = requests.post(url, data=body, headers=headers) # stop_session(t) # return response.content # def move_pnr(pnr,pccToMove,pccFrom,DestinationQueue,OriginQueue): # secToken = start_session(pccFrom) # body = \ # """ # # # # 1212 # # # 2323 # # 6B3D # TEST-6B3D # QueueMoveLLSRQ # QueueMoveLLSRQ # # 1001 # 2012-06-07T10:00:01 # 2013-06-06T23:59:59 # # # # """ \ # + secToken \ # + """ # # # # # # # # # # # # # # # """ # response = requests.post(url, data=body, headers=headers) # stop_session(secToken) # print "Moved PNR",pnr,pccToMove,DestinationQueue # # print response.content # return response.content