ChCh/CloudBot
Archived
11
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

Improved title plugin

Browse source
This commit is contained in:
Luke Rogers 2012-10-12 23:01:37 +13:00
parent cf377e1800
commit 7d9b3c35fe
1 changed files with 8 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

13
plugins/title.py
View file

@ -1,4 +1,5 @@
from util import hook, http, urlnorm from util import hook, http, urlnorm
from bs4 import BeautifulSoup
@hook.command @hook.command
@ -7,13 +8,15 @@ def title(inp):
url = urlnorm.normalize(inp.encode('utf-8'), assume_scheme="http") url = urlnorm.normalize(inp.encode('utf-8'), assume_scheme="http")
try: try:
page = http.get_html(url) page = http.open(url)
real_url = page.geturl()
soup = BeautifulSoup(page.read())
except (http.HTTPError, http.URLError): except (http.HTTPError, http.URLError):
return "Could not fetch page." return "Could not fetch page."
try: title = soup.find('title').contents[0]
title = page.find(".//title").text
except AttributeError: if not title:
return "Could not find title." return "Could not find title."
return http.unescape(title) return u"{} [{}]".format(title, real_url)