migdal/api.py

   1 # -*- coding: utf-8 -*-
   2 # This file is part of PrawoKultury, licensed under GNU Affero GPLv3 or later.
   3 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
   4 #
   5 from itertools import chain
   6 from migdal.models import Entry
   7 from migdal import app_settings
   8 from django.utils.translation import get_language
   9
  10
  11 def entry_list(entry_type=None, category=None, promobox=False,
  12             for_feed=False):
  13     lang = get_language()
  14     object_list = Entry.objects.filter(**{"published_%s" % lang: True})
  15
  16     if for_feed:
  17         object_list = object_list.order_by('-published_at_%s' % lang)
  18     else:
  19         object_list = object_list.order_by('-first_published_at')
  20
  21     if entry_type:
  22         object_list = object_list.filter(type=entry_type.db)
  23     else:
  24         object_list = object_list.filter(
  25             type__in=[t.db for t in app_settings.TYPES if t.on_main])
  26     if category:
  27         object_list = object_list.filter(categories=category)
  28
  29     if promobox:
  30         promo = list(object_list.filter(promo=True)[:promobox])
  31         #object_list = object_list.exclude(pk__in=[p.pk for p in promo])
  32
  33     object_list = object_list.filter(in_stream=True)
  34
  35     if promobox:
  36         object_list.promobox = promo
  37
  38     return object_list