Add function that returns a list of all functions in their respective category

vhag · vhag · commit 14952a961441 · 2017-08-08T11:45:03.000+02:00
diff --git a/pygorithm/data_structures/__init__.py b/pygorithm/data_structures/__init__.py
@@ -0,0 +1 @@
+from . modules import modules
diff --git a/pygorithm/data_structures/modules.py b/pygorithm/data_structures/modules.py
@@ -0,0 +1,14 @@
+import pkgutil
+def modules():
+    """
+    Find all functions in pygorithm.data_structures
+    """
+    import pygorithm.data_structures
+    package = pygorithm.data_structures
+    modules = []
+    for importer, modname, ispkg in pkgutil.iter_modules(package.__path__):
+        modules.append(modname)
+    modules.remove('modules')
+    modules.sort()
+    return modules
+
diff --git a/pygorithm/searching/__init__.py b/pygorithm/searching/__init__.py
@@ -0,0 +1 @@
+from . modules import modules
diff --git a/pygorithm/searching/modules.py b/pygorithm/searching/modules.py
@@ -0,0 +1,14 @@
+import pkgutil
+def modules():
+    """
+    Find all functions in pygorithm.searching
+    """
+    import pygorithm.searching
+    package = pygorithm.searching
+    modules = []
+    for importer, modname, ispkg in pkgutil.iter_modules(package.__path__):
+        modules.append(modname)
+    modules.remove('modules')
+    modules.sort()
+    return modules
+
diff --git a/pygorithm/sorting/modules.py b/pygorithm/sorting/modules.py
@@ -0,0 +1,14 @@
+import pkgutil
+def modules():
+    """
+    Find all functions in pygorithm.sorting
+    """
+    import pygorithm.sorting
+    package = pygorithm.sorting
+    modules = []
+    for importer, modname, ispkg in pkgutil.iter_modules(package.__path__):
+        modules.append(modname)
+    modules.remove('modules')
+    modules.sort()
+    return modules
+
