Added a prefix_lookup() function, for suggestions.

[zzz-pokedex.git] / pokedex / __init__.py

diff --git a/pokedex/__init__.py b/pokedex/__init__.py
index 73391dc..6a4ab95 100644 (file)
--- a/pokedex/__init__.py
+++ b/pokedex/__init__.py
@@ -15,7 +15,7 @@ def main():
 
     # XXX there must be a better way to get Unicode argv
     # XXX this doesn't work on Windows durp
 
     # XXX there must be a better way to get Unicode argv
     # XXX this doesn't work on Windows durp

-    enc = sys.stdin.encoding
+    enc = sys.stdin.encoding or 'utf8'

     args = [_.decode(enc) for _ in args]
 
     # Find the command as a function in this file
     args = [_.decode(enc) for _ in args]
 
     # Find the command as a function in this file


@@ -67,7 +67,12 @@ def command_lookup(name):
         print "Fuzzy-matched:"
 
     for result in results:
         print "Fuzzy-matched:"
 
     for result in results:

-        print "%s: %s" % (result.object.__tablename__, result.object.name),
+        if hasattr(result.object, 'full_name'):
+            name = result.object.full_name
+        else:
+            name = result.object.name
+
+        print "%s: %s" % (result.object.__tablename__, name),

         if result.language:
             print "(%s in %s)" % (result.name, result.language)
         else:
         if result.language:
             print "(%s in %s)" % (result.name, result.language)
         else: