← Back to branch summary

~ubuntu-es.bot/ubuntu-es-bot/botijo-trunk

« back to all changes in this revision

Viewing changes to data/rbot/plugins/imdb.rb.disabled

  • Committer: Jonathan Patrick Davies
  • Date: 2007-11-11 18:40:59 UTC
  • Revision ID: jpatrick@kubuntu.org-20071111184059-cljqhatzo183hg3f
No necesitamos estos plugins.

Show diffs side-by-side

added added

removed removed

Lines of Context:
data/rbot/plugins/imdb.rb.disabled
 
1
# IMDB plugin for RubyBot
 
2
# (c) 2005 Arnaud Cornet <arnaud.cornet@gmail.com>
 
3
# Licensed under MIT License.
 
4
 
 
5
require 'net/http'
 
6
require 'cgi'
 
7
require 'uri/common'
 
8
 
 
9
class Imdb
 
10
  def initialize(bot)
 
11
    @bot = bot
 
12
  end
 
13
 
 
14
  def search(rawstr)
 
15
    str = URI.escape(rawstr)
 
16
    @http = @bot.httputil.get_proxy(URI.parse("http://us.imdb.com/find?q=#{str}"))
 
17
    @http.start
 
18
    begin
 
19
    resp, data = @http.get("/find?q=#{str}", "User-Agent" => "Mozilla/5.0")
 
20
    rescue Net::ProtoRetriableError => detail
 
21
      head = detail.data
 
22
      if head.code == "301" or head.code == "302"
 
23
            return head['location'].gsub(/http:\/\/us.imdb.com/, "").gsub(/\?.*/, "")
 
24
        end
 
25
    end
 
26
    if resp.code == "200"
 
27
      m = /<a href="(\/title\/tt[0-9]+\/?)[^"]*"(:?[^>]*)>([^<]*)<\/a>/.match(resp.body)
 
28
      if m
 
29
        url = m[1]
 
30
        title = m[2]
 
31
        return url
 
32
      end
 
33
    elsif resp.code == "302"
 
34
      return resp['location'].gsub(/http:\/\/us.imdb.com/, "").gsub(/\?.*/, "")
 
35
    end
 
36
    return nil
 
37
  end
 
38
 
 
39
  def info(rawstr)
 
40
    sr = search(rawstr)
 
41
    if !sr
 
42
      debug "IMDB: search returned NIL"
 
43
      return nil
 
44
    end
 
45
    resp, data = @http.get(sr, "User-Agent" =>
 
46
      "Mozilla/5.0 (compatible; Konqueror/3.1; Linux)")
 
47
    if resp.code == "200"
 
48
      m = /<title>([^<]*)<\/title>/.match(resp.body)
 
49
      return nil if !m
 
50
      title = CGI.unescapeHTML(m[1])
 
51
 
 
52
      m = /<b>([0-9.]+)\/10<\/b> \(([0-9,]+) votes?\)/.match(resp.body)
 
53
      return nil if !m
 
54
      score = m[1]
 
55
      votes = m[2]
 
56
 
 
57
      genre = Array.new
 
58
      resp.body.scan(/<a href="\/Sections\/Genres\/[^\/]+\/">([^<]+)<\/a>/) do |gnr|
 
59
        genre << gnr
 
60
      end
 
61
      return ["http://us.imdb.com" + sr, title, score, votes,
 
62
        genre]
 
63
    end
 
64
    return nil
 
65
  end
 
66
end
 
67
 
 
68
class ImdbPlugin < Plugin
 
69
  def help(plugin, topic="")
 
70
    "imdb <string> => search http://www.imdb.org for <string>"
 
71
  end
 
72
 
 
73
  def privmsg(m)
 
74
    unless(m.params && m.params.length > 0)
 
75
      m.reply "incorrect usage: " + help(m.plugin)
 
76
      return
 
77
    end
 
78
 
 
79
    i = Imdb.new(@bot)
 
80
    info = i.info(m.params)
 
81
    if !info
 
82
      m.reply "Nothing found for #{m.params}"
 
83
      return nil
 
84
    end
 
85
    m.reply "#{info[1]} : #{info[0]}"
 
86
    m.reply "Ratings: #{info[2]}/10 (#{info[3]} voters). Genre: #{info[4].join('/')}"
 
87
  end
 
88
end
 
89
 
 
90
plugin = ImdbPlugin.new
 
91
plugin.register("imdb")
 
92