diff --git a/django_dbml/core/renderer.py b/django_dbml/core/renderer.py index ccbb8cc..96b92b6 100644 --- a/django_dbml/core/renderer.py +++ b/django_dbml/core/renderer.py @@ -56,7 +56,7 @@ def render_table(self, table_name: str, table: TableDefinition) -> list[str]: blocks.append(f"Table {table_name} {{") if table.note: - blocks.append(" Note: '''\n{}'''\n".format(cleanup_docstring(table.note))) + blocks.append(" Note: '''\n{}\n'''\n".format(cleanup_docstring(table.note))) for field_name, field in table.fields.items(): blocks.append(f" {field_name} {field.type} {self.render_field_attributes(field)}".rstrip()) @@ -79,7 +79,7 @@ def render_field_attributes(self, field: FieldDefinition) -> str: # noqa: PLR09 if field.note: note = field.note.replace("'", '"') if "\n" in note: - attributes.append(f"note: '''\n{note}'''") + attributes.append(f"note: '''\n{note}\n'''") else: attributes.append(f"note: '''{note}'''")