diff --git a/src/pilgrim/models/photo.py b/src/pilgrim/models/photo.py index 971cb2a..88fc5a2 100644 --- a/src/pilgrim/models/photo.py +++ b/src/pilgrim/models/photo.py @@ -25,7 +25,8 @@ class Photo(Base): back_populates="photos" ) - fk_travel_diary_id = Column(Integer, ForeignKey("travel_diaries.id"),nullable=False) + fk_travel_diary_id = Column(Integer, ForeignKey("travel_diaries.id"), nullable=False) + travel_diary = relationship("TravelDiary", back_populates="photos") __table_args__ = ( Index('idx_photo_hash_diary', 'hash', 'fk_travel_diary_id'), ) diff --git a/src/pilgrim/models/travel_diary.py b/src/pilgrim/models/travel_diary.py index 0be4463..90a89b5 100644 --- a/src/pilgrim/models/travel_diary.py +++ b/src/pilgrim/models/travel_diary.py @@ -13,6 +13,7 @@ class TravelDiary(Base): name = Column(String, nullable=False) directory_name = Column(String, nullable=False, unique=True) entries = relationship("Entry", back_populates="travel_diary", cascade="all, delete-orphan") + photos = relationship("Photo", back_populates="travel_diary", cascade="all, delete-orphan") __table_args__ = ( UniqueConstraint('directory_name', name='uq_travel_diary_directory_name'),