Do some renaming & styling

Signed-off-by: Jan Pokorný <jpokorny@redhat.com>

1 files changed, 24 insertions, 7 deletions

diff --git a/lib.py b/lib.py
index 778d2e3..4126999 100644
--- a/lib.py
+++ b/lib.py
@@ -66,17 +66,25 @@ class LibEdge(Edge):
 #
 
 
-class GraphNode(LibSubgraph):
+class SubgraphImportant(LibSubgraph):
     def __init__(self, *args, **kwargs):
-        kwargs.setdefault('style', 'bold')
-        super(GraphNode, self).__init__(*args, **kwargs)
+        kwargs.setdefault('fillcolor', '#eaeaea')
+        kwargs.setdefault('style', 'bold, filled, rounded')
+        super(SubgraphImportant, self).__init__(*args, **kwargs)
 
 
-class GraphSubnode(LibSubgraph):
+class SubgraphStandard(LibSubgraph):
     def __init__(self, *args, **kwargs):
-        kwargs.setdefault('fillcolor', 'azure')
-        kwargs.setdefault('style', 'filled')
-        super(GraphSubnode, self).__init__(*args, **kwargs)
+        kwargs.setdefault('fillcolor', '#f5f5f5')
+        kwargs.setdefault('style', 'filled, rounded')
+        super(SubgraphStandard, self).__init__(*args, **kwargs)
+
+
+class SubgraphInvisible(LibSubgraph):
+    def __init__(self, *args, **kwargs):
+        kwargs.setdefault('style', 'invisible')
+        super(SubgraphInvisible, self).__init__(*args, **kwargs)
+
 
 
 #
@@ -87,6 +95,7 @@ class GraphSubnode(LibSubgraph):
 class Program(LibNode):
     def __init__(self, *args, **kwargs):
         kwargs.setdefault('shape', 'box')
+        kwargs.setdefault('style', 'filled')
         super(Program, self).__init__(*args, **kwargs)
 
 
@@ -97,6 +106,13 @@ class Agent(Program):
         super(Agent, self).__init__(*args, **kwargs)
 
 
+class Library(Program):
+    def __init__(self, *args, **kwargs):
+        kwargs.setdefault('fillcolor', 'lavenderblush')
+        kwargs.setdefault('style', 'filled')
+        super(Library, self).__init__(*args, **kwargs)
+
+
 class Executable(Program):
     pass
 
@@ -232,6 +248,7 @@ class DelegateCIM(Delegate, CIM):
 # main-helpers
 #
 
+
 def export(fnc):
     __all__.append(fnc.__name__)
     return fnc