Ejemplo n.º 1
0
def write_stop_list_table(out_file, stop_list_map):
    out_file.write("CREATE TABLE IF NOT EXISTS trip_stops (id INTEGER PRIMARY KEY, blob STRING);\n")
    for stop_list_id, lst in stop_list_map.items():
        box = Box()
        box.add_short(len(lst))

        for stop_id, sequence in lst:
            try:
                box.add_string(stop_id)
                box.add_short(sequence)
            except:
                print((stop_id, sequence))
                raise

        out_file.write("INSERT INTO trip_stops VALUES (%d, %s);\n" % (
            stop_list_id, box.get_blob_string()
        ))
Ejemplo n.º 2
0
def write_arrivals_table(out_file, arrivals_map):
    out_file.write("CREATE TABLE IF NOT EXISTS arrivals (id INTEGER PRIMARY KEY, blob STRING);\n")
    for arrival_id, lst in arrivals_map.items():
        box = Box()
        box.add_short(len(lst))

        for arrival_seconds, departure_seconds in lst:
            if arrival_seconds % 60 != 0:
                raise Exception("arrival_seconds % 60 != 0")
            if (arrival_seconds / 60) > 0xffff or arrival_seconds < 0:
                raise Exception("arrival_seconds out of range")
            box.add_short(arrival_seconds/60)
            #box.add_int(departure_seconds)

        out_file.write("INSERT INTO arrivals VALUES (%d, %s);\n" % (
            arrival_id, box.get_blob_string()
        ))