Add proper apt-get completer

IPython/Extensions/ipy_app_completers.py

@@ -11,11 +11,7 b' ip = IPython.ipapi.get()'
 
 from ipy_completers import *
 
-ip.set_hook('complete_command', apt_completers, re_key = '.*apt-get')
-ip.set_hook('complete_command', apt_completers, re_key = '.*yum')
-
+ip.set_hook('complete_command', apt_completer, re_key = '.*apt-get')
 ip.set_hook('complete_command', svn_completer, str_key = 'svn')
-
 ip.set_hook('complete_command', hg_completer, str_key = 'hg')
-
 ip.set_hook('complete_command', bzr_completer, str_key = 'bzr')

IPython/Extensions/ipy_completers.py

@@ -165,32 +165,6 b' def vcs_completer(commands, event):'
     return ip.IP.Completer.file_matches(event.symbol)
 
 
-
-def apt_completers(self, event):
-    """ This should return a list of strings with possible completions.
-
-    Note that all the included strings that don't start with event.symbol
-    are removed, in order to not confuse readline.
-
-    """
-    # print event # dbg
-
-    # commands are only suggested for the 'command' part of package manager
-    # invocation
-
-    cmd = (event.line + "<placeholder>").rsplit(None,1)[0]
-    # print cmd
-    if cmd.endswith('apt-get') or cmd.endswith('yum'):
-        return ['update', 'upgrade', 'install', 'remove']
-
-    # later on, add dpkg -l / whatever to get list of possible
-    # packages, add switches etc. for the rest of command line
-    # filling
-
-    raise IPython.ipapi.TryNext
-
-
-
 pkg_cache = None
 
 def module_completer(self,event):
@@ -350,3 +324,32 b' def cd_completer(self, event):'
         raise IPython.ipapi.TryNext
     return found
 
+def apt_get_packages(prefix):
+    out = os.popen('apt-cache pkgnames')
+    for p in out:
+        if p.startswith(prefix):
+            yield p.rstrip()
+    
+    
+apt_commands = """\
+update upgrade install remove purge source build-dep dist-upgrade
+dselect-upgrade clean autoclean check"""
+
+def apt_completer(self, event):
+    """ Completer for apt-get (uses apt-cache internally)
+
+    """
+
+
+    cmd_param = event.line.split()
+    if event.line.endswith(' '):
+        cmd_param.append('')
+
+    if cmd_param[0] == 'sudo':
+        cmd_param = cmd_param[1:]
+
+    if len(cmd_param) == 2 or 'help' in cmd_param:
+        return apt_commands.split()
+
+    return list(apt_get_packages(event.symbol))
+

IPython/completer.py

@@ -620,8 +620,8 b' class IPCompleter(Completer):'
             except IndexError:
                 return None
         except:
-            #from IPython.ultraTB import AutoFormattedTB; # dbg
-            #tb=AutoFormattedTB('Verbose');tb() #dbg
+            from IPython.ultraTB import AutoFormattedTB; # dbg
+            tb=AutoFormattedTB('Verbose');tb() #dbg
 
             # If completion fails, don't annoy the user.
             return None

doc/ChangeLog

@@ -1,7 +1,10 b''
 2007-11-08  Ville Vainio  <[e-mail]>
-	* ipy_completer.py (import completer): assume 'xml' module exists. 
+	* ipy_completers.py (import completer): assume 'xml' module exists. 
 	Do not add every module twice anymore. Closes #196.
         
+	* ipy_completers.py, ipy_app_completers.py: Add proper apt-get
+	completer that uses apt-cache to search for existing packages.
+
 2007-11-06  Ville Vainio  <[e-mail]>
 
 	* Prompts.py: Do not update _oh and _123 when do_full_cache is not