Add parameters to method c4_graph; Refactor file structure

jieyinybl · Nov 23, 2020 · c52f1af · c52f1af
1 parent e9fa485
commit c52f1af
Show file tree

Hide file tree

Showing 5 changed files with 3 additions and 77 deletions.
diff --git a/c4/architectures/dwh.py → c4/architecture_examples/dwh.py b/c4/architectures/dwh.py → c4/architecture_examples/dwh.py
diff --git a/c4/graph.py b/c4/graph.py
@@ -1,15 +1,12 @@
-import os
 import graphviz
 import pathlib
 
-from c4.architectures.dwh import dwh
-
 
 def _content(element):
     return f'{element.name}\n\n{element.description}'
 
 
-def c4_graph(context):
+def c4_graph(context, format='png', filename='architecture_graph', directory=pathlib.Path('architectures/')):
     attr = {'node_attr': {'shape': 'record',
                           'fontsize': '6'},
             'edge_attr': {'fontsize': '6',
@@ -39,7 +36,4 @@ def c4_graph(context):
             for usage in system.usages:
                 graph.edge(_content(system), _content(usage.target), label=usage.description)
 
-    graph.render(format='png', filename='architecture_graph', directory=pathlib.Path('c4/images/'))
-
-
-c4_graph(dwh)
+    graph.render(format=format, filename=filename, directory=directory)
diff --git a/c4/images/architecture_graph b/c4/images/architecture_graph
diff --git a/c4/images/architecture_graph.png b/c4/images/architecture_graph.png
diff --git a/setup.py b/setup.py
@@ -2,7 +2,7 @@
 
 setup(
     name='c4-achitecture',
-    version='0.1dev',
+    version='0.1',
     install_requires=[
         'graphviz'
     ],