iTunesAmazon 0.0.1 → 0.0.2

data/lib/i_tunes_amazon.rb

@@ -1,7 +1,5 @@
 #!/usr/bin/env ruby -w
 
-
-
 require 'rubygems'
 require 'nokogiri'
 require 'amazon/aws'
@@ -11,10 +9,10 @@ require './lib/track.rb'
 include Amazon::AWS
 include Amazon::AWS::Search
 
-KEY_ID = '1E3447REXBE9VATGH4R2'
+KEY_ID = ''
 
 class ITunesAmazon
-  VERSION = '0.0.1'
+  VERSION = '0.0.2'
 
   attr_accessor :tracks, :albums, :total_time
 
@@ -25,6 +23,7 @@ class ITunesAmazon
     get_amazon_stats
   end
 
+  # Creates the tracks and albums objects from an iTunes file
   def parse(file)
     xml = Nokogiri::XML.parse(File.read(file))
     xml.xpath("//dict/dict/dict").each do |song|
@@ -40,7 +39,11 @@ class ITunesAmazon
           when 'Play Count' then track.play_count = attr.next.text
         end
       end
-      @tracks << track
+
+      # don't deal with Podcasts
+      unless track.genre == 'Podcast' then @tracks << track end
+
+      # Set up a hash of albums.
       key = "#{track.artist},#{track.album}"
       unless @albums.has_key?(key)
         @albums[key] = Album.new(track.album, track.artist, track)
@@ -50,6 +53,7 @@ class ITunesAmazon
     end
   end
 
+  # Runs Amazon stats on each album
   def get_amazon_stats
     @albums.each_value do |album|
       album.get_amazon_info
@@ -88,7 +92,7 @@ class ITunesAmazon
   end
 
   def random_song(seed = nil)
-    unless seed == nil then srand(seed) end
+    srand seed if seed
     @tracks[rand(@tracks.size)]
   end
 
@@ -117,18 +121,26 @@ class ITunesAmazon
     fields.sort {|a, b| b[1] <=> a[1]}[0][0]
   end
 
-  def top_rated_amazon_albums(n)
-    top_albums = @albums.values.sort {|a, b| b.amazon_average_rating.to_f <=> a.amazon_average_rating.to_f}
+  def top_selling_amazon_albums(n)
+    top_albums = @albums.values.sort_by {|album| album.amazon_sales_rank.to_i }
     top_albums[0..(n-1)]
   end
 
-
+  def how_much_is_it_worth
+    amount = 0
+    @albums.values.each do |album|
+     #unless (track.time == nil && track.time.is_a?(Numeric)) then
+        amount += album.amazon_price.to_i
+     # end
+    end
+    amount
+  end
 
   def self.run(args)
     parser = ITunesAmazon.new(args[0])
 
-    album = parser.top_rated_amazon_albums(1)[0]
-    p "Top rated Amazon album: #{album.album} by #{album.artist} rated #{album.amazon_average_rating.to_f}"
+    album = parser.top_selling_amazon_albums(1)[0]
+    p "Top selling Amazon album: #{album.album} by #{album.artist} sell rank #{album.amazon_sales_rank}"
 
     top_track = parser.top_played_songs(1)[0]
     p "Top song: #{top_track.name} by #{top_track.artist}, played #{top_track.play_count} times"

data/test/test_i_tunes_amazon.rb

@@ -1,12 +1,11 @@
 #!/usr/bin/env ruby -w
 
-
 $: << 'lib'
 
 require "test/unit"
 require 'i_tunes_amazon.rb'
 
-class ITunesParserTester < Test::Unit::TestCase
+class ITunesAmazonTester < Test::Unit::TestCase
   @@parse_this = ITunesAmazon.new("../test.itunes.xml")
 
   def setup
@@ -73,5 +72,17 @@ class ITunesParserTester < Test::Unit::TestCase
     assert_equal expected, actual
   end
 
+  def test_amazon_top_selling_album
+    @parser.get_amazon_stats
+    actual = @parser.top_selling_amazon_albums(1)[0].album
+    expected = "Live In Cook County Jail"
+    assert_equal expected, actual
+  end
 
+  def test_how_much_is_it_worth
+    @parser.get_amazon_stats
+    actual = @parser.how_much_is_it_worth
+    expected = 3547
+    assert_equal expected, actual
+  end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: iTunesAmazon
 version: !ruby/object:Gem::Version 
-  version: 0.0.1
+  version: 0.0.2
 platform: ruby
 authors: 
 - Tracie Hlavka