For upgrades

[wolnelektury.git] / src / lesmianator / models.py

diff --git a/src/lesmianator/models.py b/src/lesmianator/models.py
index bcf271c..f4c8e3b 100644 (file)
--- a/src/lesmianator/models.py
+++ b/src/lesmianator/models.py
@@ -9,7 +9,7 @@ from random import randint
 
 from django.core.files.base import ContentFile
 from django.db import models
-from django.utils.timezone import utc
+from django.utils.timezone import now
 from django.contrib.auth.models import User
 from django.contrib.contenttypes.models import ContentType
 from django.contrib.contenttypes.fields import GenericForeignKey
@@ -38,7 +38,7 @@ class Poem(models.Model):
 
     def visit(self):
         self.view_count += 1
-        self.seen_at = datetime.utcnow().replace(tzinfo=utc)
+        self.seen_at = now()
         self.save()
 
     def __str__(self):
@@ -140,7 +140,8 @@ class Continuations(models.Model):
 
     @classmethod
     def for_userlist(cls, ul):
-        cont_tabs = (cls.get(b) for b in ul.get_books())
+        cont_tabs = [cls.get(b) for b in ul.get_books()]
+        if not cont_tabs: return {}
         return reduce(cls.join_conts, cont_tabs)
 
     @classmethod