From 05458b1fb9e5f560bdcef0998b091629c93562c4 Mon Sep 17 00:00:00 2001
From: Colin Powell <colin@unbl.ink>
Date: Fri, 1 Dec 2023 01:00:58 +0100
Subject: [PATCH] Add subtitle and update play time for webpages

---
 vrobbler/apps/webpages/models.py | 28 ++++++++++++++++------------
 1 file changed, 16 insertions(+), 12 deletions(-)

diff --git a/vrobbler/apps/webpages/models.py b/vrobbler/apps/webpages/models.py
index 9492519..2e01bc0 100644
--- a/vrobbler/apps/webpages/models.py
+++ b/vrobbler/apps/webpages/models.py
@@ -44,6 +44,19 @@ class WebPage(ScrobblableMixin):
     def get_absolute_url(self):
         return reverse("webpages:webpage_detail", kwargs={"slug": self.uuid})
 
+    @property
+    def estimated_time_to_read_in_seconds(self):
+        if not self.extract:
+            return 600
+
+        words_per_minute = getattr(settings, "READING_WORDS_PER_MINUTE", 200)
+        words = len(self.extract.split(" "))
+        return int(words / words_per_minute) * 60
+
+    @property
+    def subtitle(self):
+        return self.domain
+
     def _update_data_from_web(self, force=True):
         headers = {
             "headers": "Mozilla/5.0 (X11; Ubuntu; Linux x86_64; rv:51.0) Gecko/20100101 Firefox/51.0"
@@ -60,17 +73,11 @@ class WebPage(ScrobblableMixin):
         if not self.domain or force:
             self.domain = self.url.split("//")[-1].split("/")[0]
 
+        if not self.run_time_seconds or force:
+            self.run_time_seconds = self.estimated_time_to_read_in_seconds
+
         self.save(update_fields=["title", "domain", "extract"])
 
-    @property
-    def estimated_time_to_read_in_seconds(self):
-        if not self.extract:
-            return 600
-
-        words_per_minute = getattr(settings, "READING_WORDS_PER_MINUTE", 200)
-        words = len(self.extract.split(" "))
-        return int(words / words_per_minute) * 60
-
     @classmethod
     def find_or_create(cls, data_dict: Dict) -> "GeoLocation":
         """Given a data dict from an manual URL scrobble, does the heavy lifting of looking up
@@ -86,8 +93,5 @@ class WebPage(ScrobblableMixin):
 
         if not webpage:
             webpage = cls.objects.create(url=data_dict.get("url"))
-            webpage.run_time_seconds = (
-                webpage.estimated_time_to_read_in_seconds
-            )
             webpage._update_data_from_web()
         return webpage