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

standarised format of drama.py to match other plugins

Browse source
This commit is contained in:
Luke Rogers 2013-09-04 20:27:46 +12:00
parent 8769b97dc0
commit 13d284a861
1 changed files with 7 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

12
plugins/drama.py
View file

@ -1,4 +1,5 @@
from util import hook, http from util import hook, http, text
import re
api_url = "http://encyclopediadramatica.se/api.php?action=opensearch" api_url = "http://encyclopediadramatica.se/api.php?action=opensearch"
ed_url = "http://encyclopediadramatica.se/" ed_url = "http://encyclopediadramatica.se/"
@ -10,6 +11,7 @@ def drama(inp):
" the Encyclopedia Dramatica article on <phrase>." " the Encyclopedia Dramatica article on <phrase>."
j = http.get_json(api_url, search=inp) j = http.get_json(api_url, search=inp)
if not j[1]: if not j[1]:
return "No results found." return "No results found."
article_name = j[1][0].replace(' ', '_').encode('utf8') article_name = j[1][0].replace(' ', '_').encode('utf8')
@ -17,11 +19,11 @@ def drama(inp):
url = ed_url + http.quote(article_name, '') url = ed_url + http.quote(article_name, '')
page = http.get_html(url) page = http.get_html(url)
for p in page.xpath('//div[@id="bodyContent"]/p'): for p in page.xpath('//div[@class="mw-content-ltr"]/p'):
if p.text_content(): if p.text_content():
summary = " ".join(p.text_content().splitlines()) summary = " ".join(p.text_content().splitlines())
if len(summary) > 300: summary = re.sub("\[\d+\]", "", summary)
summary = summary[:summary.rfind(' ', 0, 300)] + "..." summary = text.truncate_str(summary, 200)
return "%s :: \x02%s\x02" % (summary, url) return "%s :: %s" % (summary, url)
return "Unknown Error." return "Unknown Error."