X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/3200b733982abd6033a6b638d9c40ceda4948333..af8a54f4fbb7956ffbdb4e7b2aaddd7659de4f57:/src/catalogue/models/book.py?ds=inline

diff --git a/src/catalogue/models/book.py b/src/catalogue/models/book.py
index 10ba5b23b..eae021b18 100644
--- a/src/catalogue/models/book.py
+++ b/src/catalogue/models/book.py
@@ -471,10 +471,9 @@ class Book(models.Model):
             None, ContentFile(sync)
             )
 
-    
     def get_sync(self):
         if not self.has_sync_file():
-            return '[]'
+            return []
         with self.get_media('sync').first().file.open('r') as f:
             sync = f.read().split('\n')
         offset = float(sync[0])
@@ -484,8 +483,22 @@ class Book(models.Model):
                 continue
             start, end, elid = line.split()
             items.append([elid, float(start) + offset])
-        return json.dumps(items)
-    
+        return items
+
+    def sync_ts(self, ts):
+        elid = None
+        for cur_id, t in self.get_sync():
+            if ts >= t:
+                elid = cur_id
+            else:
+                break
+        return elid
+
+    def sync_elid(self, elid):
+        for cur_id, t in self.get_sync():
+            if cur_id == elid:
+                return t
+
     def has_audio_epub_file(self):
         return self.has_media("audio.epub")
 
@@ -988,6 +1001,12 @@ class Book(models.Model):
         elif isinstance(publisher, list):
             return ', '.join(publisher)
 
+    def recommended(self, limit=4):
+        books_qs = type(self).objects.filter(findable=True)
+        books_qs = books_qs.exclude(common_slug=self.common_slug).exclude(ancestor=self)
+        books = type(self).tagged.related_to(self, books_qs)[:limit]
+        return books
+
     @classmethod
     def tagged_top_level(cls, tags):
         """ Returns top-level books tagged with `tags`.