diff --git a/db.py b/db.py index 4fe2c31..c3447ce 100644 --- a/db.py +++ b/db.py @@ -28,6 +28,10 @@ class User(Base): back_populates='user', cascade='all, delete-orphan' ) + income_branches: Mapped[List['IncomeBranch']] = relationship( + back_populates='user', cascade='all, delete-orphan' + ) + schedule: Mapped[List['Schedule']] = relationship( back_populates='user', cascade='all, delete-orphan' ) @@ -71,7 +75,6 @@ class Profile(Base): user_id: Mapped[int] = mapped_column(ForeignKey('users.id')) scheme: Mapped[str] branch: Mapped[str] - upstream: Mapped[bool] json: Mapped[str] user: Mapped['User'] = relationship(back_populates='profiles') @@ -86,6 +89,17 @@ class Profile(Base): } +class IncomeBranch(Base): + __tablename__ = 'income_branches' + + id: Mapped[int] = mapped_column(primary_key=True) + user_id: Mapped[int] = mapped_column(ForeignKey('users.id')) + scheme: Mapped[str] + branch: Mapped[str] + + user: Mapped['User'] = relationship(back_populates='income_branches') + + class Schedule(Base): __tablename__ = 'schedule'