Small refactor in messaging.

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

diff --git a/src/annoy/models.py b/src/annoy/models.py
index 3df56b2..9ed724c 100644 (file)
--- a/src/annoy/models.py
+++ b/src/annoy/models.py
@@ -40,9 +40,9 @@ class Banner(models.Model):
 
         if hasattr(request, 'annoy_banner_exempt'):
             return cls.objects.none()
 
         if hasattr(request, 'annoy_banner_exempt'):
             return cls.objects.none()

-        
+

         if settings.DEBUG:
         if settings.DEBUG:

-            assert place in PLACES, "Banner place `{}` must be defined in annoy.places.".format(place)
+            assert place in PLACES, f"Banner place `{place}` must be defined in annoy.places."

 
         n = now()
         banners = cls.objects.filter(
 
         n = now()
         banners = cls.objects.filter(


@@ -60,8 +60,8 @@ class Banner(models.Model):
             if Membership.is_active_for(request.user):
                 banners = banners.filter(show_members=True)
         return banners
             if Membership.is_active_for(request.user):
                 banners = banners.filter(show_members=True)
         return banners

-        
-        
+
+

 class DynamicTextInsert(models.Model):
     paragraphs = models.IntegerField(_('pararaphs'))
     url = models.CharField(max_length=1024)
 class DynamicTextInsert(models.Model):
     paragraphs = models.IntegerField(_('pararaphs'))
     url = models.CharField(max_length=1024)


@@ -74,6 +74,14 @@ class DynamicTextInsert(models.Model):
     def __str__(self):
         return str(self.paragraphs)
 
     def __str__(self):
         return str(self.paragraphs)
 

+    @classmethod
+    def get_all(cls, request):
+        Membership = apps.get_model('club', 'Membership')
+        if Membership.is_active_for(request.user) and not request.user.is_staff:
+            return cls.objects.none()
+        return cls.objects.all()
+
+

     def choose(self):
         return self.dynamictextinserttext_set.order_by('?').first()
 
     def choose(self):
         return self.dynamictextinserttext_set.order_by('?').first()