cumtd 0.0.1 → 0.1.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: b9b29b282253fd7d5f524f61e7e65092e0421018
-  data.tar.gz: c0b228e68084015682b7a547aabb5a4d8bafa92f
+  metadata.gz: bd40b208a39e58b04532f3480f40bfa872fb9a9c
+  data.tar.gz: 0be1f6a864db4f9f12542d9312b67ee2f3d438d9
 SHA512:
-  metadata.gz: 4a6c9f5aebe2c84db2176df6641c377f6fe3736f46632c0266a240bf11c8679c7991e64c3b460cbffcc51bf905ae0d46507b98a373541d2d0bcbe5cc639babfa
-  data.tar.gz: 2ad657f57f944cdeb8e9705cf755250cc8d96b2841dada811b2765f465492a91602f918e2459c8266a26fd527962612462ddca0fa57e6d47cdecea48903705af
+  metadata.gz: fe7ebb6bf969afd5b457cfb0ecdf4dbc4ee08ef140b3888960ca9fb083df28388d8a71f7452484bffab3a1870a0269f85910a1a07d20dbfc5a325374fbb74395
+  data.tar.gz: 4c1689a745a21980d2fa46a48d655bbc3795ed1b4336af9551c2602cc801f84ea6e3cac0d83a08d52ff02465af0dfb899e6a632d475a38b91ad8153279a1d220

data/lib/cumtd.rb

@@ -1,23 +1,90 @@
 require 'httparty'
+require 'yaml'
 class CUMTD
+	require 'stop'
+	require 'stop_point'
+	require 'departure'
+	require 'route'
+	require 'shape'
+	require 'trip'
+	require 'vehicle'
 	include HTTParty
 	base_uri 'developer.cumtd.com/api/v2.2/json'
-	def initialize(api_key)
+	def initialize(api_key, stops_file=nil, routes_file=nil, serialize_path=File.expand_path(File.dirname(__FILE__)))
 		@api_key = api_key
+
+		@@all_stops = Array.new
+		if stops_file
+			object = nil
+			File.open(stops_file,"rb") {|f| @@all_stops = Marshal.load(f)}
+		else
+			self.get_stops
+		end
+
+		@@all_routes = Array.new
+		if routes_file
+			object = nil
+			File.open(routes_file,"rb") {|f| @@all_routes = Marshal.load(f)}
+		else
+			self.get_routes
+		end
+
+		unless serialize_path == :no_serialize
+			p File.join(serialize_path, 'stops')
+			serialize_stops(File.join(serialize_path, 'stops.yaml'))
+			serialize_routes(File.join(serialize_path, 'routes.yaml'))
+		end
+
 	end
 
 	def api_key
 		@api_key
 	end
 
-	def current_stops
-		@current_stops
+	def self.all_stops
+		@@all_stops
 	end
 
-	def get_stops(lat, lon, count=20)
-		response = self.class.get("/GetStopsByLatLon?key=#{api_key}&lat=#{lat}&lon=#{lon}&count=#{count}")
-		current_stops = response["stops"]
-		return stops = response["stops"]
+	def self.all_routes
+		@@all_routes
+	end
+
+	def serialize_stops(file_location)
+		File.open(file_location, "wb") do |file|
+			YAML.dump(@@all_stops,file)
+		end
+	end
+
+	def serialize_routes(file_location)
+		File.open(file_location, "wb") do |file|
+			YAML.dump(@@all_routes,file)
+		end
+	end
+
+	def get_stops
+		response = self.class.get("/GetStops?key=#{@api_key}")
+		@@all_stops.clear
+		response["stops"].each do |stop|
+			@@all_stops << Stop.new(stop)
+		end
+
+	end
+
+	def get_routes
+		response = self.class.get("/GetRoutes?key=#{@api_key}")
+		@@all_routes.clear
+		response["routes"].each do |route|
+			@@all_routes << Route.new(route)
+		end
+	end
+
+	def get_stops_by_lat_lon(lat, lon, count=20)
+		response = self.class.get("/GetStopsByLatLon?key=#{@api_key}&lat=#{lat}&lon=#{lon}&count=#{count}")
+		stops = Array.new
+		response["stops"].each do |stop|
+			stops << Stop.new(stop)
+		end
+		return stops
 	end
 
 	def print_all_departures(stops)
@@ -28,28 +95,37 @@ class CUMTD
 		end
 	end
 
-	def get_routes_by_stop(stop_id)
-		response = self.class.get("/GetRoutesByStop?key=#{api_key}&stop_id=#{stop_id}")
-		return response.parsed_response["routes"]
+	def get_routes_by_stop(stop)
+		response = self.class.get("/GetRoutesByStop?key=#{@api_key}&stop_id=#{stop.stop_id}")
+		routes = Array.new
+		response["routes"].each do |route|
+			routes << Route.new(route)
+		end
+		return routes
 	end
 
 	def get_stop_by_id(stop_id)
-		response = self.class.get("/GetStop?key=#{api_key}&stop_id=#{stop_id}")
-		return response.parsed_response["stops"].first["stop_name"]
-	end
-
-	def get_departures_by_stop(stop_id)
-		response = self.class.get("/GetDeparturesByStop?key=#{api_key}&stop_id=#{stop_id}")
-		return departures = response.parsed_response["departures"]
+		response = self.class.get("/GetStop?key=#{@api_key}&stop_id=#{stop_id}")
+		stop = Stop.new(response["stops"])
+		return stop
 	end
 
-	def print_departures(departures, stop)
-		puts stop["stop_name"] unless departures == []
-		departures.each do |departure|
-			puts departure["headsign"] + " " + departure["expected_mins"].to_s + "mins"
+	def get_departures_by_stop(stop)
+		response = self.class.get("/GetDeparturesByStop?key=#{@api_key}&stop_id=#{stop.stop_id}")
+		departures = Array.new
+		response["departures"].each do |departure|
+			departures << Departure.new(departure)
 		end
+		return departures
 	end
 
+	# def print_departures(departures)
+	# 	puts stop["stop_name"] unless departures == []
+	# 	departures.each do |departure|
+	# 		puts departure["headsign"] + " " + departure["expected_mins"].to_s + "mins"
+	# 	end
+	# end
+
 	# Returns an array of hashes. Each hash contains a "stop" key denoting the stop
 	# and the "departures" key contains one departure. These hashes are sorted 
 	# based on length of time until they depart, given stops to get departures for.
@@ -58,12 +134,99 @@ class CUMTD
 		master_deps = Array.new
 		stops.each do |stop|
 			# master_deps.concat(Hash["stop", stop, "departures", get_departures_by_stop(stop["stop_id"]))
-			stops_from_deps = get_departures_by_stop(stop["stop_id"])
+			stops_from_deps = get_departures_by_stop(stop)
 			stops_from_deps.each do |stop_dep|
 				master_deps << Hash["stop", stop, "departures", stop_dep]
 			end
 		end
-		return master_deps.sort_by! { |stop| stop["departures"]["expected_mins"]  }
+		return master_deps.sort_by! { |stop| stop["departures"].expected_mins  }
+	end
+
+	def get_stops_by_search(query, count=10)
+		response = self.class.get("/GetStopsBySearch?key=#{@api_key}&query=#{query}&count=#{count}")
+		stops = Array.new
+		response["stops"].each do |stop|
+			stops << Stop.new(stop)
+		end
+		return stops
+	end
+
+	def get_stop_times_by_trip(trip_id)
+		response = self.class.get("/GetStopTimesByTrip?key=#{@api_key}&trip_id=#{trip_id}")
+		stop_times = Array.new
+		response["stop_times"].each do |stop_time|
+			stop_times << StopTime.new(stop_time)
+		end
+		return stop_times
+	end
+
+	def get_vehicles
+		response = self.class.get("/GetVehicles?key=#{@api_key}")
+		vehicles = Array.new
+		response["vehicles"].each do |vehicle|
+			vehicles << Vehicle.new(vehicle)
+		end
+		return vehicles
+	end
+
+	def get_vehicle_by_id(vehicle_id)
+		response = self.class.get("/GetVehicle?key=#{@api_key}&vehicle_id=#{vehicle_id}")
+		return Vehicle.new(response["vehicles"].first)
+	end
+
+	def get_vehicles_by_route_id(route_id)
+		response = self.class.get("/GetVehiclesByRoute?key=#{@api_key}&route_id=#{route_id}")
+		vehicles = Array.new
+		response["vehicles"].each do |vehicle|
+			vehicles << Vehicle.new(vehicle)
+		end
+		return vehicles
+	end
+
+	def get_trip_by_id(trip_id)
+		response = self.class.get("/GetTrip?key=#{@api_key}&trip_id=#{trip_id}")
+		return Trip.new(response["trips"].first)
+	end
+
+	def get_trips_by_block(block_id)
+		response = self.class.get("/GetTripsByBlock?key=#{@api_key}&block_id=#{block_id}")
+		trips = Array.new
+		response["trips"].each do |trip|
+			trips << Trip.new(trip)
+		end
+		return trips
+	end
+
+	def get_shape_by_id(shape_id)
+		response = self.class.get("/GetShape?key=#{@api_key}&shape_id=#{shape_id}")
+		shapes = Array.new
+		response["shapes"].each do |shape|
+			shapes << Shape.new(shape)
+		end
+		return shapes
+	end
+
+	def get_shape_between_stops(begin_stop_id, end_stop_id, shape_id)
+		response = self.class.get("/GetShape?key=#{@api_key}&begin_stop_id=#{begin_stop_id}&end_stop_id=#{end_stop_id}&shape_id=#{shape_id}")
+		shapes = Array.new
+		response["shapes"].each do |shape|
+			shapes << Shape.new(shape)
+		end
+		return shapes
+	end
+
+	def get_stop_times_by_stop(stop_id, route_id=nil, date=Time.now.to_s)
+		if !route_id
+			response = self.class.get("/GetStopTimesByStop?key=#{@api_key}&stop_id=#{stop_id}")
+		else
+			response = self.class.get("/GetStopTimesByStop?key=#{@api_key}&stop_id=#{stop_id}&route_id=#{route_id}")
+		end
+
+		stop_times = Array.new
+		response["stop_times"].each do |stop_time|
+			stop_times << StopTime.new(stop_time)
+		end
+		return stop_times
 	end
 
 end

data/lib/departure.rb

@@ -0,0 +1,68 @@
+require 'trip'
+require 'route'
+class Departure < CUMTD
+	def initialize(json)
+		@stop_id = json["stop_id"]
+		@headsign = json["headsign"]
+		@trip = Trip.new(json["trip"])
+		@vehicle_id = json["vehicle_id"]
+		@origin = CUMTD.all_stops.select { |stop| stop.stop_points.each == \
+			json["origin"]["stop_id"] }
+		@destination = CUMTD.all_stops.select { |stop| stop.stop_points.each == \
+			json["destination"]["stop_id"] }
+		@is_monitored = json["is_monitored"]
+		@is_scheduled = json["is_scheduled"]
+		@scheduled = DateTime.parse(json["scheduled"]).to_time
+		@expected = DateTime.parse(json["expected"]).to_time
+		@expected_mins = json["expected_mins"]
+		@location = Hash[:lat, json["location"]["lat"], :lon, json["location"]["lon"]]
+	end
+	
+	def stop_id
+		@stop_id
+	end
+
+	def headsign
+		@headsign
+	end
+
+	def trip
+		@trip
+	end
+
+	def vehicle_id
+		@vehicle_id
+	end
+
+	def origin
+		@origin
+	end
+
+	def destination
+		@destination
+	end
+
+	def is_monitored
+		@is_monitored
+	end
+
+	def is_scheduled
+		@is_scheduled
+	end
+
+	def scheduled
+		@scheduled
+	end
+
+	def expected
+		@expected
+	end
+
+	def expected_mins
+		@expected_mins
+	end
+
+	def location
+		@location
+	end
+end

data/lib/route.rb

@@ -0,0 +1,29 @@
+class Route < CUMTD
+	def initialize(json)
+		@route_color = json["route_color"]
+		@route_id = json["route_id"]
+		@route_long_name = json["route_long_name"]
+		@route_short_name = json["route_short_name"]
+		@route_text_color = json["route_text_color"]
+	end
+	
+	def route_color
+		@route_color
+	end
+
+	def route_id
+		@route_id
+	end
+
+	def route_long_name
+		@route_long_name
+	end
+
+	def route_short_name
+		@route_short_name
+	end
+
+	def route_text_color
+		@route_text_color
+	end
+end

data/lib/shape.rb

@@ -0,0 +1,29 @@
+class Shape < CUMTD
+	def initialize(json)
+		@shape_dist_traveled = json["shape_dist_traveled"]
+		@shape_pt_lat = json["shape_pt_lat"]
+		@shape_pt_lon = json["shape_pt_lon"]
+		@shape_pt_sequence = json["shape_pt_sequence"]
+		@shape_stop_id = json["shape_stop_id"] if json["shape_stop_id"]
+	end
+	
+	def shape_dist_traveled
+		@shape_dist_traveled
+	end
+
+	def shape_pt_lat
+		@shape_pt_lat
+	end
+	
+	def shape_pt_lon
+		@shape_pt_lon
+	end
+
+	def shape_pt_sequence
+		@shape_pt_sequence
+	end
+
+	def shape_stop_id
+		@shape_stop_id
+	end
+end

data/lib/stop.rb

@@ -0,0 +1,29 @@
+class Stop < CUMTD
+	def initialize(json)
+		@stop_id = json["stop_id"]
+		@stop_name = json["stop_name"]
+		@code = json["code"]
+		@stop_points = Array.new
+		json["stop_points"].each do |stop_point|
+			@stop_points << StopPoint.new(stop_point)
+		end
+	end
+	
+	def stop_id
+		@stop_id
+	end
+	
+	def stop_name
+		@stop_name
+	end
+
+	def code
+		@code
+	end
+
+	def stop_points
+		@stop_points
+	end
+
+	
+end

data/lib/stop_point.rb

@@ -0,0 +1,30 @@
+class StopPoint < CUMTD
+	def initialize(json)
+		@code = json["code"]
+		@stop_id = json["stop_id"]
+		@stop_lat = json["stop_lat"]
+		@stop_lon = json["stop_lon"]
+		@stop_name = json["stop_name"]
+	end
+	
+	def code
+		@code
+	end
+
+	def stop_id
+		@stop_id
+	end
+
+	def stop_lat
+		@stop_lat
+	end
+
+	def stop_lon
+		@stop_lon
+	end
+
+	def stop_name
+		@stop_name
+	end
+end
+

data/lib/stop_time.rb

@@ -0,0 +1,32 @@
+class StopTime < CUMTD
+	require 'stop_point'
+	require 'trip'
+	def initialize(json)
+		@arrival_time = DateTime.parse(json["arrival_time"]).to_time
+		@departure_time = DateTime.parse(json["departure_time"]).to_time
+		@stop_sequence = json["stop_sequence"]
+		@stop_point = StopPoint.new(json["stop_point"]) if json["stop_point"]
+		@trip = Trip.new(json["trip"]) if json["trip"]
+	end
+	
+	def arrival_time
+		@arrival_time
+	end
+
+	def departure_time
+		@departure_time
+	end
+
+	def stop_sequence
+		@stop_sequence
+	end
+
+	def stop_point
+		@stop_point
+	end
+
+	def trip
+		@trip
+	end
+
+end

data/lib/trip.rb

@@ -0,0 +1,40 @@
+class Trip < CUMTD
+	def initialize(json)
+		@trip_id = json["trip_id"]
+		@trip_headsign = json["trip_headsign"]
+		@route_id = json["route_id"]
+		@block_id = json["block_id"]
+		@direction = json["direction"]
+		@service_id = json["service_id"]
+		@shape_id = json["shape_id"]
+	end
+	
+	def trip_id
+		@trip_id
+	end
+
+	def trip_headsign
+		@trip_headsign
+	end
+
+	def route_id
+		@route_id
+	end
+
+	def block_id
+		@block_id
+	end
+
+	def direction
+		@direction
+	end
+
+	def service_id
+		@service_id
+	end
+	
+	def shape_id
+		@shape_id
+	end
+
+end

data/lib/vehicle.rb

@@ -0,0 +1,45 @@
+class Vehicle < CUMTD
+	def initialize(json)
+		@vehicle_id = json["vehicle_id"]
+		@trip = Trip.new(json["trip"])
+		@location = Hash[:lat, json["location"]["lat"], :lon, json["location"]["lon"]]
+		@previous_stop_id = json["previous_stop_id"]
+		@next_stop_id = json["next_stop_id"]
+		@origin_stop_id = json["origin_stop_id"]
+		@destination_stop_id = json["destination_stop_id"]
+		@last_updated = DateTime.parse(json["last_updated"]).to_time
+	end
+	
+	def vehicle_id
+		@vehicle_id
+	end
+
+	def trip
+		@trip
+	end
+
+	def location
+		@location
+	end
+
+	def previous_stop_id
+		@previous_stop_id
+	end
+
+	def next_stop_id
+		@next_stop_id
+	end
+
+	def origin_stop_id
+		@origin_stop_id
+	end
+
+	def destination_stop_id
+		@destination_stop_id
+	end
+
+	def last_updated
+		@last_updated
+	end
+	
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: cumtd
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.1.1
 platform: ruby
 authors:
 - Brett Jackson
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-08-09 00:00:00.000000000 Z
+date: 2013-08-12 00:00:00.000000000 Z
 dependencies: []
 description: An interface to CUMTD's API.
 email: brett@brettjackson.org
@@ -17,7 +17,15 @@ extensions: []
 extra_rdoc_files: []
 files:
 - lib/cumtd.rb
-homepage: http://rubygems.org/gems/cumtd
+- lib/stop.rb
+- lib/stop_point.rb
+- lib/departure.rb
+- lib/route.rb
+- lib/trip.rb
+- lib/stop_time.rb
+- lib/vehicle.rb
+- lib/shape.rb
+homepage: http://github.com/bjackson/cumtd
 licenses:
 - MIT
 metadata: {}