X-Git-Url: https://jfr.im/git/erebus.git/blobdiff_plain/b13575aa6e013f8f50fba5ae169e871b5cb233dc..325937f6d7b74f3855e646a44364401f55c39ce8:/modules/youtube.py

diff --git a/modules/youtube.py b/modules/youtube.py
index cd62f5f..38b4470 100644
--- a/modules/youtube.py
+++ b/modules/youtube.py
@@ -18,12 +18,15 @@ modstop = lib.modstop
 
 # module code
 import re
+import json
 import urllib2
+import urlparse
 import HTMLParser
 from BeautifulSoup import BeautifulSoup
 
 checkfor = "youtube"
 url_regex = re.compile('http[s]?://(?:[a-zA-Z]|[0-9]|[$-_@.&+]|[!*\(\),]|(?:%[0-9a-fA-F][0-9a-fA-F]))+')
+yturl_regex = re.compile('(http|ftp|https):\/\/([\w\-_]+(?:(?:\.[\w\-_]+)+))([\w\-\.,@?^=%&:/~\+#]*[\w\-\@?^=%&/~\+#])?')
 
 @lib.hooknum("PRIVMSG")
 def privmsg_hook(bot, line):
@@ -35,9 +38,17 @@ def privmsg_hook(bot, line):
 	if checkfor not in line:
 		return # doesn't concern us
 
+	print "Meow"
 	for url in url_regex.findall(linetx):
 		if checkfor in url:
-			html_parser = HTMLParser.HTMLParser()
-			respdata = urllib2.urlopen(url).read()
-			soup = BeautifulSoup(respdata)
-			bot.msg(line.split()[2], BeautifulSoup(soup.title.string, convertEntities=BeautifulSoup.HTML_ENTITIES))
+			url_data = urlparse.urlparse(url)
+			query = urlparse.parse_qs(url_data.query)
+			video = query["v"][0]
+			api_url = 'http://gdata.youtube.com/feeds/api/videos/%s?alt=json&v=2' % video
+			respdata = urllib2.urlopen(api_url).read()
+			video_info = json.loads(respdata)
+
+			title = video_info['entry']['title']["$t"]
+			author = video_info['entry']['author'][0]['name']['$t']
+
+			bot.msg(line.split()[2], "Youtube: %s (%s)" % (title, author))