gen_requests.py

#!/usr/bin/python2.4

import itertools
import simplejson

TEST_ALL_MODES = False #set to True to make it a hard test with short walk limits and walk-only trips &c.

json_out = {}
    
# Initialize the otpprofiler DB 'requests' table with query parameters.
# Note that on-street modes are not walk-limited, so we don't want to vary the max walk param there.
# another way to do this would be to store the various values in tables, and construct this
# view as a constrained product of all the other tables (probably eliminating the synthetic keys).

# (time, arriveBy)
times = [ ("08:50:00", True ),
          ("14:00:00", True ),
          ("18:00:00", False),
          ("23:45:00", False) ]

if TEST_ALL_MODES:
    # (mode, walk, min)
    modes = [ (mode, walk, "QUICK") 
        for mode in ["WALK,TRANSIT", "BICYCLE,TRANSIT"] 
        for walk in [250, 2000, 40000] ]
    modes.append( ("WALK", 2000, "QUICK") )
    modes.extend( [("BICYCLE", 2000, minimize) for minimize in ["QUICK", "SAFE", "FLAT"]] )
else:
    # (mode, walk, min)
    modes = [ ("WALK,TRANSIT", 2000, "QUICK") ]

all_params = [(time, walk, mode, minimize, arriveBy) 
    for (time, arriveBy) in times 
    for (mode, walk, minimize) in modes] 

requests_json = []
for i, params in enumerate( all_params ) :
    # NOTE the use of double quotes to force case-sensitivity for column names. These columns 
    # represent query parameters that will be substituted directly into URLs, and URLs are defined 
    # to be case-sensitive.

    time,maxWalkDist,mode,min,arriveBy = params
    typical = (time=="08:50:00" and maxWalkDist == 2000 and "BICYCLE" not in mode)

    requests_json.append( dict(zip(('time','maxWalkDistance','mode','min','arriveBy','typical','id'),params+(typical,i))) )
json_out['requests'] = requests_json

# Initialize the otpprofiler DB with random endpoints and user-defined endpoints
import csv
endpoints_json = []
for filename, random in [("endpoints_custom.csv", False), ("endpoints_random.csv", True)] :
    endpoints = open(filename)
    reader = csv.DictReader(endpoints)

    endpoints = list(reader)

    for i, rec in enumerate( endpoints ):
        endpoint_rec = {'id':i, 'random':random, 'lon':float(rec['lon']), 'lat':float(rec['lat']), 'name':rec['name'], 'notes':None}
        endpoints_json.append( endpoint_rec )
json_out['endpoints'] = endpoints_json

fpout = open("requests.json","w")
simplejson.dump(json_out, fpout, indent=2 )
fpout.close()