Draft: Enable parsing of extension-less files

pydeps/dummymodule.py

@@ -16,10 +16,13 @@ def is_module(directory):
 
 
 def is_pysource(fname):
-    """A file name is a python source file iff it ends with '.py' and doesn't
-       start with a dot.
+    """A file name is a python source file iff (it's extensionless or
+       ends with '.py') and doesn't start with a dot.
     """
-    return not fname.startswith('.') and fname.endswith('.py')
+    if not fname.startswith('.'):
+        if len(fname.split('.')) == 1 or fname.endswith('.py'):
+            return True
+    return False
 
 
 def fname2modname(fname, package_root):

pydeps/target.py

@@ -11,6 +11,7 @@
 import tempfile
 from contextlib import contextmanager
 
+from . import dummymodule
 
 class Target(object):
     """The compilation target.
@@ -28,7 +29,7 @@ def __init__(self, path):
             sys.exit(1)
         self.is_dir = os.path.isdir(self.path)
         self.is_module = self.is_dir and '__init__.py' in os.listdir(self.path)
-        self.is_pysource = os.path.splitext(self.path)[1] in ('.py', '.pyc', '.pyo')
+        self.is_pysource = dummymodule.is_pysource(self.path)
         self.fname = os.path.basename(self.path)
         if self.is_dir:
             self.dirname = self.fname

tests/test_file.py

@@ -13,6 +13,13 @@ def test_file():
     with create_files(files) as workdir:
         assert simpledeps('a.py') == set()
 
+def test_file_no_py_extension():
+    files = """
+        a: |
+            import collections
+    """
+    with create_files(files) as workdir:
+        assert 'other -> a' in simpledeps('a', '--pylib')
 
 def test_file_pylib():
     files = """