# import graphviz # import json # from tempfile import NamedTemporaryFile # import os # def generate_class_diagram(json_input: str, output_format: str) -> str: # """ # Generates a class diagram from JSON input. # Args: # json_input (str): A JSON string describing the class diagram structure. # It must follow the Expected JSON Format Example below. # Expected JSON Format Example: # { # "classes": [ # { # "name": "Vehicle", # "type": "abstract", # "attributes": [ # {"name": "id", "type": "String", "visibility": "-"}, # {"name": "brand", "type": "String", "visibility": "#"}, # {"name": "model", "type": "String", "visibility": "#"}, # {"name": "year", "type": "int", "visibility": "#"}, # {"name": "price", "type": "double", "visibility": "+"}, # {"name": "vehicleCount", "type": "int", "visibility": "+", "static": true} # ], # "methods": [ # {"name": "Vehicle", "parameters": [{"name": "brand", "type": "String"}, {"name": "model", "type": "String"}], "return_type": "Vehicle", "visibility": "+"}, # {"name": "startEngine", "return_type": "void", "visibility": "+", "abstract": true}, # {"name": "stopEngine", "return_type": "void", "visibility": "+"}, # {"name": "getPrice", "return_type": "double", "visibility": "+"}, # {"name": "setPrice", "parameters": [{"name": "price", "type": "double"}], "return_type": "void", "visibility": "+"}, # {"name": "getTotalVehicles", "return_type": "int", "visibility": "+", "static": true} # ] # }, # { # "name": "Car", # "type": "class", # "attributes": [ # {"name": "doors", "type": "int", "visibility": "-"}, # {"name": "transmission", "type": "TransmissionType", "visibility": "-"}, # {"name": "fuelType", "type": "FuelType", "visibility": "-"} # ], # "methods": [ # {"name": "Car", "parameters": [{"name": "brand", "type": "String"}, {"name": "model", "type": "String"}, {"name": "doors", "type": "int"}], "return_type": "Car", "visibility": "+"}, # {"name": "startEngine", "return_type": "void", "visibility": "+"}, # {"name": "openTrunk", "return_type": "void", "visibility": "+"}, # {"name": "getDoors", "return_type": "int", "visibility": "+"}, # {"name": "setTransmission", "parameters": [{"name": "transmission", "type": "TransmissionType"}], "return_type": "void", "visibility": "+"} # ] # }, # { # "name": "Motorcycle", # "type": "class", # "attributes": [ # {"name": "engineSize", "type": "int", "visibility": "-"}, # {"name": "hasWindshield", "type": "boolean", "visibility": "-"} # ], # "methods": [ # {"name": "Motorcycle", "parameters": [{"name": "brand", "type": "String"}, {"name": "model", "type": "String"}], "return_type": "Motorcycle", "visibility": "+"}, # {"name": "startEngine", "return_type": "void", "visibility": "+"}, # {"name": "wheelie", "return_type": "void", "visibility": "+"}, # {"name": "getEngineSize", "return_type": "int", "visibility": "+"} # ] # }, # { # "name": "Engine", # "type": "class", # "attributes": [ # {"name": "horsepower", "type": "int", "visibility": "-"}, # {"name": "cylinders", "type": "int", "visibility": "-"}, # {"name": "fuelType", "type": "FuelType", "visibility": "-"} # ], # "methods": [ # {"name": "Engine", "parameters": [{"name": "horsepower", "type": "int"}, {"name": "cylinders", "type": "int"}], "return_type": "Engine", "visibility": "+"}, # {"name": "start", "return_type": "boolean", "visibility": "+"}, # {"name": "stop", "return_type": "void", "visibility": "+"}, # {"name": "getHorsepower", "return_type": "int", "visibility": "+"} # ] # }, # { # "name": "TransmissionType", # "type": "enum", # "attributes": [ # {"name": "MANUAL", "type": "TransmissionType", "visibility": "+", "static": true}, # {"name": "AUTOMATIC", "type": "TransmissionType", "visibility": "+", "static": true}, # {"name": "CVT", "type": "TransmissionType", "visibility": "+", "static": true} # ], # "methods": [] # }, # { # "name": "FuelType", # "type": "enum", # "attributes": [ # {"name": "GASOLINE", "type": "FuelType", "visibility": "+", "static": true}, # {"name": "DIESEL", "type": "FuelType", "visibility": "+", "static": true}, # {"name": "ELECTRIC", "type": "FuelType", "visibility": "+", "static": true}, # {"name": "HYBRID", "type": "FuelType", "visibility": "+", "static": true} # ], # "methods": [] # }, # { # "name": "VehicleService", # "type": "interface", # "attributes": [], # "methods": [ # {"name": "maintenance", "parameters": [{"name": "vehicle", "type": "Vehicle"}], "return_type": "void", "visibility": "+", "abstract": true}, # {"name": "repair", "parameters": [{"name": "vehicle", "type": "Vehicle"}, {"name": "issue", "type": "String"}], "return_type": "boolean", "visibility": "+", "abstract": true}, # {"name": "inspectVehicle", "parameters": [{"name": "vehicle", "type": "Vehicle"}], "return_type": "InspectionReport", "visibility": "+", "abstract": true} # ] # }, # { # "name": "GarageService", # "type": "class", # "attributes": [ # {"name": "garageName", "type": "String", "visibility": "-"}, # {"name": "location", "type": "String", "visibility": "-"} # ], # "methods": [ # {"name": "GarageService", "parameters": [{"name": "name", "type": "String"}], "return_type": "GarageService", "visibility": "+"}, # {"name": "maintenance", "parameters": [{"name": "vehicle", "type": "Vehicle"}], "return_type": "void", "visibility": "+"}, # {"name": "repair", "parameters": [{"name": "vehicle", "type": "Vehicle"}, {"name": "issue", "type": "String"}], "return_type": "boolean", "visibility": "+"}, # {"name": "inspectVehicle", "parameters": [{"name": "vehicle", "type": "Vehicle"}], "return_type": "InspectionReport", "visibility": "+"} # ] # } # ], # "relationships": [ # { # "from": "Car", # "to": "Vehicle", # "type": "inheritance", # "label": "extends" # }, # { # "from": "Motorcycle", # "to": "Vehicle", # "type": "inheritance", # "label": "extends" # }, # { # "from": "Car", # "to": "Engine", # "type": "composition", # "label": "has", # "multiplicity_from": "1", # "multiplicity_to": "1" # }, # { # "from": "Motorcycle", # "to": "Engine", # "type": "composition", # "label": "has", # "multiplicity_from": "1", # "multiplicity_to": "1" # }, # { # "from": "Car", # "to": "TransmissionType", # "type": "association", # "label": "uses", # "multiplicity_from": "1", # "multiplicity_to": "1" # }, # { # "from": "Vehicle", # "to": "FuelType", # "type": "association", # "label": "uses", # "multiplicity_from": "1", # "multiplicity_to": "1" # }, # { # "from": "GarageService", # "to": "VehicleService", # "type": "realization", # "label": "implements" # }, # { # "from": "GarageService", # "to": "Vehicle", # "type": "dependency", # "label": "services", # "multiplicity_from": "1", # "multiplicity_to": "*" # } # ] # } # Returns: # str: The filepath to the generated PNG image file. # """ # try: # if not json_input.strip(): # return "Error: Empty input" # data = json.loads(json_input) # if 'classes' not in data: # raise ValueError("Missing required field: classes") # # Configuración del diagrama con espaciado reducido # dot = graphviz.Digraph(comment='Class Diagram') # dot.attr(rankdir='TB', bgcolor='white', pad='0.5', nodesep='0.8', ranksep='1.2') # dot.attr('node', shape='plaintext', fontname='Arial', fontsize='11') # dot.attr('edge', color='black', fontname='Arial', fontsize='9', minlen='1') # classes = data.get('classes', []) # relationships = data.get('relationships', []) # for cls in classes: # class_name = cls.get('name') # class_type = cls.get('type', 'class') # attributes = cls.get('attributes', []) # methods = cls.get('methods', []) # if not class_name: # continue # # Crear tabla HTML para estructura de clase # html_label = '' # # Header con nombre de clase y estereotipo # if class_type == 'abstract': # html_label += f'' # elif class_type == 'interface': # html_label += f'' # elif class_type == 'enum': # html_label += f'' # else: # html_label += f'' # # Línea separadora después del nombre # html_label += '
' # # Sección de atributos # if attributes: # for attr in attributes: # visibility = attr.get('visibility', '+') # name = attr.get('name', '') # attr_type = attr.get('type', '') # is_static = attr.get('static', False) # line = f"{visibility} {name}" # if attr_type: # line += f" : {attr_type}" # if is_static: # line = f"{line}" # Subrayado para elementos estáticos # html_label += f'' # else: # # Espacio vacío si no hay atributos # html_label += '' # # Línea separadora entre atributos y métodos # html_label += '
' # # Sección de métodos # if methods: # for method in methods: # visibility = method.get('visibility', '+') # name = method.get('name', '') # parameters = method.get('parameters', []) # return_type = method.get('return_type', 'void') # is_static = method.get('static', False) # is_abstract = method.get('abstract', False) # line = f"{visibility} {name}(" # if parameters: # params = [] # for param in parameters: # param_name = param.get('name', '') # param_type = param.get('type', '') # params.append(f"{param_name}: {param_type}") # line += ", ".join(params) # line += f") : {return_type}" # if is_static: # line = f"{line}" # Subrayado para métodos estáticos # if is_abstract: # line = f"{line}" # Cursiva para métodos abstractos # html_label += f'' # else: # # Espacio vacío si no hay métodos # html_label += '' # html_label += '
<<abstract>>
{class_name}
<<interface>>
{class_name}
<<enumeration>>
{class_name}
{class_name}
{line}
{line}
' # # Agregar nodo con la tabla HTML # dot.node(class_name, f'<{html_label}>') # # Procesar relaciones con líneas rectas # for relationship in relationships: # from_class = relationship.get('from') # to_class = relationship.get('to') # rel_type = relationship.get('type', 'association') # label = relationship.get('label', '') # multiplicity_from = relationship.get('multiplicity_from', '') # multiplicity_to = relationship.get('multiplicity_to', '') # if not from_class or not to_class: # continue # edge_attrs = { # 'splines': 'ortho', # Líneas estrictamente ortogonales (solo horizontales/verticales) # 'concentrate': 'false', # Evita que las líneas se combinen # 'constraint': 'true' # Mantiene la estructura jerárquica # } # if label: # edge_attrs['label'] = label # if multiplicity_from: # edge_attrs['taillabel'] = multiplicity_from # if multiplicity_to: # edge_attrs['headlabel'] = multiplicity_to # # Configurar estilos de flecha según el tipo de relación # if rel_type == 'inheritance': # edge_attrs['arrowhead'] = 'empty' # edge_attrs['color'] = 'black' # elif rel_type == 'composition': # edge_attrs['arrowhead'] = 'normal' # edge_attrs['arrowtail'] = 'diamond' # edge_attrs['dir'] = 'both' # edge_attrs['color'] = 'black' # elif rel_type == 'aggregation': # edge_attrs['arrowhead'] = 'normal' # edge_attrs['arrowtail'] = 'odiamond' # edge_attrs['dir'] = 'both' # edge_attrs['color'] = 'black' # elif rel_type == 'realization': # edge_attrs['arrowhead'] = 'empty' # edge_attrs['style'] = 'dashed' # edge_attrs['color'] = 'black' # elif rel_type == 'dependency': # edge_attrs['arrowhead'] = 'normal' # edge_attrs['style'] = 'dashed' # edge_attrs['color'] = 'black' # else: # association # edge_attrs['arrowhead'] = 'normal' # edge_attrs['color'] = 'black' # dot.edge(from_class, to_class, **edge_attrs) # # Renderizar el diagrama # with NamedTemporaryFile(delete=False, suffix=f'.{output_format}') as tmp: # dot.render(tmp.name, format=output_format, cleanup=True) # return f"{tmp.name}.{output_format}" # except json.JSONDecodeError: # return "Error: Invalid JSON format" # except Exception as e: # return f"Error: {str(e)}" import graphviz import json from tempfile import NamedTemporaryFile import os def generate_class_diagram(json_input: str, output_format: str) -> str: """ Generates a class diagram from JSON input. Args: json_input (str): A JSON string describing the class diagram structure. It must follow the Expected JSON Format Example below. Expected JSON Format Example: { "classes": [ { "name": "Vehicle", "type": "abstract", "attributes": [ {"name": "id", "type": "String", "visibility": "-"}, {"name": "brand", "type": "String", "visibility": "#"}, {"name": "model", "type": "String", "visibility": "#"}, {"name": "year", "type": "int", "visibility": "#"}, {"name": "price", "type": "double", "visibility": "+"}, {"name": "vehicleCount", "type": "int", "visibility": "+", "static": true} ], "methods": [ {"name": "Vehicle", "parameters": [{"name": "brand", "type": "String"}, {"name": "model", "type": "String"}], "return_type": "Vehicle", "visibility": "+"}, {"name": "startEngine", "return_type": "void", "visibility": "+", "abstract": true}, {"name": "stopEngine", "return_type": "void", "visibility": "+"}, {"name": "getPrice", "return_type": "double", "visibility": "+"}, {"name": "setPrice", "parameters": [{"name": "price", "type": "double"}], "return_type": "void", "visibility": "+"}, {"name": "getTotalVehicles", "return_type": "int", "visibility": "+", "static": true} ] }, { "name": "Car", "type": "class", "attributes": [ {"name": "doors", "type": "int", "visibility": "-"}, {"name": "transmission", "type": "TransmissionType", "visibility": "-"}, {"name": "fuelType", "type": "FuelType", "visibility": "-"} ], "methods": [ {"name": "Car", "parameters": [{"name": "brand", "type": "String"}, {"name": "model", "type": "String"}, {"name": "doors", "type": "int"}], "return_type": "Car", "visibility": "+"}, {"name": "startEngine", "return_type": "void", "visibility": "+"}, {"name": "openTrunk", "return_type": "void", "visibility": "+"}, {"name": "getDoors", "return_type": "int", "visibility": "+"}, {"name": "setTransmission", "parameters": [{"name": "transmission", "type": "TransmissionType"}], "return_type": "void", "visibility": "+"} ] }, { "name": "Motorcycle", "type": "class", "attributes": [ {"name": "engineSize", "type": "int", "visibility": "-"}, {"name": "hasWindshield", "type": "boolean", "visibility": "-"} ], "methods": [ {"name": "Motorcycle", "parameters": [{"name": "brand", "type": "String"}, {"name": "model", "type": "String"}], "return_type": "Motorcycle", "visibility": "+"}, {"name": "startEngine", "return_type": "void", "visibility": "+"}, {"name": "wheelie", "return_type": "void", "visibility": "+"}, {"name": "getEngineSize", "return_type": "int", "visibility": "+"} ] }, { "name": "Engine", "type": "class", "attributes": [ {"name": "horsepower", "type": "int", "visibility": "-"}, {"name": "cylinders", "type": "int", "visibility": "-"}, {"name": "fuelType", "type": "FuelType", "visibility": "-"} ], "methods": [ {"name": "Engine", "parameters": [{"name": "horsepower", "type": "int"}, {"name": "cylinders", "type": "int"}], "return_type": "Engine", "visibility": "+"}, {"name": "start", "return_type": "boolean", "visibility": "+"}, {"name": "stop", "return_type": "void", "visibility": "+"}, {"name": "getHorsepower", "return_type": "int", "visibility": "+"} ] }, { "name": "TransmissionType", "type": "enum", "attributes": [ {"name": "MANUAL", "type": "TransmissionType", "visibility": "+", "static": true}, {"name": "AUTOMATIC", "type": "TransmissionType", "visibility": "+", "static": true}, {"name": "CVT", "type": "TransmissionType", "visibility": "+", "static": true} ], "methods": [] }, { "name": "FuelType", "type": "enum", "attributes": [ {"name": "GASOLINE", "type": "FuelType", "visibility": "+", "static": true}, {"name": "DIESEL", "type": "FuelType", "visibility": "+", "static": true}, {"name": "ELECTRIC", "type": "FuelType", "visibility": "+", "static": true}, {"name": "HYBRID", "type": "FuelType", "visibility": "+", "static": true} ], "methods": [] }, { "name": "VehicleService", "type": "interface", "attributes": [], "methods": [ {"name": "maintenance", "parameters": [{"name": "vehicle", "type": "Vehicle"}], "return_type": "void", "visibility": "+", "abstract": true}, {"name": "repair", "parameters": [{"name": "vehicle", "type": "Vehicle"}, {"name": "issue", "type": "String"}], "return_type": "boolean", "visibility": "+", "abstract": true}, {"name": "inspectVehicle", "parameters": [{"name": "vehicle", "type": "Vehicle"}], "return_type": "InspectionReport", "visibility": "+", "abstract": true} ] }, { "name": "GarageService", "type": "class", "attributes": [ {"name": "garageName", "type": "String", "visibility": "-"}, {"name": "location", "type": "String", "visibility": "-"} ], "methods": [ {"name": "GarageService", "parameters": [{"name": "name", "type": "String"}], "return_type": "GarageService", "visibility": "+"}, {"name": "maintenance", "parameters": [{"name": "vehicle", "type": "Vehicle"}], "return_type": "void", "visibility": "+"}, {"name": "repair", "parameters": [{"name": "vehicle", "type": "Vehicle"}, {"name": "issue", "type": "String"}], "return_type": "boolean", "visibility": "+"}, {"name": "inspectVehicle", "parameters": [{"name": "vehicle", "type": "Vehicle"}], "return_type": "InspectionReport", "visibility": "+"} ] } ], "relationships": [ { "from": "Car", "to": "Vehicle", "type": "inheritance" }, { "from": "Motorcycle", "to": "Vehicle", "type": "inheritance" }, { "from": "Car", "to": "Engine", "type": "composition", "multiplicity_from": "1", "multiplicity_to": "1" }, { "from": "Motorcycle", "to": "Engine", "type": "composition", "multiplicity_from": "1", "multiplicity_to": "1" }, { "from": "Car", "to": "TransmissionType", "type": "association", "multiplicity_from": "1", "multiplicity_to": "1" }, { "from": "Vehicle", "to": "FuelType", "type": "association", "multiplicity_from": "1", "multiplicity_to": "1" }, { "from": "GarageService", "to": "VehicleService", "type": "realization" }, { "from": "GarageService", "to": "Vehicle", "type": "dependency", "multiplicity_from": "1", "multiplicity_to": "*" } ] } Returns: str: The filepath to the generated PNG image file. """ try: if not json_input.strip(): return "Error: Empty input" data = json.loads(json_input) if 'classes' not in data: raise ValueError("Missing required field: classes") dot = graphviz.Digraph(comment='Class Diagram') dot.attr(rankdir='TB', bgcolor='white', pad='0.5', nodesep='0.8', ranksep='1.2', splines='ortho') dot.attr('node', shape='plaintext', fontname='Arial', fontsize='11') dot.attr('edge', color='black', fontname='Arial', fontsize='9', minlen='1') classes = data.get('classes', []) relationships = data.get('relationships', []) for cls in classes: class_name = cls.get('name') class_type = cls.get('type', 'class') attributes = cls.get('attributes', []) methods = cls.get('methods', []) if not class_name: continue html_label = '' if class_type == 'abstract': html_label += f'' elif class_type == 'interface': html_label += f'' elif class_type == 'enum': html_label += f'' else: html_label += f'' html_label += '
' if attributes: for attr in attributes: visibility = attr.get('visibility', '+') name = attr.get('name', '') attr_type = attr.get('type', '') is_static = attr.get('static', False) line = f"{visibility} {name}" if attr_type: line += f" : {attr_type}" if is_static: line = f"{line}" html_label += f'' else: html_label += '' html_label += '
' if methods: for method in methods: visibility = method.get('visibility', '+') name = method.get('name', '') parameters = method.get('parameters', []) return_type = method.get('return_type', 'void') is_static = method.get('static', False) is_abstract = method.get('abstract', False) line = f"{visibility} {name}(" if parameters: params = [] for param in parameters: param_name = param.get('name', '') param_type = param.get('type', '') params.append(f"{param_name}: {param_type}") line += ", ".join(params) line += f") : {return_type}" if is_static: line = f"{line}" if is_abstract: line = f"{line}" html_label += f'' else: html_label += '' html_label += '
<<abstract>>
{class_name}
<<interface>>
{class_name}
<<enumeration>>
{class_name}
{class_name}
{line}
{line}
' dot.node(class_name, f'<{html_label}>') for relationship in relationships: from_class = relationship.get('from') to_class = relationship.get('to') rel_type = relationship.get('type', 'association') label = relationship.get('label', '') multiplicity_from = relationship.get('multiplicity_from', '') multiplicity_to = relationship.get('multiplicity_to', '') if not from_class or not to_class: continue edge_attrs = { 'color': 'black' } if multiplicity_from: edge_attrs['taillabel'] = multiplicity_from if multiplicity_to: edge_attrs['headlabel'] = multiplicity_to if rel_type == 'inheritance': edge_attrs['arrowhead'] = 'empty' edge_attrs['color'] = 'black' elif rel_type == 'composition': edge_attrs['arrowhead'] = 'normal' edge_attrs['arrowtail'] = 'diamond' edge_attrs['dir'] = 'both' edge_attrs['color'] = 'black' elif rel_type == 'aggregation': edge_attrs['arrowhead'] = 'normal' edge_attrs['arrowtail'] = 'odiamond' edge_attrs['dir'] = 'both' edge_attrs['color'] = 'black' elif rel_type == 'realization': edge_attrs['arrowhead'] = 'empty' edge_attrs['style'] = 'dashed' edge_attrs['color'] = 'black' elif rel_type == 'dependency': edge_attrs['arrowhead'] = 'normal' edge_attrs['style'] = 'dashed' edge_attrs['color'] = 'black' else: edge_attrs['arrowhead'] = 'normal' edge_attrs['color'] = 'black' dot.edge(from_class, to_class, **edge_attrs) with NamedTemporaryFile(delete=False, suffix=f'.{output_format}') as tmp: dot.render(tmp.name, format=output_format, cleanup=True) return f"{tmp.name}.{output_format}" except json.JSONDecodeError: return "Error: Invalid JSON format" except Exception as e: return f"Error: {str(e)}"