X-Git-Url: http://git.veekun.com/zzz-dywypi.git/blobdiff_plain/c96407e6363db66ce91c2931a0dd84bf52f716d7..a806b46d032f76fa22b864987b7adfc76b4d1da0:/plugins/Pokedex/plugin.py diff --git a/plugins/Pokedex/plugin.py b/plugins/Pokedex/plugin.py index 84e9cf9..6268144 100644 --- a/plugins/Pokedex/plugin.py +++ b/plugins/Pokedex/plugin.py @@ -57,13 +57,21 @@ class Pokedex(callbacks.Plugin): Looks up in the veekun Pokédex.""" + # Fix encoding. Sigh. + if not isinstance(thing, unicode): + ascii_thing = thing + try: + thing = ascii_thing.decode('utf8') + except UnicodeDecodeError: + thing = ascii_thing.decode('latin1') + # Similar logic to the site, here. results = pokedex.lookup.lookup(thing, session=self.db, indices=self.indices) # Nothing found if len(results) == 0: - irc.reply("I don't know what that is.") + self._reply(irc, "I don't know what that is.") return # Multiple matches; propose them all @@ -81,56 +89,88 @@ class Pokedex(callbacks.Plugin): result_strings = [] for result in results: - result_string = result.name + result_string = result.object.name + + # Prepend, e.g., pokemon: if necessary if use_prefixes: # Table classes know their singular names prefix = result.object.__singlename__ result_string = prefix + ':' + result_string + + # Identify foreign language names + if result.language: + result_string += u""" ({0}: {1})""".format( + result.iso3166, result.name) + result_strings.append(result_string) - irc.reply("{0}: {1}?".format(reply, '; '.join(result_strings))) + self._reply(irc, u"{0}: {1}?".format(reply, '; '.join(result_strings))) return # If we got here, there's an exact match; hurrah! result = results[0] - if isinstance(result.object, tables.Pokemon): - irc.reply("""{name}, {type}-type Pokémon.""".format( - name=result.object.name, - type='/'.join(_.name for _ in result.object.types), + obj = result.object + if isinstance(obj, tables.Pokemon): + reply_template = \ + u"""#{id} {name}, {type}-type Pokémon. Has {abilities}. """ \ + """Is {stats}.""" + self._reply(irc, reply_template.format( + id=obj.id, + name=obj.name, + type='/'.join(_.name for _ in obj.types), + abilities=' or '.join(_.name for _ in obj.abilities), + stats='/'.join(str(_.base_stat) for _ in obj.stats), ) ) - elif isinstance(result.object, tables.Move): - irc.reply("""{name}, {type}-type move.""".format( - name=result.object.name, - type=result.object.type.name, + elif isinstance(obj, tables.Move): + reply_template = \ + u"""{name}, {type}-type {damage_class} move. """ \ + """{power} power; {accuracy}% accuracy; {pp} PP. """ \ + """{effect}""" + self._reply(irc, reply_template.format( + name=obj.name, + type=obj.type.name, + damage_class=obj.damage_class.name, + power=obj.power, + accuracy=obj.accuracy, + pp=obj.pp, + effect=unicode(obj.short_effect.as_html), ) ) - elif isinstance(result.object, tables.Type): - irc.reply("""{name}, a type.""".format( - name=result.object.name, + elif isinstance(obj, tables.Type): + self._reply(irc, u"""{name}, a type.""".format( + name=obj.name, ) ) - elif isinstance(result.object, tables.Item): - irc.reply("""{name}, an item.""".format( - name=result.object.name, + elif isinstance(obj, tables.Item): + self._reply(irc, u"""{name}, an item.""".format( + name=obj.name, ) ) - elif isinstance(result.object, tables.Ability): - irc.reply("""{name}, an ability.""".format( - name=result.object.name, + elif isinstance(obj, tables.Ability): + self._reply(irc, u"""{name}, an ability. {effect}""".format( + name=obj.name, + effect=obj.effect, ) ) else: # This can only happen if lookup.py is upgraded and we are not - irc.reply("Uhh.. I found that, but I don't know what it is. :(") + self._reply(irc, "Uhh.. I found that, but I don't know what it is. :(") pokedex = wrap(pokedex, [rest('something')]) + def _reply(self, irc, response): + """Wraps irc.reply() to do some Unicode decoding.""" + if isinstance(response, str): + irc.reply(response) + else: + irc.reply(response.encode('utf8')) + Class = Pokedex