Create AutoDiscover class in infra/module.py 🎉

Author: Douglas Farinelli

flask_toolkit/infra/module.py

@@ -0,0 +1,62 @@
+import importlib
+
+
+class AutoDiscover:
+    """
+    This class search modules in this package recursively. What it does
+    is actually search all the modules and run an `import_module` for each.
+
+    :param path: `pathlib.Path` object
+    :param pattern: Must be a string for search likes string.endswith(pattern)
+    :return: modules list
+
+    How to use:
+
+        autodiscover = Autodiscover(path=...)
+        autodiscover()
+    """
+
+    def __init__(self, path, pattern=None):
+        self.path = path
+        self.pattern = pattern
+        self.root = '.'.join(
+            reversed(self.__get_parts_until_root_recursively(path=path))
+        )
+
+    def __call__(self):
+        return self.__autodiscover(path=self.path, pattern=self.pattern)
+
+    def __get_parts_until_root_recursively(self, path):
+        parts = []
+        try:
+            next(path.glob('__init__.py'))
+        except StopIteration:
+            return parts
+        parts.append(path.name)
+        parts.extend(self.__get_parts_until_root_recursively(path.parent))
+        return parts
+
+    def __autodiscover(self, path, pattern):
+        modules = []
+
+        for obj in path.iterdir():
+            if obj.name.startswith('_'):
+                continue
+
+            if (
+                    obj.is_file()
+                and obj.suffix == '.py'
+                and obj.match(pattern or '*')
+            ):
+                module_name = self.__normalize_module_name(
+                    module_name='.'.join(obj.parts).replace('.py', '')
+                )
+                modules.append(importlib.import_module(module_name))
+
+            if obj.is_dir():
+                modules.extend(self.__autodiscover(path=obj, pattern=pattern))
+
+        return modules
+
+    def __normalize_module_name(self, module_name):
+        return module_name[module_name.find(self.root):]