code.grnet.gr Git - astakos/blob - snf-astakos-app/astakos/im/templatetags/filters.py

   1 # Copyright 2011-2012 GRNET S.A. All rights reserved.
   2 #
   3 # Redistribution and use in source and binary forms, with or
   4 # without modification, are permitted provided that the following
   5 # conditions are met:
   6 #
   7 #   1. Redistributions of source code must retain the above
   8 #      copyright notice, this list of conditions and the following
   9 #      disclaimer.
  10 #
  11 #   2. Redistributions in binary form must reproduce the above
  12 #      copyright notice, this list of conditions and the following
  13 #      disclaimer in the documentation and/or other materials
  14 #      provided with the distribution.
  15 #
  16 # THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS
  17 # OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
  18 # WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  19 # PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR
  20 # CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  21 # SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  22 # LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
  23 # USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
  24 # AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  25 # LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
  26 # ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
  27 # POSSIBILITY OF SUCH DAMAGE.
  28 #
  29 # The views and conclusions contained in the software and
  30 # documentation are those of the authors and should not be
  31 # interpreted as representing official policies, either expressed
  32 # or implied, of GRNET S.A.
  33
  34 import calendar
  35 import datetime
  36
  37 from django import template
  38 from django.core.paginator import Paginator, InvalidPage
  39 from django.db.models.query import QuerySet
  40
  41 from astakos.im.settings import PAGINATE_BY
  42
  43 register = template.Library()
  44
  45 DELIM = ','
  46
  47 @register.filter
  48 def monthssince(joined_date):
  49     now = datetime.datetime.now()
  50     date = datetime.datetime(year=joined_date.year, month=joined_date.month, day=1)
  51     months = []
  52
  53     month = date.month
  54     year = date.year
  55     timestamp=calendar.timegm( date.utctimetuple() )
  56
  57     while date < now:
  58         months.append((year, month, timestamp))
  59
  60         if date.month < 12:
  61             month = date.month + 1
  62             year = date.year
  63         else:
  64             month = 1
  65             year = date.year + 1
  66
  67         date = datetime.datetime(year=year, month=month, day=1)
  68         timestamp=calendar.timegm( date.utctimetuple() )
  69
  70     return months
  71
  72 @register.filter
  73 def lookup(d, key):
  74     return d.get(key)
  75
  76
  77 @register.filter
  78 def dkeys(d):
  79     return d.keys()
  80
  81
  82 @register.filter
  83 def month_name(month_number):
  84     return calendar.month_name[month_number]
  85
  86
  87 @register.filter
  88 def todate(value, arg = ''):
  89     secs = int(value) / 1000
  90     return datetime.datetime.fromtimestamp(secs)
  91
  92
  93 @register.filter
  94 def rcut(value, chars = '/'):
  95     return value.rstrip(chars)
  96
  97 @register.filter
  98 def paginate(l, args):
  99     page, delim, sorting = args.partition(DELIM)
 100     if sorting:
 101         if isinstance(l, QuerySet):
 102             l = l.order_by(sorting)
 103         elif isinstance(l, list):
 104             default = ''
 105             if sorting.endswith('_date'):
 106                 default = datetime.datetime.utcfromtimestamp(0)
 107             l.sort(key=lambda i: getattr(i, sorting) \
 108                         if getattr(i, sorting) else default)
 109
 110     paginator = Paginator(l, PAGINATE_BY)
 111
 112     try:
 113         page_number = int(page)
 114     except ValueError:
 115         if page == 'last':
 116             page_number = paginator.num_pages
 117         else:
 118             page_number = 1
 119     page = globals()['page'] = paginator.page(page_number)
 120     return page
 121
 122 @register.filter
 123 def concat(str1, str2):
 124     if not str2:
 125         return str(str1)
 126     return '%s%s%s' % (str1, DELIM, str2)