diff --git a/src/sqlacodegen/generators.py b/src/sqlacodegen/generators.py index 7b4901a7..61a9d998 100644 --- a/src/sqlacodegen/generators.py +++ b/src/sqlacodegen/generators.py @@ -1127,7 +1127,9 @@ def render_models(self, models: list[Model]) -> str: def render_class(self, model: ModelClass) -> str: sections: list[str] = [] - + comments = self.render_table_comment(model) + if comments: + sections.append(comments) # Render class variables / special declarations class_vars: str = self.render_class_variables(model) if class_vars: @@ -1166,6 +1168,10 @@ def render_class_declaration(self, model: ModelClass) -> str: ) return f"class {model.name}({parent_class_name}):" + def render_table_comment(self, model: ModelClass) -> str: + if model.table.comment: + return f'"""{model.table.comment}"""' + def render_class_variables(self, model: ModelClass) -> str: variables = [f"__tablename__ = {model.table.name!r}"]