Contents

require 'earth/locality'
class NationalTransitDatabaseCompany < ActiveRecord::Base
  self.primary_key = "id"
  self.table_name = :ntd_companies
  
  has_many :ntd_records, :foreign_key => 'company_id', :primary_key => 'id', :class_name => 'NationalTransitDatabaseRecord'
  
  def self.rail_companies
    NationalTransitDatabaseRecord.rail_records.map(&:ntd_company).uniq
  end
  
  def rail_records
    ntd_records.where(:mode_code => NationalTransitDatabaseMode.rail_modes)
  end
  
  # Methods to calculate summary statistics from rail_records
  [:rail_passengers, :rail_passenger_distance, :rail_vehicle_distance, :rail_vehicle_time, :rail_electricity, :rail_diesel].each do |method|
    define_method method do
      attribute = method.to_s.split('rail_')[1].to_sym
      rail_records.sum(attribute) > 0 ? rail_records.sum(attribute) : nil 
    end
  end
  
  # Methods to look up units from from rail_records
  # TODO this looks inefficient
  [:rail_passenger_distance_units, :rail_vehicle_distance_units, :rail_vehicle_time_units, :rail_electricity_units, :rail_diesel_units].each do |method|
    define_method method do
      attribute = method.to_s.split('rail_')[1].to_sym
      units = rail_records.map(&attribute).uniq.compact
      (units.length == 1 and units[0].present?) ? units[0] : raise("Error: units missing or multiple units in #{name}'s NTD records")
    end
  end
  
  col :id
  col :name
  col :acronym
  col :zip_code_name
  col :duns_number
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
earth-0.11.20	lib/earth/rail/national_transit_database_company.rb
earth-0.11.19	lib/earth/rail/national_transit_database_company.rb