flightcheck 0.0.7 → 0.0.8

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: bdb7135f2e28deaff5f27571923eb7b603fc58ec
-  data.tar.gz: e3d30b474b4bc5a2b2e199eaa1ca1975d8e746f6
+  metadata.gz: 8d7279bcf039e1885bbd60fd2fb7d808a8592c70
+  data.tar.gz: 5fe29ad744055798a6a52fb33f92c29c319e181c
 SHA512:
-  metadata.gz: ef8e83ffee2b4f3657f47a378af5c697bb1a2c7e2bdedb6871b91b931f45e39bfa55d19a338f0d68009babdc12bec9bee0283a0ee6bf13bf0910156bd61e6318
-  data.tar.gz: d7c12d9d9bcc327af9038842499334d274f650c9cea3dccd823d396a6a069b4f47f82a6bcd71de5ffa731a41a903649cbbaf7b43ed69b0cd61c05225f042d5f3
+  metadata.gz: d5cd02f37e734ca4db3f6594a69cd1c2bf7cbcdcfa3c5bb6f64e48fc8083faf7639797183cb8c5d441cd3f8a5e3a429cb083c21430bc518621f5d10537177de1
+  data.tar.gz: 1219acd2740153dc1c266d42014c453e876e0003248235bdee13f19457a6b51c146b3f6ebbaacb4dd4b187c4b1ace3eb64979727c6b79ab71b103608730bca38

data/lib/flightcheck.rb

@@ -4,11 +4,11 @@ class Flight
   attr_reader :flight_number, :departure_date, :title, :status, :route, :url,
   :raw_airline_iata, :airline_iata, :route_array, :departure_airport_iata,
   :departure_airport_city, :departure_country_iata, :arriving_airport_iata,
-  :arriving_airport_city, :arriving_country_iata
+  :arriving_airport_city, :arriving_country_iata, :multiple_segments,
+  :segment_1, :segment_2, :selected_segment, :segment_1, :segment_2
 
   ## Initialize
   def initialize(flight_number, departure_date)
-
     # Flight number
     # e.g. "BA2589"
     @flight_number = flight_number
@@ -23,25 +23,63 @@ class Flight
 
   ## Use Mechanize to scrape all flight information
   def check
-
     # Instantiate a new Mechanize object
     a = Mechanize.new
 
     # Visit the flightstats URL to scrape information
     a.get(@url) do |page|
-
-      # Page title contains the basic information for the flight
-      # e.g. "(BA) British Airways 2589 Flight Status"
-	    @title = page.search("head > title").text.strip
-
-      # Flight status
-      # e.g. "Landed - Delayed 103 minutes"
-	    @status = page.search("#mainAreaLeftColumn > div.uiComponent674.flightStatusByFlightBlock > div > div.statusBlock > div.statusType").text.strip
-
-      # Flight route
-      # e.g. "(VCE) Venice, IT to (LGW) London, EN, GB"
-      @route = page.search("#mainAreaLeftColumn > div.uiComponent674.flightStatusByFlightBlock > div > div.route").text.strip
-
+      # Check if flight has multiple segments
+      # Returns a boolean value
+      @multiple_segments = page.search("//*[@id='mainAreaLeftColumn']/div/div/text()").text.strip === "This flight has multiple segments. Please select a segment from the list." ? true : false
+
+      if @multiple_segments == true
+
+        @segment_1 = page.search("//*[@id='mainAreaLeftColumn']/div/div/p[1]/a").text.strip
+        @segment_2 = page.search("//*[@id='mainAreaLeftColumn']/div/div/p[2]/a").text.strip
+
+        puts   "This flight has two flight segments,"
+        puts   "please select the correct one"
+        puts   "========================================"
+        puts   "Segment 1: #{@segment_1}"
+        puts   "Segment 2: #{@segment_2}"
+        puts   "----------------------------------------"
+        print  "1 or 2? Make your choice: "
+        @selected_segment = gets.chomp
+
+        if @selected_segment == "1"
+          @segment_1 = page.links[0].click
+
+          # Page title contains the basic information for the flight
+          # e.g. "(BA) British Airways 2589 Flight Status"
+          @title = @segment_1.search("head > title").text.strip
+
+          # Flight status
+          # e.g. "Landed - Delayed 103 minutes"
+          @status = @segment_1.search("#mainAreaLeftColumn > div.uiComponent674.flightStatusByFlightBlock > div > div.statusBlock > div.statusType").text.strip
+
+          # Flight route
+          # e.g. "(VCE) Venice, IT to (LGW) London, EN, GB"
+          @route = @segment_1.search("#mainAreaLeftColumn > div.uiComponent674.flightStatusByFlightBlock > div > div.route").text.strip
+
+        elsif @selected_segment == "2"
+          @segment_2 = page.links[1].click
+
+          # Page title contains the basic information for the flight
+          # e.g. "(BA) British Airways 2589 Flight Status"
+          @title = @segment_2.search("head > title").text.strip
+
+          # Flight status
+          # e.g. "Landed - Delayed 103 minutes"
+          @status = @segment_2.search("#mainAreaLeftColumn > div.uiComponent674.flightStatusByFlightBlock > div > div.statusBlock > div.statusType").text.strip
+
+          # Flight route
+          # e.g. "(VCE) Venice, IT to (LGW) London, EN, GB"
+          @route = @segment_2.search("#mainAreaLeftColumn > div.uiComponent674.flightStatusByFlightBlock > div > div.route").text.strip
+
+        else
+          puts "The input was unknown. Please use '1' for segment 1 and '2' for segment 2"
+        end
+      end
     end
 
     ## Airline IATA
@@ -87,8 +125,5 @@ class Flight
     puts @title
     puts @status
     puts @route
-
-
-
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: flightcheck
 version: !ruby/object:Gem::Version
-  version: 0.0.7
+  version: 0.0.8
 platform: ruby
 authors:
 - Narek Aramjan
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-05-31 00:00:00.000000000 Z
+date: 2014-06-06 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: mechanize