fixed "from ..." import completion

Trundle · Trundle · commit 399692f79e8a · 2009-05-30T13:25:05.000+02:00
diff --git a/bpython/importcompletion.py b/bpython/importcompletion.py
@@ -34,18 +34,17 @@ def complete(line, cw):
     """Construct a full list of possibly completions for imports."""
     tokens = line.split()
     if tokens[0] in ['from', 'import']:
+        completing_from = False
         if tokens[0] == 'from':
-            completing_from = True
             if len(tokens) > 3:
                 if '.' in cw:
                     # This will result in a SyntaxError, so do not return
                     # any matches
                     return list()
+                completing_from = True
                 cw = '%s.%s' % (tokens[1], cw)
             elif len(tokens) == 3:
                 return ['import']
-        else:
-            completing_from = False
         
         matches = list()
         for name in modules:
