X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/1227b1201c537ea17e310f0faab4ead7cf0d25af..e852900e42eb0bde98daf7070d84816e44c3308e:/src/annoy/models.py diff --git a/src/annoy/models.py b/src/annoy/models.py index 3df56b2ea..9ed724c0e 100644 --- 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 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( @@ -60,8 +60,8 @@ class Banner(models.Model): 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) @@ -74,6 +74,14 @@ class DynamicTextInsert(models.Model): 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()