Refactor attributes handling: conciseness and flexibility

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

1 files changed, 9 insertions, 8 deletions

diff --git a/lib_subgraphs.py b/lib_subgraphs.py
index 9d74e42..fd233b0 100644
--- a/lib_subgraphs.py
+++ b/lib_subgraphs.py
@@ -7,17 +7,18 @@
 """Subgraph library incl. visual aspects"""
 
 from lib import LibSubgraph
+from lib_attributes import *
 
 
 class SubgraphImportant(LibSubgraph):
-    def __init__(self, *args, **kwargs):
-        kwargs.setdefault('fillcolor', '#eaeaea')
-        kwargs.setdefault('style', 'bold, filled, rounded')
-        super(SubgraphImportant, self).__init__(*args, **kwargs)
+    defaults = dict((
+        FILLCOLOR('#eaeaea'),
+        STYLE('bold, filled, rounded'),
+    ))
 
 
 class SubgraphStandard(LibSubgraph):
-    def __init__(self, *args, **kwargs):
-        kwargs.setdefault('fillcolor', '#f5f5f5')
-        kwargs.setdefault('style', 'filled, rounded')
-        super(SubgraphStandard, self).__init__(*args, **kwargs)
+    defaults = dict((
+        FILLCOLOR('#f5f5f5'),
+        STYLE('filled, rounded'),
+    ))