lib/imdb_party/imdb.rb in maksar-imdb_party-0.6.2 vs lib/imdb_party/imdb.rb in maksar-imdb_party-0.6.3

@@ -39,11 +39,11 @@
         results["data"]["results"].each do |result_section|
           result_section["list"].each do |r|
             next unless r["tconst"] && r["title"]
             h = {:title => r["title"], :year => r["year"], :imdb_id => r["tconst"], :kind => r["type"]}
             h.merge!(:poster_url => r["image"]["url"]) if r["image"] && r["image"]["url"]
-            movie_results << h
+            movie_results << SearchResult.new(h)
           end
         end
       end
       
       movie_results
@@ -58,17 +58,17 @@
     
     def top_250
       url = build_url('/chart/top')
 
       results = self.class.get(url).parsed_response
-      results["data"]["list"]["list"].map { |r| {:title => r["title"], :imdb_id => r["tconst"], :year => r["year"], :poster_url => (r["image"] ? r["image"]["url"] : nil)} }
+      results["data"]["list"]["list"].map { |r| SearchResult.new({:title => r["title"], :imdb_id => r["tconst"], :year => r["year"], :poster_url => (r["image"] ? r["image"]["url"] : nil)}) }
     end
 
     def popular_shows
       url = build_url('/chart/tv')
 
       results = self.class.get(url).parsed_response
-      results["data"]["list"].map { |r| {:title => r["title"], :imdb_id => r["tconst"], :year => r["year"], :poster_url => (r["image"] ? r["image"]["url"] : nil)} }
+      results["data"]["list"].map { |r| SearchResult.new({:title => r["title"], :imdb_id => r["tconst"], :year => r["year"], :poster_url => (r["image"] ? r["image"]["url"] : nil)}) }
     end
     
   end
 end