~bnrubin/supytube/main : revision 4

1

###

2

3

4

#

5

# Redistribution and use in source and binary forms, with or without

6

# modification, are permitted provided that the following conditions are met:

7

#

8

# * Redistributions of source code must retain the above copyright notice,

9

# this list of conditions, and the following disclaimer.

10

# * Redistributions in binary form must reproduce the above copyright notice,

11

# this list of conditions, and the following disclaimer in the

12

# documentation and/or other materials provided with the distribution.

13

# * Neither the name of the author of this software nor the name of

14

# contributors to this software may be used to endorse or promote products

15

# derived from this software without specific prior written consent.

16

#

17

# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"

18

# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE

19

# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE

20

# ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE

21

# LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR

22

# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF

23

# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS

24

# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN

25

# CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)

26

# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE

27

# POSSIBILITY OF SUCH DAMAGE.

28

29

###

30

31

import supybot.utils as utils

32

from supybot.commands import *

33

import supybot.plugins as plugins

34

import supybot.ircutils as ircutils

35

import supybot.callbacks as callbacks

36

import supybot.ircmsgs as ircmsgs

37

38

import supybot.conf as conf

39

import supybot.log as log

40

from urlparse import urlparse

41

import urllib

42

43

import xml.sax

44

45

class YoutubeHandler(xml.sax.handler.ContentHandler):

46

def __init__(self):

47

self.inTitle = 0

48

self.inAuthor = 0

49

self.inRating = 0

50

self.inRatingCount = 0

51

self.inViews = 0

52

self.title = ''

53

self.author = ''

54

self.rating = ''

55

self.rating_count = ''

56

self.views = ''

57

58

59

def startElement(self, name, attributes):

60

if(name == 'title'):

61

self.title = ''

62

self.inTitle = 1

63

if(name == 'author'):

64

self.author = ""

65

self.inAuthor = 1

66

if(name == 'rating_avg'):

67

self.rating = ''

68

self.inRating = 1

69

if(name == 'rating_count'):

70

self.rating_count = ''

71

self.inRatingCount = 1

72

if(name == 'view_count'):

73

self.views = ''

74

self.inViews = 1

75

76

def characters(self,data):

77

if(self.inTitle):

78

self.title += data

79

if(self.inAuthor):

80

self.author += data

81

if(self.inRating):

82

self.rating += data

83

if(self.inRatingCount):

84

self.rating_count += data

85

if(self.inViews):

86

self.views += data

87

def endElement(self,name):

88

if(name == 'title'):

89

self.inTitle=0

90

if(name == 'author'):

91

self.inAuthor = 0

92

if(name == 'rating_avg'):

93

self.inRating = 0

94

if(name == 'rating_count'):

95

self.inRatingCount = 0

96

self.rating = (float(self.rating)/5)*100

97

if(name == 'view_count'):

98

self.inViews = 0

99

100

class Supytube(callbacks.Plugin):

101

"""Add the help for "@plugin help Supytube" here

102

This should describe *how* to use this plugin."""

103

threaded = True

104

105

def doPrivmsg(self, irc, msg):

106

if(self.registryValue('enable',msg.args[0])):

107

# If this is a youtube link, commence lookup

108

if(msg.args[1].find("youtube") != -1):

109

for word in msg.args[1].split(' '):

110

if(word.find("youtube") != -1):

111

videoid = urlparse(word)[4].split('=')[1].split('&')[0]

112

devid = self.registryValue('devid')

113

#log.critical(videoid)

114

#log.critical(devid)

115

f = urllib.urlopen('http://www.youtube.com/api2_rest?method=youtube.videos.get_details&dev_id=%s&video_id=%s' % (devid,videoid))

116

parser = xml.sax.make_parser()

117

handler = YoutubeHandler()

118

parser.setContentHandler(handler)

119

parser.parse(f)

120

#log.critical(handler.title)

121

irc.queueMsg(ircmsgs.privmsg(msg.args[0],'Title: %s, Views: %s, Rating: %s%%' % (ircutils.bold(handler.title), ircutils.bold(handler.views),ircutils.bold(handler.rating))))

122

123

124

125

126

127

Class = Supytube

128

129

130

# vim:set shiftwidth=4 tabstop=4 expandtab textwidth=79: