From 954b35b1d0023a1d2c3a82017d35f24d8b6ce6f8 Mon Sep 17 00:00:00 2001 From: Colin Powell Date: Sat, 15 Apr 2023 13:02:07 -0400 Subject: [PATCH] Set stop_timestamp when stopping scrobbles --- vrobbler/apps/scrobbles/models.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/vrobbler/apps/scrobbles/models.py b/vrobbler/apps/scrobbles/models.py index bee3494..ad70295 100644 --- a/vrobbler/apps/scrobbles/models.py +++ b/vrobbler/apps/scrobbles/models.py @@ -693,10 +693,17 @@ class Scrobble(TimeStampedModel): return scrobble def stop(self, force_finish=False) -> None: + self.stop_timestamp = timezone.now() if force_finish: self.played_to_completion = True self.in_progress = False - self.save(update_fields=["in_progress", "played_to_completion"]) + self.save( + update_fields=[ + "in_progress", + "played_to_completion", + "stop_timestamp", + ] + ) logger.info(f"stopping {self.id} from {self.source}") class_name = self.media_obj.__class__.__name__