mirror of https://github.com/gmbrax/Pilgrim.git
48 lines
1.6 KiB
Python
48 lines
1.6 KiB
Python
from pathlib import Path
|
|
from typing import List
|
|
|
|
from pilgrim import Photo, TravelDiary
|
|
|
|
|
|
class PhotoService:
|
|
def __init__(self, session):
|
|
self.session = session
|
|
|
|
def create(self, filepath:Path, name:str, travel_diary_id, addition_date=None, caption=None, ) -> Photo | None:
|
|
travel_diary = self.session.query(TravelDiary).filter(TravelDiary.id == travel_diary_id).first()
|
|
if not travel_diary:
|
|
return None
|
|
new_photo = Photo(filepath, name, addition_date=addition_date, caption=caption)
|
|
self.session.add(new_photo)
|
|
self.session.commit()
|
|
self.session.refresh(new_photo)
|
|
|
|
return new_photo
|
|
def read_by_id(self, photo_id:int) -> Photo:
|
|
return self.session.query(Photo).get(photo_id)
|
|
|
|
def read_all(self) -> List[Photo]:
|
|
return self.session.query(Photo).all()
|
|
|
|
def update(self,photo_src:Photo,photo_dst:Photo) -> Photo | None:
|
|
original:Photo = self.read_by_id(photo_src.id)
|
|
if original:
|
|
original.filepath = photo_dst.filepath
|
|
original.name = photo_dst.name
|
|
original.addition_date = photo_dst.addition_date
|
|
original.caption = photo_dst.caption
|
|
original.entries.extend(photo_dst.entries)
|
|
self.session.commit()
|
|
self.session.refresh(original)
|
|
return original
|
|
return None
|
|
|
|
def delete(self, photo_src:Photo) -> Photo | None:
|
|
excluded = self.read_by_id(photo_src.id)
|
|
if excluded:
|
|
self.session.delete(excluded)
|
|
self.session.commit()
|
|
self.session.refresh(excluded)
|
|
return excluded
|
|
return None
|