Revert "fix: Fixes Issue 89 with missing find_module (ml-tooling#91)"

This reverts commit fc1b6fe.

Author: ltan10

README.md

@@ -38,7 +38,7 @@ Lazydocs makes it easy to generate beautiful markdown documentation for your Pyt
 
 ### Installation
 
-> _Requirements: Python 3.9+._
+> _Requirements: Python 3.6+._
 
 ```bash
 pip install lazydocs

setup.py

@@ -15,7 +15,7 @@
 EMAIL = "team@mltooling.org"
 AUTHOR = "ML Tooling Team"
 LICENSE = "MIT"
-REQUIRES_PYTHON = ">=3.9"
+REQUIRES_PYTHON = ">=3.6"
 VERSION = None  # Only set version if you like to overwrite the version in _about.py
 
 PWD = os.path.abspath(os.path.dirname(__file__))

src/lazydocs/generation.py

@@ -624,14 +624,6 @@ def _lines_isvalid(lines: list, start_index: int, blockindent: int,
             prev_blank_line_count += 1
     return "".join(out)
 
-
-def get_module(loader, module_name: str) -> Optional[Any]:
-    spec = loader.find_spec(module_name)
-    if spec is None:
-        raise ImportError(f"Cannot find module {module_name}")
-    return spec.loader.load_module(spec.name)
-
-
 class MarkdownGenerator(object):
     """Markdown generator class."""
 
@@ -1258,7 +1250,8 @@ def generate_docs(
                         mod = importlib.util.module_from_spec(mod_spec)
                         mod_spec.loader.exec_module(mod)
                     except AttributeError:
-                        mod = get_module(loader, module_name)
+                        # For older python version compatibility
+                        mod = loader.find_module(module_name).load_module(module_name)  # type: ignore
                     module_md = generator.module2md(mod, is_mdx=is_mdx, include_toc=include_toc)
                     if not module_md:
                         # Module md is empty -> ignore module and all submodules
@@ -1341,7 +1334,8 @@ def generate_docs(
                                 mod = importlib.util.module_from_spec(mod_spec)
                                 mod_spec.loader.exec_module(mod)
                             except AttributeError:
-                                mod = get_module(loader, module_name)
+                                # For older python version compatibility
+                                mod = loader.find_module(module_name).load_module(module_name)  # type: ignore
                             module_md = generator.module2md(mod, is_mdx=is_mdx, include_toc=include_toc)
 
                             if not module_md:

tests/test_generation.py

@@ -1,7 +1,6 @@
 import hashlib
 
-from lazydocs import MarkdownGenerator, generate_docs
-from tempfile import TemporaryDirectory
+from lazydocs import MarkdownGenerator
 
 
 def test_import2md() -> None:
@@ -36,31 +35,3 @@ def test_func2md() -> None:
     # Remove whitespaces: fix changes between py version 3.6 3.7 in signature method
     md_hash = hashlib.md5(markdown.replace(" ", "").encode("utf-8")).hexdigest()
     assert md_hash == "797bad8c00ee6f189cb6f578eaec02c4"
-
-
-def test_integration_generate_docs(capsys) -> None:
-    test_class = """
-class TestClass:
-   \"\"\"just a test class\"\"\"
-    """
-    with TemporaryDirectory() as d:
-        test_module_name = "test_module"
-        with open(f"{d}/{test_module_name}.py", "w") as f:
-            f.write(test_class)
-
-        overview_file_name = "DOCS.md"
-        overview_file = f"{d}/output/{overview_file_name}"
-        generate_docs(
-            paths=[d],
-            output_path=f"{d}/output/",
-            overview_file=overview_file_name
-        )
-
-        captured = capsys.readouterr()
-
-        with open(overview_file) as f:
-            result = f.read()
-
-        assert test_module_name in result
-        assert f"{test_module_name}.TestClass" in result
-        assert "Failed to generate docs for module" not in captured.out