mirror of https://github.com/gmbrax/Pilgrim.git
57 lines
1.9 KiB
Python
57 lines
1.9 KiB
Python
from datetime import datetime
|
|
from typing import List
|
|
|
|
from ..models.entry import Entry
|
|
from ..models.travel_diary import TravelDiary
|
|
from ..models.photo import Photo # ✨ Importe o modelo Photo
|
|
|
|
|
|
class EntryService:
|
|
def __init__(self, session):
|
|
self.session = session
|
|
|
|
# ✨ Modifique a assinatura para aceitar a lista de fotos
|
|
def create(self, travel_diary_id: int, title: str, text: str, date: datetime, photos: List[Photo]):
|
|
travel_diary = self.session.query(TravelDiary).filter(TravelDiary.id == travel_diary_id).first()
|
|
if not travel_diary:
|
|
return None
|
|
|
|
new_entry = Entry(title, text, date, travel_diary_id,photos=photos)
|
|
|
|
# ✨ Atribua a relação ANTES de adicionar e fazer o commit
|
|
new_entry.photos = photos
|
|
|
|
self.session.add(new_entry)
|
|
self.session.commit()
|
|
self.session.refresh(new_entry)
|
|
return new_entry
|
|
|
|
def read_by_id(self, entry_id: int) -> Entry:
|
|
entry = self.session.query(Entry).filter(Entry.id == entry_id).first()
|
|
return entry
|
|
|
|
def read_all(self) -> List[Entry]:
|
|
entries = self.session.query(Entry).all()
|
|
return entries
|
|
|
|
def update(self, entry_src: Entry, entry_dst: Entry) -> Entry | None:
|
|
original: Entry = self.read_by_id(entry_src.id)
|
|
if original:
|
|
original.title = entry_dst.title
|
|
original.text = entry_dst.text
|
|
original.date = entry_dst.date
|
|
original.fk_travel_diary_id = entry_dst.fk_travel_diary_id
|
|
original.photos = entry_dst.photos
|
|
self.session.commit()
|
|
self.session.refresh(original)
|
|
return original
|
|
return None
|
|
|
|
def delete(self, entry_src: Entry) -> Entry | None:
|
|
excluded = self.read_by_id(entry_src.id)
|
|
if excluded is not None:
|
|
self.session.delete(excluded)
|
|
self.session.commit()
|
|
return excluded
|
|
return None
|