track_geom_lonlats = graph.geom2lonlats(track_geom) track_geom_lonlats_length = len(track_geom_lonlats) dropoff_lonlat = track_geom_lonlats[track_geom_lonlats_length - 1] dropoff_dict['dropoff_lonlat'] = dropoff_lonlat dropoff_time = track_desc.split(',')[track_geom_lonlats_length - 1] dropoff_dict['dropoff_time'] = dropoff_time dropoff_dict['track_oorc'] = track_oorc return dropoff_dict from_date = (2012, 11, 1) to_date = (2012, 11, 30) tbnames_suffix = lib.dates_to_tbnames(from_date, to_date) tracks_tbname_prefix = 'taxi_tracks_occupied_crusing_' paths_tbname_prefix = 'taxi_paths_occupied_crusing_st_' tracks_tbname = [ tracks_tbname_prefix + tbname_suffix.split('_')[1] for tbname_suffix in tbnames_suffix ] paths_tbname = [ paths_tbname_prefix + tbname_suffix.split('_')[1] for tbname_suffix in tbnames_suffix ] paths_tb_length = len(paths_tbname) for i in range(paths_tb_length):
import psycopg2 import sys import time import lib import os #proc_input_dir = '/home/donghao/ITS_project/taxi_finder/data/data_by_cuid_count/' proc_input_dir = '/home/donghao/ITS_project/taxi_finder/data/data_by_cuid_count_filter_1/' fromdate = (2012,11,1) todate = (2012,11,4) tbnames = lib.dates_to_tbnames(fromdate,todate) #tbnames = ['tb_20121101'] class Tracks2Postgres: def __init__(self): self.conn = psycopg2.connect(database = 'beijing_taxi_201211',user = '******',password = '******') print 'postgres successfully connected' def cuid_start_callback(self,cuid): self.cuid = cuid self.coords = list() self.descs = list() self.OorC = '' self.pre_occupied = '-1' def cuid_end_callback(self,cuid): pass def line_callback(self,line,postgres_tbname): if line != '': attrs = lib.parse_line(line) #taxi_status_tx = str(attrs['taxi_status']) time_tx = attrs['time']
dropoff_dict['track_id'] = track_id track_geom_lonlats = graph.geom2lonlats(track_geom) track_geom_lonlats_length = len(track_geom_lonlats) dropoff_lonlat = track_geom_lonlats[track_geom_lonlats_length - 1] dropoff_dict['dropoff_lonlat'] = dropoff_lonlat dropoff_time = track_desc.split(',')[track_geom_lonlats_length - 1] dropoff_dict['dropoff_time'] = dropoff_time dropoff_dict['track_oorc'] = track_oorc return dropoff_dict from_date = (2012,11,1) to_date = (2012,11,30) tbnames_suffix = lib.dates_to_tbnames(from_date, to_date) tracks_tbname_prefix = 'taxi_tracks_occupied_crusing_' paths_tbname_prefix = 'taxi_paths_occupied_crusing_st_' tracks_tbname = [tracks_tbname_prefix + tbname_suffix.split('_')[1] for tbname_suffix in tbnames_suffix] paths_tbname = [paths_tbname_prefix + tbname_suffix.split('_')[1] for tbname_suffix in tbnames_suffix] paths_tb_length = len(paths_tbname) for i in range(paths_tb_length): gw = graph.new_gw() track_reader = db.TrackReader(tbname = tracks_tbname[i], limit = 10000, offset = 0) path_reader = db.PathReader(tbname = paths_tbname[i], gw = gw, limit = 10000, offset = 0) dropoff_file = '/home/donghao/ITS_project/taxi_finder/data/data_dropoff/dropoff' + '_' +tbnames_suffix[i].split('_')[1] dropoff_f = open(dropoff_file, 'w')