app/models/torrent.rb app/views/movies/_movie.erb app/views/movies/index.html.erb app/views/movies/index.rss.builder app/views/movies/show.erb app/views/movies/show2.html.haml db/migrate/20081211160533_create_torrents.rb lib/tasks/fetch.rake test/fixtures/torrents.yml test/unit/torrent_test.rb @@ -7,21 +7,5 @@ class MoviesController < ApplicationController def show @movie = Movie.find(params[:id]) end - - def fetch - count = Movie.fetch_from_this_month - if count == 0 - flash[:notice] = "Nothing new, sorry!" - else - flash[:notice] = "I've got #{count} movies!" - end - redirect_to movies_path - end - - def rss - headers["Content-Type"] = "application/xml" - @movies = Movie.all - render :layout => false - end end \ No newline at end of file app/controllers/movies_controller.rb @@ -1,32 +1,10 @@ -class Movie < ActiveRecord::Base - named_scope :recent, lambda { {:conditions => ['created_at > ?', 5.days.ago]}} +class Movie < ActiveRecord::Base - belongs_to :genre + has_many :torrents validates_presence_of :title, :imdb_id - validates_uniqueness_of :imdb_id, :title - - def self.fetch_from_this_month - movies = [] - ids = Movie.latest_ids - # sciagamy pełne informacje o filmach - imdb = IMDB::FullInformation.new - ids.each {|imdb_id| movies << imdb.information(imdb_id)[:result] } - Movie.bulk_save(movies) # zapis w bazie - end - - def self.latest_ids - # sciagamy idki imdb nowosci z tego miesiaca - IMDB::NowPlaying.new.movies(Time.now.year, Time.now.month)[:results].map{|movie| movie[:imdb_id]} - end - - protected - - def self.bulk_save(movies = nil) - count = 0 - movies[0..100].each { |m| count +=1 if Movie.create(:title => m[:title], :imdb_id => m[:imdb_id]).valid? } - count - end + validates_uniqueness_of :imdb_id + validates_numericality_of [:votes,:rating, :year] def self.fake_movies {:query=>{:month=>6, :year=>2008}, :results=>[{:imdb_id=>"0960144", :title=>"You Don't Mess with the Zohan"}, {:imdb_id=>"0416044", :title=>"Mongol"}, app/models/movie.rb @@ -5,6 +5,7 @@ %title Movie Pole = stylesheet_link_tag "application" = javascript_include_tag :defaults + = auto_discovery_link_tag(:rss, movies_path(:format => :rss)) %body #doc #hd app/views/layouts/application.html.haml @@ -16,6 +16,7 @@ module IMDB info[:release_date] = parse_release_date(main_div) info[:genres] = parse_genres(main_div) info[:tagline] = parse_tagline(main_div) + info[:imdb_id] = imdb_id result[:result] = info result end backends/imdb/lib/full_information.rb @@ -2,9 +2,9 @@ require "rubygems" require "open-uri" require "hpricot" -require "backends/imdb/lib/power_search" -require "backends/imdb/lib/now_playing" -require "backends/imdb/lib/full_information" +require File.expand_path(File.dirname(__FILE__) + "/power_search") +require File.expand_path(File.dirname(__FILE__) + "/now_playing") +require File.expand_path(File.dirname(__FILE__) + "/full_information") module IMDB BASE_URI = "http://www.imdb.com" backends/imdb/lib/imdb.rb @@ -1,14 +1,21 @@ require File.expand_path(File.dirname(__FILE__) + "/../spec_helper") describe IMDB::NowPlaying, "with 2009-07 fixture" do + before do @now_playing = IMDB::NowPlaying.new @now_playing.stub!(:open).and_return(open(File.dirname(__FILE__) + "/../fixtures/2009-07.html")) end - it "should parse HTML and return hash" do + it "should parse HTML and return hash with query" do + result = @now_playing.movies(2009, 07) + result.should include(:query) + result[:query].should include(:month) + result[:query].should include(:year) + end + + it "should parse HTML and return hash with results" do result = @now_playing.movies(2009, 07) - result.should include(:query => { :month => 7, :year => 2009 }) results = result[:results] results.should include(:imdb_id => "0417741", :title => "Harry Potter and the Half-Blood Prince") results.should include(:imdb_id => "0775552", :title => "They Came from Upstairs") backends/imdb/spec/lib/now_playing_spec.rb @@ -4,7 +4,7 @@ class AddMoreAttrToMovies < ActiveRecord::Migration add_column :movies, :rating, :integer add_column :movies, :votes, :integer add_column :movies, :genre_id, :integer - add_column :movies, :imdb_id, :integer + add_column :movies, :imdb_id, :integer end def self.down db/migrate/20081208150509_add_more_attr_to_movies.rb app/models/genre.rb app/views/movies/index.html.haml app/views/movies/rss.rxml app/views/movies/show.html.haml c94c9ef11622b22e6a734197f99ddfa6e141b59d marek marek@.(none) http://github.com/ncr/movie_pole/commit/2a24ad95ef1ae3efe8a7823d6c07ad2e48251c93 2a24ad95ef1ae3efe8a7823d6c07ad2e48251c93 2008-12-11T10:30:11-08:00 2008-12-11T10:30:11-08:00 added rake tasks for populating movies table from IMDB backend added torrents table added some erb views for education purposes :) b364ae32988e503f4a7aba17bd5e73a8c2a6140d marek marek@.(none)