Added watch start/end autoset

Author: volbil

app/schemas.py

@@ -421,6 +421,8 @@ class WatchResponseBase(CustomModel):
     duration: int = Field(examples=[24])
     episodes: int = Field(examples=[3])
     score: int = Field(examples=[8])
+    start_date: datetime_pd | None
+    end_date: datetime_pd | None
 
 
 class ReadResponseBase(CustomModel):

app/watch/service.py

@@ -103,30 +103,41 @@ async def save_watch(
     # Create watch record if missing
     if not (watch := await get_anime_watch(session, anime, user)):
         log_type = constants.LOG_WATCH_CREATE
+
         watch = AnimeWatch()
+        watch.start_date = now
         watch.created = now
         watch.anime = anime
         watch.user = user
 
+        session.add(watch)
+
     log_before = {}
     log_after = {}
 
     # Set attributes from args to watch record
     for key, new_value in args.model_dump().items():
         # Here we add changes to log's before/after dicts only if there are changes
         old_value = getattr(watch, key)
+
         if old_value != new_value:
             log_before[key] = old_value
             setattr(watch, key, new_value)
             log_after[key] = new_value
 
+        if (
+            key == "status"
+            and new_value == constants.WATCH_COMPLETED
+            and watch.end_date is None
+        ):
+            watch.end_date = now
+
     # Calculate duration and update updated field
     watch.duration = calculate_watch_duration(watch)
     watch.updated = now
 
     # Update user last list update
     user.updated = now
-    session.add_all([watch, user])
 
     await session.commit()
 

tests/watch/test_watch_add.py

@@ -28,6 +28,10 @@ async def test_watch_add(
     )
 
     assert response.status_code == status.HTTP_200_OK
+
+    assert response.json()["start_date"] is not None
+    assert response.json()["end_date"] is None
+
     assert response.json()["anime"]["slug"] == "bocchi-the-rock-9e172d"
     assert response.json()["status"] == "watching"
     assert response.json()["duration"] == 230  # 10 episodes * 23 minutes
@@ -72,6 +76,10 @@ async def test_watch_add(
     )
 
     assert response.status_code == status.HTTP_200_OK
+
+    assert response.json()["start_date"] is not None
+    assert response.json()["end_date"] is not None
+
     assert response.json()["anime"]["slug"] == "bocchi-the-rock-9e172d"
     assert response.json()["status"] == "completed"
     assert response.json()["note"] == "Good anime!"