1 | #!/usr/bin/perl -w |
---|
2 | |
---|
3 | # |
---|
4 | # This perl script is intended to perform movie data lookups based on |
---|
5 | # the themoviedb.org website |
---|
6 | # |
---|
7 | # For more information on MythVideo's external movie lookup mechanism, see |
---|
8 | # the README file in this directory. |
---|
9 | # |
---|
10 | # Author: William Stewart, Stuart Morgan |
---|
11 | # |
---|
12 | # v0.1 |
---|
13 | # - Initial script |
---|
14 | |
---|
15 | use File::Basename; |
---|
16 | use lib dirname($0); |
---|
17 | |
---|
18 | use MythTV::MythVideoCommon; |
---|
19 | |
---|
20 | eval "use DateTime::Format::Strptime"; my $has_date_format = $@ ? 0 : 1; |
---|
21 | |
---|
22 | use vars qw($opt_h $opt_r $opt_d $opt_i $opt_v $opt_D $opt_M $opt_P $opt_B); |
---|
23 | use Getopt::Std; |
---|
24 | use Data::Dumper; |
---|
25 | use LWP::Simple; |
---|
26 | use XML::Simple; |
---|
27 | |
---|
28 | $title = "themoviedb Query"; |
---|
29 | $version = "v0.1.0"; |
---|
30 | $author = "William Stewart, Stuart Morgan"; |
---|
31 | push(@MythTV::MythVideoCommon::URL_get_extras, ($title, $version)); |
---|
32 | |
---|
33 | my @countries = qw(USA UK Canada Japan); |
---|
34 | |
---|
35 | my $base_url = "http://api.themoviedb.org/2.0/"; |
---|
36 | my $posterimdb_url = "http://api.themoviedb.org/cover.php?imdb="; |
---|
37 | |
---|
38 | # themoviedb.org api key given by Travis Bell for Mythtv |
---|
39 | my $api_key = "c27cb71cff5bd76e1a7a009380562c62"; |
---|
40 | |
---|
41 | binmode(STDOUT, ":utf8"); |
---|
42 | |
---|
43 | # display usage |
---|
44 | sub usage { |
---|
45 | print "usage: $0 -hdrviMPD [parameters]\n"; |
---|
46 | print " -h help\n"; |
---|
47 | print " -d debug\n"; |
---|
48 | print " -r dump raw query result data only\n"; |
---|
49 | print " -v display version\n"; |
---|
50 | print " -i display info\n"; |
---|
51 | print "\n"; |
---|
52 | print " -M <query> get movie list\n"; |
---|
53 | print " -P <movieid> get movie poster\n"; |
---|
54 | print " -B <movieid> get movie backdrop\n"; |
---|
55 | print " -D <movieid> get movie data\n"; |
---|
56 | exit(-1); |
---|
57 | } |
---|
58 | |
---|
59 | # display 1-line of info that describes the version of the program |
---|
60 | sub version { |
---|
61 | print "$title ($version) by $author\n" |
---|
62 | } |
---|
63 | |
---|
64 | # display 1-line of info that can describe the type of query used |
---|
65 | sub info { |
---|
66 | print "Performs queries using the themoviedb.org website.\n"; |
---|
67 | } |
---|
68 | |
---|
69 | # display detailed help |
---|
70 | sub help { |
---|
71 | version(); |
---|
72 | info(); |
---|
73 | usage(); |
---|
74 | } |
---|
75 | |
---|
76 | # get Movie Data |
---|
77 | sub getMovieData { |
---|
78 | my ($movieid)=@_; # grab movieid parameter |
---|
79 | if (defined $opt_d) { printf("# looking for movie id: '%s'\n", $movieid);} |
---|
80 | |
---|
81 | # get the search results page via Movie.imdbLookup |
---|
82 | my $request = $base_url . "Movie.imdbLookup?imdb_id=tt" . $movieid . "&api_key=" . $api_key; |
---|
83 | if (defined $opt_d) { printf("# request: '%s'\n", $request); } |
---|
84 | my ($rc, $response) = myth_url_get($request); |
---|
85 | if (defined $opt_r) { printf("%s", $response); } |
---|
86 | |
---|
87 | my $xs = new XML::Simple(suppressempty => ''); |
---|
88 | my $xml = $xs->XMLin($response); |
---|
89 | |
---|
90 | if ($xml->{"opensearch:totalResults"} > 0) { |
---|
91 | # |
---|
92 | # now get the movie data via Movie.getInfo, Movie.imdbLookup does not provide us all the data |
---|
93 | # |
---|
94 | my $tmdbid = $xml->{moviematches}->{movie}->{id}; |
---|
95 | $request = $base_url . "Movie.getInfo?id=" . $tmdbid . "&api_key=" . $api_key; |
---|
96 | if (defined $opt_d) { printf("# request: '%s'\n", $request); } |
---|
97 | my ($rc, $response) = myth_url_get($request); |
---|
98 | if (defined $opt_r) { printf("%s", $response); } |
---|
99 | |
---|
100 | $xml = $xs->XMLin($response, KeyAttr => ['key', 'id'], forcearray => [ 'category']); |
---|
101 | |
---|
102 | my $title = $xml->{moviematches}->{movie}->{title}; |
---|
103 | my $releasedate = $xml->{moviematches}->{movie}->{release}; |
---|
104 | my $year = substr($releasedate, 0, 4); |
---|
105 | my $plot = $xml->{moviematches}->{movie}->{short_overview}; |
---|
106 | my $userrating = $xml->{moviematches}->{movie}->{rating}; |
---|
107 | my $runtime = $xml->{moviematches}->{movie}->{runtime}; |
---|
108 | my $budget = $xml->{moviematches}->{movie}->{budget}; |
---|
109 | my $revenue = $xml->{moviematches}->{movie}->{revenue}; |
---|
110 | my $trailer = $xml->{moviematches}->{movie}->{trailer}->{content}; |
---|
111 | my $homepage = $xml->{moviematches}->{movie}->{homepage}; |
---|
112 | my $country = $xml->{moviematches}->{movie}->{production_countries}->{country}->{name}; |
---|
113 | |
---|
114 | # Genre |
---|
115 | my @lgenres; |
---|
116 | my $j = 0; |
---|
117 | my @catlist = @{$xml->{moviematches}->{movie}->{categories}->{category}}; |
---|
118 | for (my $i = 0; $i < @catlist; $i++) |
---|
119 | { |
---|
120 | $lgenres[$j++] = $catlist[$i]->{name}; |
---|
121 | } |
---|
122 | my $genres = join(',', @lgenres); |
---|
123 | |
---|
124 | # People |
---|
125 | my @lcast; |
---|
126 | my @ldirector; |
---|
127 | my @lwriter; |
---|
128 | my $dc = 0; |
---|
129 | my $cc = 0; |
---|
130 | my $wc = 0; |
---|
131 | my @castlist = @{$xml->{moviematches}->{movie}->{people}->{person}}; |
---|
132 | for (my $i = 0; $i < @castlist; $i++) |
---|
133 | { |
---|
134 | if ($castlist[$i]->{job} eq "actor") |
---|
135 | { |
---|
136 | $lcast[$cc++] = $castlist[$i]->{name}; |
---|
137 | } |
---|
138 | |
---|
139 | if ($castlist[$i]->{job} eq "director") |
---|
140 | { |
---|
141 | $ldirector[$dc++] = $castlist[$i]->{name}; |
---|
142 | } |
---|
143 | |
---|
144 | if ($castlist[$i]->{job} eq "writer") |
---|
145 | { |
---|
146 | $lwriter[$wc++] = $castlist[$i]->{name}; |
---|
147 | } |
---|
148 | |
---|
149 | } |
---|
150 | my $cast = join(',', @lcast); |
---|
151 | my $director = join(',', @ldirector); |
---|
152 | my $writer = join(',', @lwriter); |
---|
153 | |
---|
154 | # output fields (these field names must match what MythVideo is looking for) |
---|
155 | print "Title:$title\n"; |
---|
156 | print "Year:$year\n"; |
---|
157 | print "ReleaseDate:$releasedate\n"; |
---|
158 | print "Director:$director\n"; |
---|
159 | print "Plot:$plot\n"; |
---|
160 | print "UserRating:$userrating\n"; |
---|
161 | # print "MovieRating:$movierating\n"; |
---|
162 | print "Runtime:$runtime\n"; |
---|
163 | print "Writers: $writer\n"; |
---|
164 | print "Cast:$cast\n"; |
---|
165 | print "Genres: $genres\n"; |
---|
166 | if ($country) {print "Countries: $country\n";} |
---|
167 | print "Budget: $budget\n"; |
---|
168 | print "Revenue: $revenue\n"; |
---|
169 | if ($trailer) {print "trailer: $trailer\n";} |
---|
170 | if ($homepage) {print "Homepage: $homepage\n";} |
---|
171 | } |
---|
172 | } |
---|
173 | |
---|
174 | # dump Movie Poster |
---|
175 | sub getMoviePoster { |
---|
176 | my ($movieid)=@_; # grab movieid parameter |
---|
177 | if (defined $opt_d) { printf("# looking for poster for movie id: '%s'\n", $movieid);} |
---|
178 | |
---|
179 | # get the search results page |
---|
180 | my $request = $base_url . "Movie.imdbLookup?imdb_id=tt" . $movieid . "&api_key=" . $api_key; |
---|
181 | if (defined $opt_d) { printf("# request: '%s'\n", $request); } |
---|
182 | my ($rc, $response) = myth_url_get($request); |
---|
183 | if (defined $opt_r) { printf("%s", $response); } |
---|
184 | |
---|
185 | my $xml = XMLin($response, forcearray => [ 'poster', 'backdrop' ]); |
---|
186 | |
---|
187 | if ($xml->{"opensearch:totalResults"} > 0) { |
---|
188 | my $poster = $xml->{moviematches}->{movie}->{poster}->[0]->{content}; |
---|
189 | |
---|
190 | if ($poster) { |
---|
191 | # check for a mid sized poster |
---|
192 | foreach my $p (@{$xml->{moviematches}->{movie}->{poster}}) |
---|
193 | { |
---|
194 | if ($p->{size} eq "mid") |
---|
195 | { |
---|
196 | $poster = $p->{content}; |
---|
197 | last; |
---|
198 | } |
---|
199 | } |
---|
200 | |
---|
201 | if ($poster) { |
---|
202 | print "$poster\n"; |
---|
203 | } |
---|
204 | } |
---|
205 | # else { |
---|
206 | # my $backdrop = $xml->{moviematches}->{movie}->{backdrop}->[0]->{content}; |
---|
207 | # |
---|
208 | # if ($backdrop) |
---|
209 | # { |
---|
210 | # print "$backdrop\n"; |
---|
211 | # } |
---|
212 | # } |
---|
213 | } |
---|
214 | } |
---|
215 | |
---|
216 | # dump Movie Backdrop |
---|
217 | sub getMovieBackdrop { |
---|
218 | my ($movieid)=@_; # grab movieid parameter |
---|
219 | if (defined $opt_d) { printf("# looking for backdrop for movie id: '%s'\n", $movieid);} |
---|
220 | |
---|
221 | # get the search results page via Movie.imdbLookup |
---|
222 | my $request = $base_url . "Movie.imdbLookup?imdb_id=tt" . $movieid . "&api_key=" . $api_key; |
---|
223 | if (defined $opt_d) { printf("# request: '%s'\n", $request); } |
---|
224 | my ($rc, $response) = myth_url_get($request); |
---|
225 | if (defined $opt_r) { printf("%s", $response); } |
---|
226 | |
---|
227 | my $xs = new XML::Simple(suppressempty => ''); |
---|
228 | my $xml = $xs->XMLin($response); |
---|
229 | |
---|
230 | if ($xml->{"opensearch:totalResults"} > 0) { |
---|
231 | # |
---|
232 | # now get the movie data via Movie.getInfo, Movie.imdbLookup does not provide us all the data |
---|
233 | # |
---|
234 | my $tmdbid = $xml->{moviematches}->{movie}->{id}; |
---|
235 | $request = $base_url . "Movie.getInfo?id=" . $tmdbid . "&api_key=" . $api_key; |
---|
236 | if (defined $opt_d) { printf("# request: '%s'\n", $request); } |
---|
237 | my ($rc, $response) = myth_url_get($request); |
---|
238 | if (defined $opt_r) { printf("%s", $response); } |
---|
239 | |
---|
240 | $xml = XMLin($response, KeyAttr => ['key', 'id']); |
---|
241 | |
---|
242 | foreach my $backdrop (@{$xml->{moviematches}->{movie}->{backdrop}}) |
---|
243 | { |
---|
244 | # print "$backdrop->{content}\n"; |
---|
245 | |
---|
246 | if ($backdrop->{size} eq "original") |
---|
247 | { |
---|
248 | print "$backdrop->{content}\n"; |
---|
249 | } |
---|
250 | } |
---|
251 | } |
---|
252 | } |
---|
253 | |
---|
254 | # dump Movie list: 1 entry per line, each line as 'movieid:Movie Title' |
---|
255 | sub getMovieList { |
---|
256 | my ($filename)=@_; # grab parameters |
---|
257 | |
---|
258 | # |
---|
259 | # Convert filename into a query string |
---|
260 | # (use same rules that Metadata::guesTitle does) |
---|
261 | my $query = cleanTitleQuery($filename); |
---|
262 | |
---|
263 | if (defined $opt_d) { |
---|
264 | printf("# query: '%s'\n", $query); |
---|
265 | } |
---|
266 | |
---|
267 | # get the search results page |
---|
268 | my $request = $base_url . "Movie.search?title=" . $query . "&api_key=" . $api_key; |
---|
269 | if (defined $opt_d) { printf("# request: '%s'\n", $request); } |
---|
270 | my ($rc, $response) = myth_url_get($request); |
---|
271 | if (defined $opt_r) { |
---|
272 | print $response; |
---|
273 | exit(0); |
---|
274 | } |
---|
275 | |
---|
276 | my $xs = new XML::Simple(suppressempty => ''); |
---|
277 | my $xml = $xs->XMLin($response, forcearray => [ 'movie' ]); |
---|
278 | |
---|
279 | if ($xml->{"opensearch:totalResults"} > 0) { |
---|
280 | my @movies; |
---|
281 | my $j = 0; |
---|
282 | |
---|
283 | foreach my $key (keys(%{$xml->{moviematches}->{movie}})) |
---|
284 | { |
---|
285 | my $movienum = $xml->{moviematches}->{movie}->{$key}->{imdb}; |
---|
286 | my $moviename = $xml->{moviematches}->{movie}->{$key}->{title}; |
---|
287 | my $release = $xml->{moviematches}->{movie}->{$key}->{release}; |
---|
288 | my $movieyear = 0; |
---|
289 | |
---|
290 | if ($release) { |
---|
291 | $movieyear = substr($release, 0, 4); |
---|
292 | } |
---|
293 | |
---|
294 | if ($movienum) { |
---|
295 | if ($release) { |
---|
296 | $movies[$j++]= substr($movienum,2) . ":" . $moviename . " (". $movieyear . ")"; |
---|
297 | } |
---|
298 | else { |
---|
299 | $movies[$j++]= substr($movienum,2) . ":" . $moviename; |
---|
300 | } |
---|
301 | } |
---|
302 | } |
---|
303 | # display array of values |
---|
304 | |
---|
305 | for $movie (@movies) { print "$movie\n"; } |
---|
306 | } |
---|
307 | } |
---|
308 | |
---|
309 | # |
---|
310 | # Main Program |
---|
311 | # |
---|
312 | |
---|
313 | # parse command line arguments |
---|
314 | getopts('ohrdivDMPB'); |
---|
315 | |
---|
316 | # print out info |
---|
317 | if (defined $opt_v) { version(); exit 1; } |
---|
318 | if (defined $opt_i) { info(); exit 1; } |
---|
319 | |
---|
320 | # print out usage if needed |
---|
321 | if (defined $opt_h || $#ARGV<0) { help(); } |
---|
322 | |
---|
323 | if (defined $opt_D) { |
---|
324 | # take movieid from cmdline arg |
---|
325 | $movieid = shift || die "Usage : $0 -D <movieid>\n"; |
---|
326 | getMovieData($movieid); |
---|
327 | } |
---|
328 | |
---|
329 | elsif (defined $opt_P) { |
---|
330 | # take movieid from cmdline arg |
---|
331 | $movieid = shift || die "Usage : $0 -P <movieid>\n"; |
---|
332 | getMoviePoster($movieid); |
---|
333 | } |
---|
334 | |
---|
335 | elsif (defined $opt_B) { |
---|
336 | # take movieid from cmdline arg |
---|
337 | $movieid = shift || die "Usage : $0 -P <movieid>\n"; |
---|
338 | getMovieBackdrop($movieid); |
---|
339 | } |
---|
340 | |
---|
341 | elsif (defined $opt_M) { |
---|
342 | # take query from cmdline arg |
---|
343 | $query = shift || die "Usage : $0 -M <query>\n"; |
---|
344 | getMovieList($query); |
---|
345 | } |
---|
346 | # vim: set expandtab ts=3 sw=3 : |
---|