diff options
Diffstat (limited to 'helpers/python')
-rw-r--r-- | helpers/python | 10 |
1 files changed, 9 insertions, 1 deletions
diff --git a/helpers/python b/helpers/python index 23c14bde..b6c4d5ed 100644 --- a/helpers/python +++ b/helpers/python @@ -1,6 +1,14 @@ # -*- python -*- import pkgutil +import sys -for mod in pkgutil.iter_modules(): +# walk_packages() is much slower than iter_modules(), use it only when +# completing something with a dot in it. +if len(sys.argv) > 1 and "." in sys.argv[1]: + walker = pkgutil.walk_packages +else: + walker = pkgutil.iter_modules + +for mod in walker(): print(mod[1]) |