source: trunk/sara_cmt/sara_cmt/cluster/filters.py @ 14194

#    This file is part of CMT, a Cluster Management Tool made at SARA.
#    Copyright (C) 2012  Sil Westerveld
#
#    This program is free software; you can redistribute it and/or modify
#    it under the terms of the GNU General Public License as published by
#    the Free Software Foundation; either version 2 of the License, or
#    (at your option) any later version.
#
#    This program is distributed in the hope that it will be useful,
#    but WITHOUT ANY WARRANTY; without even the implied warranty of
#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#    GNU General Public License for more details.
#
#    You should have received a copy of the GNU General Public License
#    along with this program; if not, write to the Free Software
#    Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA

from django.db import models
from django.contrib.admin.filterspecs import FilterSpec, ChoicesFilterSpec, DateFieldFilterSpec
#from django.utils.encoding import smart_unicode
from django.utils.translation import ugettext as _
from datetime import date, timedelta

class InSupportFilterSpec(DateFieldFilterSpec):
    """
        Adds filtering by warranty-status values in the admin filter sidebar.
        Set the in_support_filter filter in the model field attribute
        'in_support_filter'.
        
        my_model_field.in_support_filter = True
    """

    def __init__(self, f, request, params, model, model_admin):
        super(InSupportFilterSpec, self).__init__(f, request, params, model, model_admin)
        
        today = date.today()
        days_thirty = today + timedelta(30)
        self.links = (
            (_('All'), {}),
            (_('In Support'), {'%s__gte' % self.field.name: str(today), }),
            (_('Expiring in 30 days'), {'%s__gte' % self.field.name: str(today), '%s__lte' % self.field.name: str(days_thirty) }),
            (_('Expired'), {'%s__lte' % self.field.name: str(today), }),
        )

    def title(self):
        return 'warranty status'

# register the filter
FilterSpec.filter_specs.insert(0, (lambda f: getattr(f, 'in_support_filter', False), InSupportFilterSpec))