From 8cc42e390ad7de21ff66ad014f6600344b411e88 Mon Sep 17 00:00:00 2001
From: Gustavo Henrique Santos Souza de Miranda <gustavohssmiranda@gmail.com>
Date: Thu, 3 Jul 2025 21:08:24 -0300
Subject: [PATCH] Added Hash capability to the model photo.py and also updated
 the service to create hashes and update hashes

---
 src/pilgrim/models/photo.py          |  4 +++-
 src/pilgrim/service/photo_service.py | 16 ++++++++++++++--
 2 files changed, 17 insertions(+), 3 deletions(-)

diff --git a/src/pilgrim/models/photo.py b/src/pilgrim/models/photo.py
index a164474..02fdcc0 100644
--- a/src/pilgrim/models/photo.py
+++ b/src/pilgrim/models/photo.py
@@ -16,6 +16,7 @@ class Photo(Base):
     name = Column(String)
     addition_date = Column(DateTime, default=datetime.now)
     caption = Column(String)
+    photo_hash = Column(String,name='hash')
     entries = relationship(
         "Entry",
         secondary=photo_entry_association,
@@ -24,7 +25,7 @@ class Photo(Base):
 
     fk_travel_diary_id = Column(Integer, ForeignKey("travel_diaries.id"),nullable=False)
 
-    def __init__(self, filepath, name, addition_date=None, caption=None, entries=None, fk_travel_diary_id=None, **kw: Any):
+    def __init__(self, filepath, name, photo_hash, addition_date=None, caption=None, entries=None, fk_travel_diary_id=None, **kw: Any):
         super().__init__(**kw)
         # Convert Path to string if needed
         if isinstance(filepath, Path):
@@ -34,6 +35,7 @@ class Photo(Base):
         self.name = name
         self.addition_date = addition_date if addition_date is not None else datetime.now()
         self.caption = caption
+        self.photo_hash = photo_hash
         self.entries = entries if entries is not None else []
         if fk_travel_diary_id is not None:
             self.fk_travel_diary_id = fk_travel_diary_id
diff --git a/src/pilgrim/service/photo_service.py b/src/pilgrim/service/photo_service.py
index df87902..ed1b2e3 100644
--- a/src/pilgrim/service/photo_service.py
+++ b/src/pilgrim/service/photo_service.py
@@ -1,6 +1,7 @@
 from pathlib import Path
 from typing import List
 from datetime import datetime
+import hashlib
 
 
 from pilgrim.models.photo import Photo
@@ -9,6 +10,12 @@ from pilgrim.models.travel_diary import TravelDiary
 class PhotoService:
     def __init__(self, session):
         self.session = session
+    def _hash_file(self,filepath):
+        hash_func = hashlib.new('sha3_384')
+        with open(filepath, 'rb') as f:
+            while chunk := f.read(8192):
+                hash_func.update(chunk)
+        return hash_func.hexdigest()
 
     def create(self, filepath: Path, name: str, travel_diary_id: int, caption=None, addition_date=None) -> Photo | None:
         travel_diary = self.session.query(TravelDiary).filter(TravelDiary.id == travel_diary_id).first()
@@ -27,7 +34,8 @@ class PhotoService:
             name=name, 
             caption=caption, 
             fk_travel_diary_id=travel_diary_id,
-            addition_date=addition_date
+            addition_date=addition_date,
+            photo_hash=self._hash_file(filepath)
         )
         self.session.add(new_photo)
         self.session.commit()
@@ -47,6 +55,7 @@ class PhotoService:
             original.name = photo_dst.name
             original.addition_date = photo_dst.addition_date
             original.caption = photo_dst.caption
+            original.photo_hash = original.photo_hash
             if photo_dst.entries and len(photo_dst.entries) > 0:
                 if original.entries is None:
                     original.entries = []
@@ -66,7 +75,10 @@ class PhotoService:
                 addition_date=excluded.addition_date,
                 caption=excluded.caption,
                 fk_travel_diary_id=excluded.fk_travel_diary_id,
-                id=excluded.id
+                id=excluded.id,
+                photo_hash=excluded.photo_hash,
+                entries=excluded.entries,
+
             )
             
             self.session.delete(excluded)