ZahirJS commited on
Commit
6fee7d6
·
verified ·
1 Parent(s): 97301c1

Update class_diagram_generator.py

Browse files
Files changed (1) hide show
  1. class_diagram_generator.py +4 -0
class_diagram_generator.py CHANGED
@@ -222,6 +222,8 @@ def generate_class_diagram(json_input: str, output_format: str) -> str:
222
  else:
223
  html_label += f'<TR><TD ALIGN="CENTER" BGCOLOR="{header_color}"><B>{class_name}</B></TD></TR>'
224
 
 
 
225
  if attributes:
226
  for attr in attributes:
227
  visibility = attr.get('visibility', '+')
@@ -239,6 +241,8 @@ def generate_class_diagram(json_input: str, output_format: str) -> str:
239
  else:
240
  html_label += f'<TR><TD ALIGN="LEFT" BGCOLOR="{body_color}"> </TD></TR>'
241
 
 
 
242
  if methods:
243
  for method in methods:
244
  visibility = method.get('visibility', '+')
 
222
  else:
223
  html_label += f'<TR><TD ALIGN="CENTER" BGCOLOR="{header_color}"><B>{class_name}</B></TD></TR>'
224
 
225
+ html_label += '<HR/>'
226
+
227
  if attributes:
228
  for attr in attributes:
229
  visibility = attr.get('visibility', '+')
 
241
  else:
242
  html_label += f'<TR><TD ALIGN="LEFT" BGCOLOR="{body_color}"> </TD></TR>'
243
 
244
+ html_label += '<HR/>'
245
+
246
  if methods:
247
  for method in methods:
248
  visibility = method.get('visibility', '+')