Separate group diagrams

tests.py

@@ -2890,6 +2890,89 @@ def _functionArgsMatch(callArgs: [], funcArgs: []):
     return callArgsCtr >= funcArgsCtr
 
 
+def _moduleInGroups(modName: str, includeGroups: [], modGroups: {}) -> bool:
+    """Is the given module within the included groups list?
+    """
+    for groupName in includeGroups:
+        if modName in modGroups[groupName]:
+            return True
+    return False
+
+
+def _diagramGroups(includeGroups: [],
+                   modules: {}, modGroups: {},
+                   maxModuleCalls: int) -> None:
+    """Draws a dot diagram containing only the given module groups
+    """
+    callGraphStr = 'digraph EpicyonGroups {\n\n'
+    callGraphStr += '  graph [fontsize=10 fontname="Verdana" compound=true];\n'
+    callGraphStr += '  node [fontsize=10 fontname="Verdana"];\n\n'
+    excludeModulesFromDiagram = (
+        'setup', 'tests', '__init__'
+    )
+    # colors of modules nodes
+    for modName, modProperties in modules.items():
+        if modName in excludeModulesFromDiagram:
+            continue
+        if not _moduleInGroups(modName, includeGroups, modGroups):
+            continue
+        if not modProperties.get('calls'):
+            callGraphStr += '  "' + modName + \
+                '" [fillcolor=yellow style=filled];\n'
+            continue
+        if len(modProperties['calls']) <= int(maxModuleCalls / 8):
+            callGraphStr += '  "' + modName + \
+                '" [fillcolor=green style=filled];\n'
+        elif len(modProperties['calls']) < int(maxModuleCalls / 4):
+            callGraphStr += '  "' + modName + \
+                '" [fillcolor=orange style=filled];\n'
+        else:
+            callGraphStr += '  "' + modName + \
+                '" [fillcolor=red style=filled];\n'
+    callGraphStr += '\n'
+    # connections between modules
+    for modName, modProperties in modules.items():
+        if modName in excludeModulesFromDiagram:
+            continue
+        if not _moduleInGroups(modName, includeGroups, modGroups):
+            continue
+        if not modProperties.get('calls'):
+            continue
+        for modCall in modProperties['calls']:
+            if modCall in excludeModulesFromDiagram:
+                continue
+            if not _moduleInGroups(modCall, includeGroups, modGroups):
+                continue
+            callGraphStr += '  "' + modName + '" -> "' + modCall + '";\n'
+    # module groups/clusters
+    clusterCtr = 1
+    for groupName, groupModules in modGroups.items():
+        if groupName not in includeGroups:
+            continue
+        callGraphStr += '\n'
+        callGraphStr += \
+            '  subgraph cluster_' + str(clusterCtr) + ' {\n'
+        callGraphStr += '    node [style=filled];\n'
+        for modName in groupModules:
+            if modName not in excludeModulesFromDiagram:
+                callGraphStr += '    ' + modName + ';\n'
+        callGraphStr += '    label = "' + groupName + '";\n'
+        callGraphStr += '    color = blue;\n'
+        callGraphStr += '  }\n'
+        clusterCtr += 1
+    callGraphStr += '\n}\n'
+    filename = 'epicyon_groups'
+    for groupName in includeGroups:
+        filename += '_' + groupName.replace(' ', '-')
+    filename += '.dot'
+    with open(filename, 'w+') as fp:
+        fp.write(callGraphStr)
+        print('Graph saved to ' + filename)
+        print('Plot using: ' +
+              'sfdp -x -Goverlap=false -Goverlap_scaling=2 ' +
+              '-Gsep=+100 -Tx11 epicyon_modules.dot')
+
+
 def _testFunctions():
     print('testFunctions')
     function = {}
@@ -3139,51 +3222,17 @@ def _testFunctions():
                         else:
                             modules[modName]['calls'] = [modCall]
             lineCtr += 1
-    callGraphStr = 'digraph EpicyonModules {\n\n'
+
-    callGraphStr += '  graph [fontsize=10 fontname="Verdana" compound=true];\n'
+    _diagramGroups(['Commandline Interface', 'ActivityPub', 'Core'],
-    callGraphStr += '  node [shape=record fontsize=10 fontname="Verdana"];\n\n'
+                   modules, modGroups, maxModuleCalls)
-    # colors of modules nodes
+    _diagramGroups(['Timeline', 'Core'],
-    for modName, modProperties in modules.items():
+                   modules, modGroups, maxModuleCalls)
-        if not modProperties.get('calls'):
+    _diagramGroups(['Web Interface', 'Core'],
-            callGraphStr += '  "' + modName + \
+                   modules, modGroups, maxModuleCalls)
-                '" [fillcolor=yellow style=filled];\n'
+    _diagramGroups(['Web Interface Columns', 'Core'],
-            continue
+                   modules, modGroups, maxModuleCalls)
-        if len(modProperties['calls']) <= int(maxModuleCalls / 8):
+    _diagramGroups(['Core'],
-            callGraphStr += '  "' + modName + \
+                   modules, modGroups, maxModuleCalls)
-                '" [fillcolor=green style=filled];\n'
-        elif len(modProperties['calls']) < int(maxModuleCalls / 4):
-            callGraphStr += '  "' + modName + \
-                '" [fillcolor=orange style=filled];\n'
-        else:
-            callGraphStr += '  "' + modName + \
-                '" [fillcolor=red style=filled];\n'
-    callGraphStr += '\n'
-    # connections between modules
-    for modName, modProperties in modules.items():
-        if not modProperties.get('calls'):
-            continue
-        for modCall in modProperties['calls']:
-            callGraphStr += '  "' + modName + '" -> "' + modCall + '";\n'
-    # module groups/clusters
-    clusterCtr = 1
-    for groupName, groupModules in modGroups.items():
-        callGraphStr += '\n'
-        callGraphStr += \
-            '  subgraph cluster_' + str(clusterCtr) + ' {\n'
-        callGraphStr += '    node [style=filled];\n'
-        for modName in groupModules:
-            callGraphStr += '    ' + modName + ';\n'
-        callGraphStr += '    label = "' + groupName + '";\n'
-        callGraphStr += '    color = blue;\n'
-        callGraphStr += '  }\n'
-        clusterCtr += 1
-    callGraphStr += '\n}\n'
-    with open('epicyon_modules.dot', 'w+') as fp:
-        fp.write(callGraphStr)
-        print('Modules call graph saved to epicyon_modules.dot')
-        print('Plot using: ' +
-              'sfdp -x -Goverlap=false -Goverlap_scaling=2 ' +
-              '-Gsep=+100 -Tx11 epicyon_modules.dot')
 
     callGraphStr = 'digraph Epicyon {\n\n'
     callGraphStr += '  size="8,6"; ratio=fill;\n'