|
12 | 12 | import re |
13 | 13 | Base = declarative_base() |
14 | 14 |
|
15 | | -def generate_data_model_diagram(models, output_file='my_data_model_diagram', add_labels=True): |
| 15 | +def generate_data_model_diagram(models, output_file='my_data_model_diagram', add_labels=True, view_diagram=True): |
16 | 16 | # Initialize graph with more advanced visual settings |
17 | 17 | dot = graphviz.Digraph(comment='Interactive Data Models', format='svg', |
18 | 18 | graph_attr={'bgcolor': '#EEEEEE', 'rankdir': 'TB', 'splines': 'spline'}, |
@@ -59,7 +59,7 @@ def generate_data_model_diagram(models, output_file='my_data_model_diagram', add |
59 | 59 | dot.edge(name, target_name, label=rel.key if add_labels else None, tooltip=tooltip, color="#1E88E5", style="dashed") |
60 | 60 |
|
61 | 61 | # Render the graph to a file and open it |
62 | | - dot.render(output_file, view=True) |
| 62 | + dot.render(output_file, view=view_diagram) |
63 | 63 |
|
64 | 64 |
|
65 | 65 | def add_web_font_and_interactivity(input_svg_file, output_svg_file): |
|
0 commit comments