Esempio n. 1
0
    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')