99 lines
2.9 KiB
Python
99 lines
2.9 KiB
Python
from util import hook, http, web
|
|
import json
|
|
import urllib2
|
|
|
|
|
|
def get_sound_info(game, search):
|
|
# Currently, http.get_json doesn't support adding headers.
|
|
def get_json(url, headers):
|
|
request = urllib2.Request(url)
|
|
request.add_header('User-Agent', http.ua_cloudbot)
|
|
opener = urllib2.build_opener()
|
|
return json.loads(opener.open(request))
|
|
|
|
search = search.replace(" ", "+")
|
|
try:
|
|
data = get_json("http://p2sounds.blha303.com.au/search/%s/%s" % (game, search))
|
|
except urllib2.HTTPError as e:
|
|
return "Error: " + json.loads(e.read())["error"]
|
|
items = []
|
|
for item in data["items"]:
|
|
if "music" in game:
|
|
textsplit = item["text"].split('"')
|
|
text = ""
|
|
for i in xrange(len(textsplit)):
|
|
if i % 2 != 0 and i < 6:
|
|
if text:
|
|
text += " / " + textsplit[i]
|
|
else:
|
|
text = textsplit[i]
|
|
else:
|
|
text = item["text"]
|
|
items.append("{} - {} {}".format(item["who"],
|
|
text if len(text) < 325 else text[:325] + "...",
|
|
item["listen"] ) )
|
|
if len(items) == 1:
|
|
return items[0]
|
|
else:
|
|
return "{} (and {} others: {})".format(items[0], len(items) - 1, web.haste("\n".join(items)) )
|
|
|
|
|
|
@hook.command
|
|
def portal2(inp):
|
|
"""portal2 <quote> - Look up Portal 2 quote.
|
|
Example: .portal2 demand to see life's manager"""
|
|
return get_sound_info("portal2", inp)
|
|
|
|
|
|
@hook.command
|
|
def portal2dlc(inp):
|
|
"""portal2dlc <quote> - Look up Portal 2 DLC quote.
|
|
Example: .portal2dlc1 these exhibits are interactive"""
|
|
return get_sound_info("portal2dlc1", inp)
|
|
|
|
|
|
@hook.command("portal2pti")
|
|
@hook.command
|
|
def portal2dlc2(inp):
|
|
"""portal2dlc2 <quote> - Look up Portal 2 Perpetual Testing Inititive quote.
|
|
Example: .portal2 Cave here."""
|
|
return get_sound_info("portal2dlc2", inp)
|
|
|
|
|
|
@hook.command
|
|
def portal2music(inp):
|
|
"""portal2music <title> - Look up Portal 2 music.
|
|
Example: .portal2music turret opera"""
|
|
return get_sound_info("portal2music", inp)
|
|
|
|
|
|
@hook.command('portal1')
|
|
@hook.command
|
|
def portal(inp):
|
|
"""portal <quote> - Look up Portal quote.
|
|
Example: .portal The last thing you want to do is hurt me"""
|
|
return get_sound_info("portal1", inp)
|
|
|
|
|
|
@hook.command('portal1music')
|
|
@hook.command
|
|
def portalmusic(inp):
|
|
"""portalmusic <title> - Look up Portal music.
|
|
Example: .portalmusic still alive"""
|
|
return get_sound_info("portal1music", inp)
|
|
|
|
|
|
@hook.command('tf2sound')
|
|
@hook.command
|
|
def tf2(inp):
|
|
"""tf2 [who - ]<quote> - Look up TF2 quote.
|
|
Example: .tf2 may i borrow your earpiece"""
|
|
return get_sound_info("tf2", inp)
|
|
|
|
|
|
@hook.command
|
|
def tf2music(inp):
|
|
"""tf2music title - Look up TF2 music lyrics.
|
|
Example: .tf2music rocket jump waltz"""
|
|
return get_sound_info("tf2music", inp)
|