Merge pull request #3 from cocodrips/0.2.0

0.2.0
cocodrips · Feb 6, 2019 · 8e61007 · 8e61007
2 parents 73886fb + 369e603
commit 8e61007
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 4 deletions.
diff --git a/doccov/__init__.py b/doccov/__init__.py
@@ -1,7 +1,7 @@
 """
 doccov
 """
-__version__ = '0.1.1'
+__version__ = '0.2.0'
 name = "doc-cov"
 
 from .main import walk, summary
diff --git a/doccov/main.py b/doccov/main.py
@@ -200,16 +200,22 @@ def walk(root_path):
     coverages = []
     summary = Coverage()
     for importer, modname, ispkg in packages:
-        spec = pkgutil._get_spec(importer, modname)
         try:
-            object = importlib._bootstrap._load(spec)
+            if ispkg:
+                spec = pkgutil._get_spec(importer, modname)
+                object = importlib._bootstrap._load(spec)
+            else:
+                import_path = f"{package_name}.{modname}"
+                object = importlib.import_module(import_path)
             counter = count_module(object)
 
             coverages.append(counter)
             summary += counter
         except ImportError as e:
             logger.error(f"Failed to import {modname}: {e}")
             continue
+        except Exception as e:
+            continue
 
     summary.name = 'coverage'
     return coverages, summary

diff --git a/tests/sample_project/module_fulldoc.py b/tests/sample_project/module_fulldoc.py
@@ -1,5 +1,5 @@
 """Full document module."""
-from sample_project import package_A
+from . import package_A
 import logging # unused 
 
 class FullDoc():