From 77af941a05a3d3441c78e533aaf624ce9c15c418 Mon Sep 17 00:00:00 2001
From: Antony Lee <antony.lee@berkeley.edu>
Date: Sun, 26 Jul 2015 09:53:57 +0900
Subject: faster method of generating list of python modules

---
 Completion/Unix/Type/_python_modules | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

(limited to 'Completion')

diff --git a/Completion/Unix/Type/_python_modules b/Completion/Unix/Type/_python_modules
index b30848d83..e82f8efbf 100644
--- a/Completion/Unix/Type/_python_modules
+++ b/Completion/Unix/Type/_python_modules
@@ -26,10 +26,8 @@ _python_modules () {
   if ( [[ ${(P)+array_name} -eq 0 ]] || _cache_invalid $cache_id ) &&
       ! _retrieve_cache $cache_id; then
 
-    local script='import sys, pydoc
-def f(p,m,d):
-    if m.find(".") < 0: sys.stdout.write(m+"\n")
-pydoc.ModuleScanner().run(f)'
+   local script='import pkgutil
+for importer, name, ispkg in pkgutil.iter_modules(): print(name)'
 
     typeset -agU $array_name
     set -A $array_name \
-- 
cgit 1.4.1