Переглянути джерело

Fix !urban following API change.

main
Ben Kurtovic 6 роки тому
джерело
коміт
72dc0bfeff
1 змінених файлів з 24 додано та 17 видалено
  1. +24
    -17
      commands/urbandictionary.py

+ 24
- 17
commands/urbandictionary.py Переглянути файл

@@ -1,6 +1,6 @@
# -*- coding: utf-8 -*-
#
# Public domain, 2013 Legoktm
# Public domain, 2013 Legoktm; 2013, 2018 Ben Kurtovic
#

from json import loads
@@ -15,27 +15,34 @@ class UrbanDictionary(Command):
name = "urban"
commands = ["urban", "urbandictionary", "dct", "ud"]

@staticmethod
def _normalize(word):
return re.sub(r"\W", "", word.lower())

def process(self, data):
if not data.args:
self.reply(data, "What do you want to define?")
return

url = "http://api.urbandictionary.com/v0/define?term={0}"
q = quote(' '.join(data.args), safe="")
query = urlopen(url.format(q)).read()
arg = " ".join(data.args)
query = urlopen(url.format(quote(arg, safe=""))).read()
res = loads(query)
if res['result_type'] == 'exact':
definition = re.sub(r"\s+", " ", res['list'][0]['definition'])
example = re.sub(r"\s+", " ", res['list'][0]['example'])
url = res['list'][0]['permalink'].rsplit('/', 1)[0]
if definition and definition[-1] not in (".", "!", "?"):
definition += "."
msg = '{0} \x02Example\x0F: {1} {2}'.format(
definition.encode("utf8"), example.encode("utf8"), url)
self.reply(data, msg)
elif res['result_type'] == 'fulltext':
L = [i['word'] for i in res['list']]
msg = 'Here are some close matches: {0}.'
self.reply(data, msg.format(u", ".join(L).encode("utf8")))
else:
results = res.get("list")
if not results:
self.reply(data, 'Sorry, no results found.')
return

result = results[0]
definition = re.sub(r"\s+", " ", result["definition"])
example = re.sub(r"\s+", " ", result["example"])
url = result["permalink"].rsplit("/", 1)[0]
if definition and definition[-1] not in (".", "!", "?"):
definition += "."

msg = "{0} \x02Example\x0F: {1} {2}".format(
definition.encode("utf8"), example.encode("utf8"), url)
if self._normalize(result["word"]) != self._normalize(arg):
msg = "\x02{0}\x0F: {1}".format(result["word"].encode("utf8"), msg)

self.reply(data, msg)

Завантаження…
Відмінити
Зберегти