From 5e9f427072ce092a0b10dba14ca80f773b662cda Mon Sep 17 00:00:00 2001 From: Ivan Vazhenin Date: Mon, 13 Nov 2023 21:49:41 +0700 Subject: [PATCH] Update config schema for replication --- db.py | 16 +++++++++++++++- 1 file changed, 15 insertions(+), 1 deletion(-) 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'