X-Git-Url: https://git.netwichtig.de/gitweb/?a=blobdiff_plain;f=data%2Frbot%2Fplugins%2Fimdb.rb;h=57cae635d4535bb75cccc1c1dfaa6360ed6b12ee;hb=6cf365c49ce5fbe24c0a4ff0663550390b501fea;hp=a12b77843073cbed34f71f850b0dd96add8f9dfe;hpb=b4e07b8fb4f1c79cc8c6a09fcb18bb3d4e25ea98;p=user%2Fhenk%2Fcode%2Fruby%2Frbot.git
diff --git a/data/rbot/plugins/imdb.rb b/data/rbot/plugins/imdb.rb
index a12b7784..57cae635 100644
--- a/data/rbot/plugins/imdb.rb
+++ b/data/rbot/plugins/imdb.rb
@@ -15,7 +15,7 @@ class Imdb
IMDB = "http://www.imdb.com"
TITLE_OR_NAME_MATCH = /]*)>([^<]*)<\/a>/
TITLE_MATCH = /]*)>([^<]*)<\/a>/
- NAME_MATCH = /]*)>([^<]*)<\/a>/
+ NAME_MATCH = /]*)>([^<]*)<\/a>/
CHAR_MATCH = /]*)>([^<]*)<\/a>/
CREDIT_NAME_MATCH = /#{NAME_MATCH}\s*<\/td>\s*
]+>\s*\.\.\.\s*<\/td>\s* | ]+>\s*(.+?)\s*<\/td>/m
FINAL_ARTICLE_MATCH = /, ([A-Z]\S{0,2})$/
@@ -33,8 +33,12 @@ class Imdb
end
def search(rawstr, rawopts={})
+ # allow the user to search directly for (movie) IDs
+ if rawstr.match /$tt\d+^/ then
+ return ["/movie/#{rawstr}/"]
+ end
str = CGI.escape(rawstr)
- str << ";site=aka" if @bot.config['imdb.aka']
+ str << "&site=aka" if @bot.config['imdb.aka']
opts = rawopts.dup
opts[:type] = :both unless opts[:type]
return do_search(str, opts)
@@ -128,6 +132,8 @@ class Imdb
def info_title(sr, opts={})
resp = nil
begin
+ # movie urls without tailing / trigger a redirect
+ sr += '/' if sr[-1,1] != '/'
resp = @bot.httputil.get_response(IMDB + sr, :max_redir => -1)
rescue Exception => e
error e.message
@@ -175,14 +181,18 @@ class Imdb
end
ratings = "no votes"
- m = resp.body.match(/([0-9.]+)<\/b>]+>\/10<\/span><\/span>\s*[^<]+]*href="ratings"[^>]+>([0-9,]+) votes?<\/a>/m)
- if m
- ratings = "#{m[1]}/10 (#{m[2]} voters)"
+ # parse imdb rating value:
+ if resp.body.match(/itemprop="ratingValue">([^<]+))
+ ratings = "#{$1}/10"
+ end
+ # parse imdb rating count:
+ if resp.body.match(/itemprop="ratingCount">([^<]+))
+ ratings += " (#{$1} voters)"
end
genre = Array.new
- resp.body.scan(/([^<]+)<\/a>/) do |gnr|
- genre << gnr
+ resp.body.scan(/([^<]+)<\/a>/) do |gnr|
+ genre << gnr.first.strip
end
plot = resp.body.match(DESC_MATCH)[3] rescue nil
@@ -309,8 +319,8 @@ class Imdb
}
end
- urls.map { |url|
- info = info_name(url, :movies_by_year => true)
+ urls.map { |u|
+ info = info_name(u, :movies_by_year => true)
debug info.inspect
|