[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

src/sqlacodegen/generators.py

@@ -782,7 +782,10 @@ def remove_table_from_metadata(self, table: Table) -> None:
             for fk in list(other_table.foreign_keys):
                 if fk.column.table.name == table.name:
                     for constraint in list(other_table.constraints):
-                        if isinstance(constraint, ForeignKeyConstraint) and fk in constraint.elements:
+                        if (
+                            isinstance(constraint, ForeignKeyConstraint)
+                            and fk in constraint.elements
+                        ):
                             other_table.constraints.discard(constraint)
                     fk.parent.foreign_keys.discard(fk)
                     fks_to_remove.append(fk)
@@ -813,8 +816,13 @@ def generate_models(self) -> list[Model]:
         # them normally
         links: defaultdict[str, list[Model]] = defaultdict(lambda: [])
         for table in self.metadata.sorted_tables:
-            if not table.primary_key and not self.generate_tables_for_relations_without_pk:
-                self.warnings.append(f"table {table.name!r} has no primary key and will not be included in the generated models")
+            if (
+                not table.primary_key
+                and not self.generate_tables_for_relations_without_pk
+            ):
+                self.warnings.append(
+                    f"table {table.name!r} has no primary key and will not be included in the generated models"
+                )
                 self.remove_table_from_metadata(table)
                 continue
 

tests/test_generator_sqlmodel.py

@@ -210,4 +210,7 @@ class TestTableWithPkey(SQLModel, table=True):
         """,
     )
     assert len(generator.warnings) == 1
-    assert generator.warnings[0] == "table 'test_table' has no primary key and will not be included in the generated models"
+    assert (
+        generator.warnings[0]
+        == "table 'test_table' has no primary key and will not be included in the generated models"
+    )