def get_youtube(query): if yt is None: data = requests.get("https://gdata.youtube.com/feeds/api/videos?q=%s&alt=json" % urllib.quote_plus(query)).text data = json.loads(data) title = data["feed"]["entry"][0]["title"]["$t"] data = data["feed"]["entry"][0]["link"][0]["href"] data = re.findall("v=(.+)&", data)[0] # Quick heuristic: test if there are common words between the titles qfilter = {"".join(filter(str.isalpha, i)).lower() for i in query.split()} qtfilter = {"".join(filter(str.isalpha, i)).lower() for i in query.split()} tfilter = {"".join(filter(str.isalpha, i)).lower() for i in title.split()} if len({i for i in qfilter & tfilter if len(i) > 2}) >= 2 and qtfilter & tfilter: return url.format("http://youtu.be/" + data) else: return url.format("http://youtu.be/" + yt.get_music_video(query)[1]) # Don't check title for now.
def shorten_urls(data): """ Shorten and format all URLs. """ jobs = [] for line in data: jobs.append((lambda i: lambda: re.sub("http://[^ ]+", lambda x: url.format(url.shorten(x.group(0))), i))(line)) return util.parallelise(jobs)
def collage(self, server, message, flags, data, user): tapmusic = "http://nsfcd.com/lastfm/collage.php?" if not user: user = message.address.nick lowername = server.lower(user) if lowername in self.users: user = self.users[lowername] defaults = {"user": user, "type": "7day", "size": "3x3"} for i in (flags or []): if i in "cap": defaults[{ "c": "caption", "a": "artistonly", "p": "playcount" }[i]] = "true" if data: for i in data.split(): if "x" in i: defaults["size"] = i else: defaults["type"] = { "7d": "7day", "1m": "1month", "3m": "3month", "6m": "6month", "12m": "12month", "overall": "overall" }[i] return "04│ " + url.format( url.shorten(tapmusic + urlencode(defaults)))
def shortgo(server, message, url): """ Shorten a URL. If no URL is provided, shorten the last posted URL. """ if not url: url = grabber.last(server.lower(message.context)) return "12%s│ %s" % ("🔗" * message.text.startswith("@"), URL.format(URL.shorten(url)))
def mean_girls(server, msg): params = { "start": random.randrange(1, 75), "safe": "off", "v": "1.0", "rsz": 1, "q": "mean girls quotes" } pic = requests.get( "https://ajax.googleapis.com/ajax/services/search/images", params=params).json()["responseData"]["results"][0]["url"] return url.format(url.shorten(pic))
def mean_girls(server, msg): params = { "start": random.randrange(1,75), "safe": "off", "v": "1.0", "rsz": 1, "q": "mean girls quotes" } pic = requests.get( "https://ajax.googleapis.com/ajax/services/search/images", params=params ).json()["responseData"]["results"][0]["url"] return url.format(url.shorten(pic))
def push_text(push): fields = [] if push["type"] in ["note", "link", "file"]: message_field = [] if "file_url" in push: fields.append(url.format(url.shorten(push["file_url"]))) if "title" in push: message_field.append("\x0303%s\x03" % push["title"]) if "body" in push: message_field.append(push["body"].replace("\n", " · ")) # TODO: temporary if message_field: fields.append(" ".join(message_field)) if "url" in push: try: fields.append(url.format(url.shorten(push["url"]))) except: fields.append(url.format(push["url"])) elif push["type"] == "address": if "name" in push: fields.append("\x0303 📍 %s\x03" % push["name"]) if "address" in push: fields.append(push["address"]) return " · ".join(fields)
def urban_lookup(bot, msg, arg, index): ''' UrbanDictionary lookup. ''' url = 'http://www.urbandictionary.com/iphone/search/define' params = {'term': arg} nick = msg.address.nick try: index = int(index.strip()) - 1 except: index = 0 request = requests.get(url, params=params) data = request.json() defs = None output = "" try: defs = data['list'] if data['result_type'] == 'no_results': return failmsg() % (nick, params['term']) output = "%s 15│ %s" % (defs[index]['word'], re.sub( r"\[(.+?)\]", "\x02\\1\x02", re.sub(r"\[word\](.+?)\[/word\]", "\x02\\1\x02", defs[index]['definition']))) except: return failmsg() % params['term'] output = " ".join(output.split()) sentences = re.split("([!.?]+)", output) output = sentences.pop(0) i = 1 while sentences: s = sentences.pop(0) if i % 2 or len(output) + len(s) < 350: output += s else: sentences = [s] + sentences # put it back gently break i += 1 if sentences: output += '\n15│ Read more: %s' % format( shorten(defs[index]['permalink'])) return "15│ %s" % output
def urban_lookup(bot, msg, arg, index): ''' UrbanDictionary lookup. ''' url = 'http://www.urbandictionary.com/iphone/search/define' params = {'term': arg} nick = msg.address.nick try: index = int(index.strip()) - 1 except: index = 0 request = requests.get(url, params=params) data = request.json() defs = None output = "" try: defs = data['list'] if data['result_type'] == 'no_results': return failmsg() % (nick, params['term']) output = "%s 15│ %s" % (defs[index]['word'], re.sub(r"\[(.+?)\]", "\x02\\1\x02", defs[index]['definition'])) except: return failmsg() % params['term'] output = " ".join(output.split()) sentences = re.split("([!.?]+)", output) output = sentences.pop(0) i = 1 while sentences: s = sentences.pop(0) if i % 2 or len(output) + len(s) < 350: output += s else: sentences = [s] + sentences # put it back gently break i += 1 if sentences: output += '\n15│ Read more: %s' % format(shorten(defs[index]['permalink'])) return "15│ %s" % output
def collage(self, server, message, flags, data, user): tapmusic = "http://tapmusic.net/lastfm/collage.php?" if not user: user = message.address.nick lowername = server.lower(user) if lowername in self.users: user = self.users[lowername] defaults = {"user": user, "type": "7d", "size": "3x3"} for i in (flags or []): if i in "cap": defaults[{"c":"caption","a":"artistonly","p":"playcount"}[i]] = "true" if data: for i in data.split(): if "x" in i: defaults["size"] = i else: defaults["type"] = {"7d": "7day", "1m": "1month", "3m": "3month", "6m": "6month", "12m": "12month", "overall":"overall"}[i] return "04│ "+url.format(url.shorten(tapmusic + urlencode(defaults)))
def get_youtube(query): return url.format("http://youtu.be/" + yt.get_music_video(query)[1])
def wolfram_format(self, query, category=None, h_max=None, user=None, wasettings={}): try: if self.last is not None: query = query.replace("$_", self.last) answer, url = parallelise([ lambda: self.wolfram(query, **wasettings), lambda: URL.shorten("http://www.wolframalpha.com/input/?i=%s" % urllib.parse.quote_plus(query)) ]) if "Result" in answer and "(" not in answer["Result"]: self.last = answer["Result"] else: self.last = "(%s)" % query except urllib.error.URLError: return "05Wolfram08Alpha failed to respond. Try again later or go to " + URL.format( url) if not answer: return "05Wolfram08Alpha returned no results for '07%s'" % query for i in self.input_categories: if i in answer: header = answer[i] remove = i break else: header = "'%s'" % query remove = False header = str.join(" ", header.split()) h_max = h_max or self.h_max if not category: results = answer if remove: results = collections.OrderedDict([(k, v) for k, v in answer.items() if k != remove]) if category is not None: # Guess the category for i in self.results: if i in answer: results = {i: results[i]} break else: results = { list(results.keys())[0]: list(results.values())[0] } else: results = max(answer, key=lambda x: difflib.SequenceMatcher( None, category, x).ratio()) results = {results: answer[results]} output = [ spacepad("05Wolfram08Alpha 04 ", " %s" % header, self.t_max) ] t_max = striplen(output[0]) results = collections.OrderedDict([ (k, self.breakdown(v.split("\n"), t_max - 3)) for k, v in results.items() ]) total_lines = sum([min(len(results[x]), h_max) for x in results]) if total_lines > self.t_lines: # Too many lines, print the available categories instead. for i in justifiedtable( sorted(results.keys(), key=len) + ["05Categories"], t_max - 3): output.append(" 04⎪ %s" % i) output[-1] = " 04⎩" + output[-1][5:] elif results: if len(results) == 1 and len(list(results.values())[0]) == 1: # Single line: Shorten output catname = list(results.keys())[0] if catname in self.results: output = ["08│ %s " % list(results.values())[0][0]] else: output = [ spacepad("08│ %s " % list(results.values())[0][0], "07%s" % catname, self.t_max) ] else: for category in results: lines = [x.rstrip() for x in results[category]] output.append( spacepad(" 08⎨ %s " % lines.pop(0), " 07%s" % category, t_max)) truncated = lines[:h_max] for line in truncated: output.append(" 08⎪ " + line) if len(truncated) < len(lines): omission = "%d more lines" % (len(lines) - h_max + 1) length = t_max - len(omission) - 6 - len(url) output[-1] = " 08� " + url + " " + ( "-" * int(length)) + " 07%s" % omission else: output.append(" 08‣ 05No plaintext results. See " + URL.format( URL.shorten("http://www.wolframalpha.com/input/?i=%s" % urllib.parse.quote_plus(query)))) return "\n".join(i.rstrip() for i in output)
def shortgo(message, url): if not url: url = lg.links[server.lower(message.context)][-1] return "12%s│ %s" % ("bit.ly" * message.text.startswith("@"), URL.format(URL.shorten(url)))
def wolfram_format(self, query, category=None, h_max=None): try: answer = self.wolfram(query) except urllib.error.URLError: return "05Wolfram08Alpha failed to respond. Try again later or go to " + URL.format(URL.shorten("http://www.wolframalpha.com/input/?i=%s" % urllib.parse.quote_plus(query))) if not answer: return "05Wolfram08Alpha returned no results for '07%s'" % query for i in self.input_categories: if i in answer: header = answer[i] remove = i break else: header ="'%s'"%query remove = False header = str.join(" ", header.split()) h_max = h_max or self.h_max if not category: results = answer if remove: results = collections.OrderedDict([(k, v) for k, v in answer.items() if k != remove]) if category is not None: # Guess the category for i in self.results: if i in answer: results = {i: results[i]} break else: results = {list(results.keys())[0]: list(results.values())[0]} else: results = max(answer, key=lambda x:difflib.SequenceMatcher(None, category, x).ratio()) results = {results:answer[results]} output = [spacepad("05Wolfram08Alpha 04 ", " %s" % header, self.t_max)] t_max = striplen(output[0]) results = collections.OrderedDict([(k, self.breakdown(v.split("\n"), t_max - 3)) for k, v in results.items()]) total_lines = sum([min(len(results[x]), h_max) for x in results]) if total_lines > self.t_lines: # Too many lines, print the available categories instead. for i in justifiedtable(sorted(results.keys(), key=len) + ["05Categories"], t_max-3): output.append(" 04⎪ %s" % i) output[-1] = " 04⎩" + output[-1][5:] elif results: if len(results) == 1 and len(list(results.values())[0]) == 1: # Single line: Shorten output catname = list(results.keys())[0] if catname in self.results: output = ["08│ %s " % list(results.values())[0][0]] else: output = [spacepad("08│ %s " % list(results.values())[0][0], "07%s" % catname, t_max)] else: for category in results: lines = [x.rstrip() for x in results[category]] output.append(spacepad(" 08⎨ %s " % lines.pop(0), " 07%s" % category, t_max)) truncated = lines[:h_max] for line in truncated: output.append(" 08⎪ " + line) if len(truncated) < len(lines): omission = "%d more lines" % (len(lines) - h_max) length = t_max - len(omission) - 5 output[-1] = " 08⎬�" + ("-"*int(length)) + " 07%s" % omission else: output.append(" 08‣ 05No plaintext results. See " + URL.format(URL.shorten("http://www.wolframalpha.com/input/?i=%s" % urllib.parse.quote_plus(query)))) return "\n".join(i.rstrip() for i in output)