[gentoo-commits] proj/catalyst:master commit in: bin/

[Mike Frysinger]

commit:     ea895af0e5a9f22f71ba695dcb7fd6c8f6f0ce70
Author:     Mike Frysinger <vapier <AT> gentoo <DOT> org>
AuthorDate: Sun Oct 11 18:19:13 2015 +0000
Commit:     Mike Frysinger <vapier <AT> gentoo <DOT> org>
CommitDate: Sun Oct 11 18:19:13 2015 +0000
URL:        https://gitweb.gentoo.org/proj/catalyst.git/commit/?id=ea895af0

pylint: scan all modules by default

Add a quick shortcut to scan all the modules in the tree.

 bin/pylint | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/bin/pylint b/bin/pylint
index 1a50609..b001827 100755
--- a/bin/pylint
+++ b/bin/pylint
@@ -10,10 +10,27 @@ import os
 import sys
 
 
+def find_all_modules(source_root):
+	"""Locate all python modules in the tree for scanning"""
+	ret = []
+
+	for root, _dirs, files in os.walk(source_root, topdown=False):
+		# Add all of the .py modules in the tree.
+		ret += [os.path.join(root, x) for x in files if x.endswith('.py')]
+
+	# Add the main scripts that don't end in .py.
+	ret += [os.path.join(source_root, 'bin', x) for x in ('catalyst', 'pylint')]
+
+	return ret
+
+
 def main(argv):
 	"""The main entry point"""
 	source_root = os.path.dirname(os.path.dirname(os.path.realpath(__file__)))
 
+	if not argv:
+		argv = find_all_modules(source_root)
+
 	pympath = source_root
 	pythonpath = os.environ.get('PYTHONPATH')
 	if pythonpath is None: