flight_segment.rb in earth-0.5.0

- old
+ new
@@ -1,24 +1,30 @@
+# need this for association with Aircraft through loose_tight_dictionary_cached_results
+require 'loose_tight_dictionary/cached_result'
+
 class FlightSegment < ActiveRecord::Base
   set_primary_key :row_hash
   
   extend CohortScope
   self.minimum_cohort_size = 1
   
-  belongs_to :airline,             :foreign_key => 'airline_iata_code',                 :primary_key => 'iata_code'
-  belongs_to :origin_airport,      :foreign_key => 'origin_airport_iata_code',          :primary_key => 'iata_code',     :class_name => 'Airport'
-  belongs_to :destination_airport, :foreign_key => 'destination_airport_iata_code',     :primary_key => 'iata_code',     :class_name => 'Airport'
-  belongs_to :aircraft,            :foreign_key => 'aircraft_bts_code',                 :primary_key => 'bts_code'
+  # If airport iata code is missing, associate with all airports in a city
+  # We need this to calculate distance when importing ICAO segments - see cm1 flight_segment.rb
+  has_many :origin_city_airports,      :foreign_key => 'city', :primary_key => 'origin_airport_city',      :class_name => 'Airport'
+  has_many :destination_city_airports, :foreign_key => 'city', :primary_key => 'destination_airport_city', :class_name => 'Airport'
   
-  falls_back_on :distance      => lambda { weighted_average(:distance,      :weighted_by => :passengers) }, # 2077.1205         data1 10-12-2010
-                :seats         => lambda { weighted_average(:seats,         :weighted_by => :passengers) }, # 144.15653537046   data1 10-12-2010
-                :load_factor   => lambda { weighted_average(:load_factor,   :weighted_by => :passengers) }, # 0.78073233770097  data1 10-12-2010
-                :freight_share => lambda { weighted_average(:freight_share, :weighted_by => :passengers) }  # 0.022567224170157 data1 10-12-2010
+  # Enable flight_segment.aircraft
+  cache_loose_tight_dictionary_matches_with :aircraft, :primary_key => :aircraft_description, :foreign_key => :description
   
+  falls_back_on :distance         => lambda { weighted_average(:distance,         :weighted_by => :passengers) }, # 2077.1205         data1 10-12-2010
+                :seats_per_flight => lambda { weighted_average(:seats_per_flight, :weighted_by => :passengers) }, # 144.15653537046   data1 10-12-2010
+                :load_factor      => lambda { weighted_average(:load_factor,      :weighted_by => :passengers) }, # 0.78073233770097  data1 10-12-2010
+                :freight_share    => lambda { weighted_average(:freight_share,    :weighted_by => :passengers) }  # 0.022567224170157 data1 10-12-2010
+  
   data_miner do
-    tap "Brighter Planet's sanitized T100 data", Earth.taps_server
+    tap "Brighter Planet's sanitized flight segment data", Earth.taps_server
     
-    process "pull dependencies" do
+    process "Pull dependencies" do
       run_data_miner_on_belongs_to_associations
     end
   end
 end