Skip to content

Commit

Permalink
wip
Browse files Browse the repository at this point in the history
  • Loading branch information
ciur committed Oct 28, 2024
1 parent 0254da5 commit c9b087e
Show file tree
Hide file tree
Showing 2 changed files with 11 additions and 4 deletions.
10 changes: 7 additions & 3 deletions papermerge/core/db/models.py
Original file line number Diff line number Diff line change
Expand Up @@ -61,15 +61,15 @@ class User(Base):
back_populates="user", primaryjoin="User.id == Node.user_id"
)
home_folder_id: Mapped[UUID] = mapped_column(
ForeignKey("core_folder.basetreenode_ptr_id", deferrable=True)
ForeignKey("core_folder.basetreenode_ptr_id", deferrable=True), nullable=True
)
home_folder: Mapped["Folder"] = relationship(
primaryjoin="User.home_folder_id == Folder.id",
back_populates="user",
viewonly=True,
)
inbox_folder_id: Mapped[UUID] = mapped_column(
ForeignKey("core_folder.basetreenode_ptr_id", deferrable=True)
ForeignKey("core_folder.basetreenode_ptr_id", deferrable=True), nullable=True
)
inbox_folder: Mapped["Folder"] = relationship(
primaryjoin="User.home_folder_id == Folder.id",
Expand Down Expand Up @@ -103,7 +103,11 @@ class Node(Base):
user: Mapped["User"] = relationship(
back_populates="nodes", primaryjoin="User.id == Node.user_id"
)
user_id: Mapped[UUID] = mapped_column(ForeignKey("core_user.id", use_alter=True))
user_id: Mapped[UUID] = mapped_column(
ForeignKey(
"core_user.id", use_alter=True, name="core_basetreenode_user_id_fkey"
)
)
parent_id: Mapped[UUID] = mapped_column(
ForeignKey("core_basetreenode.id"), nullable=True
)
Expand Down
5 changes: 4 additions & 1 deletion papermerge/core/features/document_types/tests/conftest.py
Original file line number Diff line number Diff line change
Expand Up @@ -110,7 +110,10 @@ def _maker(username: str, is_superuser: bool = True):
lang="de",
user_id=user_id,
)
db_session.add_all([db_home, db_inbox, db_user])
db_session.add(db_inbox)
db_session.add(db_home)
db_session.add(db_user)
db_session.commit()
db_user.home_folder_id = db_home.id
db_user.inbox_folder_id = db_inbox.id
db_session.commit()
Expand Down

0 comments on commit c9b087e

Please sign in to comment.