stalkr 0.9.1 → 0.9.2

data/VERSION

@@ -1 +1 @@
-0.9.1
+0.9.2

data/lib/stalkr.rb

@@ -9,13 +9,16 @@ require 'scrapi'
 require 'htmlentities'
 require 'date'
 
-require 'stalkr/base'
-require 'stalkr/result'
-require 'stalkr/error'
 
-require 'stalkr/ups'
-require 'stalkr/usps'
-require 'stalkr/fedex'
+stalkr_dir = File.dirname(__FILE__)
+require "#{stalkr_dir}/stalkr/base"
+require "#{stalkr_dir}/stalkr/result"
+require "#{stalkr_dir}/stalkr/error"
+
+require "#{stalkr_dir}/stalkr/ups"
+require "#{stalkr_dir}/stalkr/usps"
+require "#{stalkr_dir}/stalkr/fedex"
+require "#{stalkr_dir}/stalkr/china_post"
 
 if not DateTime.new.public_methods.include? "to_time" then
     # monkey patch DateTime to add to_time (exists in Ruby 1.9.2 and above)

data/lib/stalkr/base.rb

@@ -7,8 +7,14 @@ module Stalkr
             attr_accessor :regex
         end
 
-        def fetchurl(url)
-            return Net::HTTP.get_response(URI.parse(URI.escape(url))).body
+        def fetchurl(url, data = nil)
+            url = URI.parse(URI.escape(url))
+            if data.nil? then
+                # GET request
+                return Net::HTTP.get_response(url).body
+            else
+                return Net::HTTP.post_form(url, data).body
+            end
         end
 
         def strip_tags(html)

data/lib/stalkr/china_post.rb

@@ -0,0 +1,68 @@
+
+module Stalkr
+
+class ChinaPost < Base
+
+    # RA#########CN
+    # not sure if the RA is constant or not
+    self.regex = /\b(RA\d{9}CN)\b/i
+
+    def track(id)
+
+        url = "http://www.emsairmailtracking.com/chinapost.php"
+        html = fetchurl(url, {"itemNo" => id})
+
+        begin
+
+            td_scraper = Scraper.define do
+                array :tds
+                process "td", :tds => :text
+                result :tds
+            end
+
+            row_scraper = Scraper.define do
+                array :rows
+                process "tr", :rows => td_scraper
+                result :rows
+            end
+
+            rows = row_scraper.scrape(html)
+
+            if not rows or rows.size < 2 then
+                raise "ChinaPost scraper failed"
+            end
+
+            ret = Result.new(:ChinaPost)
+
+            # ["Item No.", "Year", "Status", "Location", "Destination Country", "Status", "Recipient's Signature"]
+            # get rid of the header row
+            rows.shift
+
+            status = rows.first[2].strip.downcase
+            puts status
+            if status =~ /delivered/ then
+                ret.status = DELIVERED
+            elsif status =~ /arrival|departure/ then
+                ret.status = IN_TRANSIT
+            else
+                ret.status = UNKNOWN
+            end
+
+            updated_at = cleanup_html( rows.first[5] )
+            ret.updated_at = DateTime.strptime( updated_at, "%Y-%m-%d" ).to_time
+
+            if ret.status == DELIVERED then
+                ret.delivered_at = ret.updated_at
+            end
+
+            return ret
+
+        rescue Exception => ex
+            raise Stalkr::Error.new(ex, html)
+
+        end
+
+    end
+
+end # class ChinaPost
+end # module Stalkr

data/test/stalkr/china_post_test.rb

@@ -0,0 +1,44 @@
+require File.dirname(__FILE__) + '/../test_helper.rb'
+
+class ChinaPost_Test < Test::Unit::TestCase
+
+    def test_track
+        id = "RA103044224CN"
+        info = Stalkr::ChinaPost.new.track(id)
+        assert(info.shipper == :ChinaPost)
+        assert(info.status == Stalkr::IN_TRANSIT)
+        assert(info.updated_at.kind_of? Time)
+    end
+
+    def test_track_bad_code
+
+        begin
+            id = "foobar"
+            info = Stalkr::ChinaPost.new.track(id)
+            flunk("no exception was thrown")
+        rescue Exception => ex
+            if ex.message != "ChinaPost scraper failed" then
+                flunk("wrong exception was thrown")
+            end
+        end
+
+    end
+
+    def test_valid_id
+        id = "RA103044224CN"
+        ret = Stalkr::ChinaPost.is_valid?(id)
+        assert(ret)
+    end
+
+    def test_invalid_id
+        id = "1X799470341200708"
+        ret = Stalkr::ChinaPost.is_valid?(id)
+        assert(!ret)
+    end
+
+    def test_extract_id
+        str = "asdf RA103044224CN asdf"
+        assert(Stalkr::ChinaPost.extract_id(str) == "RA103044224CN")
+    end
+
+end

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: stalkr
 version: !ruby/object:Gem::Version 
-  hash: 57
+  hash: 63
   prerelease: 
   segments: 
   - 0
   - 9
-  - 1
-  version: 0.9.1
+  - 2
+  version: 0.9.2
 platform: ruby
 authors: 
 - Chetan Sarva
@@ -15,7 +15,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2011-02-16 00:00:00 -05:00
+date: 2011-02-17 00:00:00 -05:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency 
@@ -111,11 +111,13 @@ files:
 - VERSION
 - lib/stalkr.rb
 - lib/stalkr/base.rb
+- lib/stalkr/china_post.rb
 - lib/stalkr/error.rb
 - lib/stalkr/fedex.rb
 - lib/stalkr/result.rb
 - lib/stalkr/ups.rb
 - lib/stalkr/usps.rb
+- test/stalkr/china_post_test.rb
 - test/stalkr/fedex_test.rb
 - test/stalkr/ups_test.rb
 - test/stalkr/usps_test.rb
@@ -156,6 +158,7 @@ signing_key:
 specification_version: 3
 summary: Ruby library for tracking packages
 test_files: 
+- test/stalkr/china_post_test.rb
 - test/stalkr/fedex_test.rb
 - test/stalkr/ups_test.rb
 - test/stalkr/usps_test.rb