sys.path.append('/Users/lorenamesa/Desktop/pyten/lib/python2.7/site-packages/') from api.cta_api import CTABustracker, CTATraintracker bus_tracker = CTABustracker() predictions = bus_tracker.get_predictions_for_stops(route_id=72, stp_id="890,944") cali_predictions = bus_tracker.get_predictions_for_stops(route_id=52, stp_id="3183,3006") with open("/Users/lorenamesa/Desktop/pytennessee/bus_data.csv", "ab") as csvdata: headers = ['distance_to_stop', 'route', 'vehicle_id', 'arrival_time', 'requested_time', 'route_direction', 'stop_name'] if predictions: logging.info("Writing data for bus data...") wr = csv.writer(csvdata, dialect='excel') for prediction in predictions: wr.writerow(prediction.__dict__.values()) for cali_p in cali_predictions: wr.writerow(cali_p.__dict__.values()) # ['distance_to_stop', 'route', 'vehicle_id', 'arrival_time', 'requested_time', 'route_direction', 'stop_name'] train_tracker = CTATraintracker() train_predictions = train_tracker.get_predictions_for_stops(stp_id="30112,30116,30153",max=10) with open("/Users/lorenamesa/Desktop/pytennessee/train_data.csv", "ab") as traindata: if train_predictions: wr = csv.writer(traindata, dialect='excel') logging.info("Writing data for cta data...") for prediction in train_predictions: wr.writerow(prediction.__dict__.values()) # "('is_delayed', u'0')","('route', u'Blue')","('is_fault', u'0')","('arrival_time', u'20160124 13:35:50')","('requested_time', u'20160124 13:23:50')","('is_scheduled', u'0')","('stop_name', u'Monroe')","('is_approached', u'0')","('train_direction', u'1')"
prediction.eta)) for cali_p in cali_predictions: db.query("INSERT INTO bus_predictions " + "(stop_name, route, direction, arrival_time, requested_time, distance, vehicle_id, eta) " + "VALUES ('{0}', '{1}', '{2}', {3}, {4}, {5}, {6}, {7})".format(prediction.stop_name, prediction.route, prediction.direction, prediction.arrival_time, prediction.requested_time, prediction.distance, prediction.vehicle_id, prediction.eta)) train_tracker = CTATraintracker() train_predictions = train_tracker.get_predictions_for_stops(requested_time=requested_time, stp_id="30112,30116,30153", max=10) if train_predictions: logging.info("Writing cta data...") for prediction in train_predictions: db.query("INSERT INTO train_predictions " + "(stop_name, route, direction, arrival_time, requested_time, delayed, scheduled, approaching, fault, eta) " + "VALUES ('{0}', '{1}', '{2}', {3}, {4}, {5}, {6}, {7}, {8}, {9})".format(prediction.stop_name, prediction.route, prediction.direction, prediction.arrival_time, prediction.requested_time, prediction.delayed, prediction.scheduled, prediction.approaching,