File Coverage

blib/lib/WWW/Netflix.pm

Criterion	Covered	Total	%
statement	9	71	12.6
branch	0	34	0.0
condition	0	6	0.0
subroutine	3	11	27.2
pod	8	8	100.0
total	20	130	15.3

package WWW::Netflix;

use strict;
use warnings;

use WWW::Mechanize;

our $VERSION = 0.05;

sub new {
    my $ref = shift;
    my $class = ref( $ref ) || $ref;

    my $self = bless {
        www => new WWW::Mechanize(),
    }, $class;

    return $self;
}

sub login {
    my ( $self, $user, $pass ) = @_;
    die "Netflix requires a username and password"
        unless ( $user && $pass );

    $self->{ www }->get('http://www.netflix.com/Login');
    die "couldn't find login form"
        unless ( $self->{ www }->content =~ /login_form/ );

    $self->{ www }->form_name( 'login_form' );
    $self->{ www }->set_fields(
        email     => $user,
        password1 => $pass,
    );
    $self->{ www }->submit();
    die "login incorrect\n"
        if ( $self->{ www }->content =~ /does not match an account/ );
    $self->{ www }->get("http://www.netflix.com");
    die ( "not logged in" )
        unless ( $self->{ www }->content =~ /Your Account/ );
}

sub getRatings {
    my ( $self ) = @_;

    my $movies;
    my $body = 'alt="Next"';
    my $cur = 1;

    # loop through each page of ratings
    # the page that is being checked for a 'Next' button
    # is not the page that is being scraped
    while ( $body =~ /alt="Next"/i ) {
        $self->{ www }->get( "http://www.netflix.com/MoviesYouveSeen?lnkctr=wizMovUC&pageNum=$cur" );

        # the page gets returned even if not logged in or
        # if you don't have any ratings,
        # but this text doesn't appear unless you are logged in
        # and you have ratings
        return
            unless ( $self->{ www }->content =~ /this is the list of movies you've seen/ );
        $body = $self->{ www }->content();

        # loop through each movie on the page
        #  id="stM60025026_1" class=star alt="5.0 Stars">
        while ( $body =~ /id="stM(\d+)_[^ ]+ class=star alt="(\d).0 Stars/gs ) {
            my ( $movie_id, $rating ) = ( $1, $2 );
            $movies->{ $movie_id }->{ rating } = $rating;
        } # end of looping through movies

        # Adaptation
        while ( $body =~ /
73
            $movies->{ $2 }->{ title } = $1;
        }
        ++$cur;
    } # end of looping through pages

    # return all of the ratings as a hashref
    return $self->{ rated_movies } = $movies;
}

sub getRating {
    my ( $self, $movie_id ) = @_;
    $self->getRatings unless ( exists $self->{ rated_movies } );
    return $self->{ rated_movies }{ $movie_id }{ rating };
}

sub getTitle {
    my ( $self, $movie_id ) = @_;
    $self->getRatings unless ( exists $self->{ rated_movies } );
    return $self->{ rated_movies }{ $movie_id }{ title };
}


sub setRating {
    my ( $self, $movie_id, $rating ) = @_;

    die 'movie id must be an integer'
        unless ( $movie_id =~ /^\d+$/ );
    die 'rating must be an integer between 1 and 5'
        unless ( $rating =~ /^[1-5]$/ );

    # my $uri = "http://www.netflix.com/SetRating?movieid=${movie_id}&value=${rating}&url=http%3A%2F%2Fwww.netflix.com%2FMovieDisplay%3Fmovieid%3D${movie_id}";
    my $uri = "http://www.netflix.com/SetRating?movieid=${movie_id}&value=${rating}";
    $self->{ www }->get( $uri );

    # keep our hash up to date with reality
    $self->{ rated_movies }{ $movie_id }{ rating } = $rating if ( defined $self->{ rated_movies } );
}


sub getQueue {
  my ( $self ) = @_;

    $self->{www}->get( 'http://www.netflix.com/Queue' );
    my $body = $self->{www}->content();

    # this relies on the current state of the HTML where
    # each of the headers and movies is on its own line
	$self->{ queue }{ home } = [];
    $self->{ queue }{ queued } = [];
    $self->{ queue }{ saved }  = [];
    my $section = 'throwaway';
    for my $line ( split "\n", $body ) {
        chomp $line;
        if ( $line =~ m{DVDs At Home} ) {
            $section = 'home';
        }
        elsif ( $line =~ m{DVD Queue} ) {
            $section = 'queued';
        }
        elsif ( $line  =~ m{Saved DVDs} ) {
            $section = 'saved';
        }
        elsif ( $line =~ m{
136
            # $1 is movie ID; $2 is movie title
            push @{ $self->{ queue }{ $section } }, [ $1 => $2 ]
                unless ( $section eq 'throwaway' ); # throwaway = recently watched movies
       }
    }

    # er, returning a hashref that the using code could modify ... that's a no-no.
    return $self->{ queue };
}

sub queueMovie {
  my ( $self, $id ) = @_;

    die "Movie ID must be a series of one or more numerical digits.\n"
        unless $id =~ /^\d+$/;

    #http://www.netflix.com/AddToQueue?movieid=70027897&ftype=DD&trkid=199898

    my $url = "http://www.netflix.com/AddToQueue?movieid=$id";
    $self->{ www }->get( $url );

    return $self->{ www }->uri() =~ /QueueAddConfirmation/ ? 1 : 0;
}



1;

__END__