Skip to content

Commit

Permalink
add linebreaks
Browse files Browse the repository at this point in the history
  • Loading branch information
rbuffat committed Mar 4, 2023
1 parent a18d600 commit a46f018
Show file tree
Hide file tree
Showing 2 changed files with 135 additions and 65 deletions.
30 changes: 20 additions & 10 deletions tests/test_generator_dataclass2.py
Original file line number Diff line number Diff line change
Expand Up @@ -100,7 +100,8 @@ def test_onetomany_optional(self, generator: CodeGenerator) -> None:
from typing import List, Optional
from sqlalchemy import ForeignKey, Integer
from sqlalchemy.orm import DeclarativeBase, Mapped, MappedAsDataclass, mapped_column, relationship
from sqlalchemy.orm import DeclarativeBase, Mapped, MappedAsDataclass, mapped_column, \
relationship
class Base(MappedAsDataclass, DeclarativeBase):
pass
Expand All @@ -111,16 +112,19 @@ class SimpleContainers(Base):
id: Mapped[int] = mapped_column(Integer, primary_key=True)
simple_items: Mapped[List['SimpleItems']] = relationship('SimpleItems', back_populates='container')
simple_items: Mapped[List['SimpleItems']] = relationship('SimpleItems', \
back_populates='container')
class SimpleItems(Base):
__tablename__ = 'simple_items'
id: Mapped[int] = mapped_column(Integer, primary_key=True)
container_id: Mapped[Optional[int]] = mapped_column(ForeignKey('simple_containers.id'))
container_id: Mapped[Optional[int]] = \
mapped_column(ForeignKey('simple_containers.id'))
container: Mapped['SimpleContainers'] = relationship('SimpleContainers', back_populates='simple_items')
container: Mapped['SimpleContainers'] = relationship('SimpleContainers', \
back_populates='simple_items')
""",
)

Expand Down Expand Up @@ -148,7 +152,8 @@ def test_manytomany(self, generator: CodeGenerator) -> None:
from typing import List
from sqlalchemy import Column, ForeignKey, Integer, Table
from sqlalchemy.orm import DeclarativeBase, Mapped, MappedAsDataclass, mapped_column, relationship
from sqlalchemy.orm import DeclarativeBase, Mapped, MappedAsDataclass, mapped_column, \
relationship
class Base(MappedAsDataclass, DeclarativeBase):
pass
Expand All @@ -159,15 +164,17 @@ class SimpleContainers(Base):
id: Mapped[int] = mapped_column(Integer, primary_key=True)
item: Mapped[List['SimpleItems']] = relationship('SimpleItems', secondary='container_items', back_populates='container')
item: Mapped[List['SimpleItems']] = relationship('SimpleItems', \
secondary='container_items', back_populates='container')
class SimpleItems(Base):
__tablename__ = 'simple_items'
id: Mapped[int] = mapped_column(Integer, primary_key=True)
container: Mapped[List['SimpleContainers']] = relationship('SimpleContainers', secondary='container_items', back_populates='item')
container: Mapped[List['SimpleContainers']] = relationship('SimpleContainers', \
secondary='container_items', back_populates='item')
t_container_items = Table(
Expand Down Expand Up @@ -200,7 +207,8 @@ def test_named_foreign_key_constraints(self, generator: CodeGenerator) -> None:
from typing import List, Optional
from sqlalchemy import ForeignKeyConstraint, Integer
from sqlalchemy.orm import DeclarativeBase, Mapped, MappedAsDataclass, mapped_column, relationship
from sqlalchemy.orm import DeclarativeBase, Mapped, MappedAsDataclass, mapped_column, \
relationship
class Base(MappedAsDataclass, DeclarativeBase):
pass
Expand All @@ -211,7 +219,8 @@ class SimpleContainers(Base):
id: Mapped[int] = mapped_column(Integer, primary_key=True)
simple_items: Mapped[List['SimpleItems']] = relationship('SimpleItems', back_populates='container')
simple_items: Mapped[List['SimpleItems']] = relationship('SimpleItems', \
back_populates='container')
class SimpleItems(Base):
Expand All @@ -224,7 +233,8 @@ class SimpleItems(Base):
id: Mapped[int] = mapped_column(Integer, primary_key=True)
container_id: Mapped[Optional[int]] = mapped_column(Integer)
container: Mapped['SimpleContainers'] = relationship('SimpleContainers', back_populates='simple_items')
container: Mapped['SimpleContainers'] = relationship('SimpleContainers', \
back_populates='simple_items')
""",
)

Expand Down

0 comments on commit a46f018

Please sign in to comment.