- corrected version of vendored search_packages_info() from pip

- handle relative imports

Author: r1chardj0n3s

CHANGELOG.rst

@@ -1,6 +1,10 @@
 Release History
 ---------------
 
+1.1.2
+- corrected version of vendored search_packages_info() from pip
+- handle relative imports
+
 1.1.1
 - fixed handling of import from __future__
 - self-tested and added own requirements.txt

pip_missing_reqs/find_missing_reqs.py

@@ -41,12 +41,13 @@ def search_packages_info(query):  # pragma: no cover
                 lines = dist.get_metadata_lines('RECORD')
                 paths = [l.split(',')[0] for l in lines]
                 paths = [os.path.join(dist.location, p) for p in paths]
-                file_list = [os.path.relpath(p, dist.egg_info)
-                             for p in paths]
+                file_list = [os.path.relpath(p, dist.location) for p in paths]
         else:
             # Otherwise use pip's log for .egg-info's
             if dist.has_metadata('installed-files.txt'):
-                file_list = dist.get_metadata_lines('installed-files.txt')
+                paths = dist.get_metadata_lines('installed-files.txt')
+                paths = [os.path.join(dist.egg_info, p) for p in paths]
+                file_list = [os.path.relpath(p, dist.location) for p in paths]
         # use and short-circuit to check for None
         package['files'] = file_list and sorted(file_list)
         yield package
@@ -81,6 +82,9 @@ def visit_ImportFrom(self, node):
             # not an actual module
             return
         for alias in node.names:
+            if node.module is None:
+                # relative import
+                continue
             self.__addModule(node.module + '.' + alias.name, node.lineno)
 
     def __addModule(self, modname, lineno):

pip_missing_reqs/test_find_missing_reqs.py

@@ -104,7 +104,8 @@ def test_find_imported_modules(monkeypatch, caplog, ignore_ham, ignore_hashlib,
     class FakeFile():
         contents = [
             'from os import path\nimport ast, hashlib',
-            'from __future__ import braces\nimport ast, sys',
+            'from __future__ import braces\nimport ast, sys\n'
+            'from . import friend',
         ]
 
         def __init__(self, filename):

setup.py

@@ -12,7 +12,7 @@
 
 setup(
     name='pip_missing_reqs',
-    version='1.1.1',
+    version='1.1.2',
     description='Find packages that should be in requirements for a project',
     long_description=long_description,
     url='https://github.com/r1chardj0n3s/pip-missing-reqs',