qpxexpress.py

import json



key = "fdksaljvdispoajhp89vdsa8sFDSpvdsa"


def flightsearch(origin, destination, depart_data):
  url = "https://www.googleapis.com/qpxExpress/v1/trips/search?key=%s" % key
  code = {
      "request": {
          "passengers": {
              "kind": "qpxexpress#passengerCounts",
              "adultCount": 1,
          },
          "slice": [
              {
                  "kind": "qpxexpress#sliceInput",
                  "origin": origin,  # "LGA"
                  "destination": destination,  # "MIA"
                  "date": depart_data,  # "2015-06-09"
              }
          ],
          "refundable": "false",
          "solutions": 20
      }
  }
  jsonreq = json.dumps(code, encoding='utf-8')
  req = urllib2.Request(url, jsonreq, {'Content-Type': 'application/json'})
  flight = urllib2.urlopen(req)
  response = flight.read()
  flight.close()
  return response


def flightsearchresults(origin, destination, depart_data):
  flight = []
  # read json data from file
  j = json.load(file('%s-%s.json' % (origin, destination)))
  # j = flightsearch(origin, destination, depart_data)
  for t in j['trips']['tripOption']:
    price = float(t['saleTotal'][3:])
    depart = t['slice'][0]['segment'][0]['leg'][0]['departureTime'][11:16]
    arrive = t['slice'][0]['segment'][-1]['leg'][0]['arrivalTime'][11:16]
    origin = t['slice'][0]['segment'][0]['leg'][0]['origin']
    dest = t['slice'][0]['segment'][-1]['leg'][0]['destination']
    flight.append((depart, arrive, int(price)))
  return flight


def createschedule(people, dest, dep, ret):
  flights = {}
  for p in people:
    name, origin = p
    flights[(origin, dest)] = flightsearchresults(origin, dest, dep)
    flights[(dest, origin)] = flightsearchresults(dest, origin, ret)
  return flights