Make iterSourceCode take many paths.

1 files changed, 13 insertions, 7 deletions

diff --git a/pyflakes/scripts/pyflakes.py b/pyflakes/scripts/pyflakes.py
index b17271f..4725696 100644
--- a/pyflakes/scripts/pyflakes.py
+++ b/pyflakes/scripts/pyflakes.py
@@ -137,16 +137,22 @@ def checkPath(filename, reporter=None):
 
 
 
-def iterSourceCode(directory):
+def iterSourceCode(paths):
     """
-    Iterate over all Python source files in C{directory}.
+    Iterate over all Python source files in C{paths}.
 
-    @param directory: A path to a directory.
+    @param paths: A list of paths.  Directories will be recursed into and
+        any .py files found will be yielded.  Any non-directories will be
+        yielded as-is.
     """
-    for dirpath, dirnames, filenames in os.walk(directory):
-        for filename in filenames:
-            if filename.endswith('.py'):
-                yield os.path.join(dirpath, filename)
+    for path in paths:
+        if os.path.isdir(path):
+            for dirpath, dirnames, filenames in os.walk(path):
+                for filename in filenames:
+                    if filename.endswith('.py'):
+                        yield os.path.join(dirpath, filename)
+        else:
+            yield path