Rapports: trie des rapports en WIP

[auf_rh_dae.git] / project / rh / templatetags / rapports.py

# -*- coding: utf-8 -*-

import datetime

from django.utils.encoding import smart_unicode
from django.template import Library
from django.utils.http import urlencode

from datamaster_modeles.models import Implantation, Region
from rh.models import TypeContrat


register = Library()


COMBLE_CHOICES = (('c', 'Comblé'), ('n', 'Vacant'))


@register.inclusion_tag('admin/filter.html', takes_context=True)
def filter_comble(context):
    return {'title': 'comblé',
            'choices': prepare_choices(COMBLE_CHOICES, 'comble', context)}


@register.inclusion_tag('admin/filter_select.html', takes_context=True)
def filter_region(context):
    return {'title': u"région",
            'choices': prepare_choices(Region.objects.values_list('id', 'nom'), 'implantation__region', context, remove=['pays', 'nord_sud'])}


@register.inclusion_tag('admin/filter_select.html', takes_context=True)
def filter_implantation(context):
    return {'title': u"implantation",
            'choices': prepare_choices(Implantation.objects.values_list('id', 'nom'), 'implantation', context)}


@register.inclusion_tag('admin/filter_select.html', takes_context=True)
def filter_region_contrat(context):
    return {'title': u"région",
            'choices': prepare_choices(Region.objects.values_list('id', 'nom'), 'dossier__poste__implantation__region', context, remove=['pays', 'nord_sud'])}


@register.inclusion_tag('admin/filter_select.html', takes_context=True)
def filter_implantation_contrat(context):
    return {'title': u"implantation",
            'choices': prepare_choices(Implantation.objects.values_list('id', 'nom'), 'dossier__poste__implantation', context)}


@register.inclusion_tag('admin/filter_select.html', takes_context=True)
def filter_type_contrat(context):
    return {'title': u"type de contrat",
            'choices': prepare_choices(TypeContrat.objects.values_list('id', 'nom'), 'type_contrat', context)}

@register.inclusion_tag('admin/filter_select.html', takes_context=True)
def filter_echeance_contrat(context):
    today = datetime.date.today()
    three_months = today + datetime.timedelta(days=3*30)
    six_months = today + datetime.timedelta(days=6*30)
    twelve_months = today + datetime.timedelta(days=12*30)

    field_name = 'date_fin'
    return {'title': u"échéance",
            'choices': prepare_choices_date(field_name, context, links=(
                ('Tous', {}),
                ('moins de 3 mois', {'%s__gte' % field_name: today.strftime('%Y-%m-%d'),
                                    '%s__lte' % field_name: three_months.strftime('%Y-%m-%d')}),
                ('3 à 6 mois', {'%s__gte' % field_name: three_months.strftime('%Y-%m-%d'),
                                '%s__lte' % field_name: six_months.strftime('%Y-%m-%d')}),
                ('6 à 12 mois', {'%s__gte' % field_name: six_months.strftime('%Y-%m-%d'),
                                '%s__lte' % field_name: twelve_months.strftime('%Y-%m-%d')}),
            )
    )}

@register.inclusion_tag('admin/filter_select.html', takes_context=True)
def filter_debut_contrat(context):
    year = datetime.date.today().timetuple()[0]
    this_year = datetime.date(year, 1, 1)
    next_year = datetime.date(year + 1, 1, 1)
    last_year = datetime.date(year - 1, 12,31)

    field_name = 'date_debut'
    return {'title': u"date début",
            'choices': prepare_choices_date(field_name, context, links=(
                ('Tous', {}),
                ('anées à venirs', {'%s__gte' % field_name: next_year.strftime('%Y-%m-%d')}),
                ('cette anneée', {'%s__gte' % field_name: this_year.strftime('%Y-%m-%d'),
                                '%s__lt' % field_name: next_year.strftime('%Y-%m-%d')}),
                ('ans passés', {'%s__lte' % field_name: last_year.strftime('%Y-%m-%d')}),
            )
    )}

@register.inclusion_tag('admin/filter_select.html', takes_context=True)
def filter_a_venir(context):
    today = datetime.date.today()
    year, month, day = datetime.date.today().timetuple()[:3]
    mois_prochain  = datetime.date(year+((month+1)/13), (month+1)%12, 1)
    this_month  = datetime.date(year, month, 1)

    field_name = 'date_debut'
    return {'title': u"à venir",
            'choices': prepare_choices_date(field_name, context, links=(
                ('Tous', {}),
                ('à venir', {'%s__gt' % field_name: today.strftime('%Y-%m-%d')}),
                ('à venir mois prochain', {'%s__gte' % field_name: mois_prochain.strftime('%Y-%m-%d')}),
                ('à venir ce mois', {'%s__gte' % field_name: this_month.strftime('%Y-%m-%d'),
                                    '%s__lt' % field_name: mois_prochain.strftime('%Y-%m-%d')}),
            )
    )}




@register.inclusion_tag('admin/filter_select.html', takes_context=True)
def filter_region_remun(context):
    return {'title': u"région",
            'choices': prepare_choices(Region.objects.values_list('id', 'nom'), 'dossiers__poste__implantation__region', context, remove=['pays', 'nord_sud'])}


@register.inclusion_tag('admin/filter_select.html', takes_context=True)
def filter_implantation_remun(context):
    return {'title': u"implantation",
            'choices': prepare_choices(Implantation.objects.values_list('id', 'nom'), 'dossiers__poste__implantation', context)}

@register.inclusion_tag('admin/table_header.html', takes_context=True)
def table_header(context, headers):
        return {'headers': headers}

def get_query_string(request, new_params=None, remove=None):
    if new_params is None: new_params = {}
    if remove is None: remove = []
    p = dict(request.GET.items())
    for r in remove:
        for k in p.keys():
            if k.startswith(r):
                del p[k]
    for k, v in new_params.items():
        if v is None:
            if k in p:
                del p[k]
        else:
            p[k] = v
    return '?%s' % urlencode(p)


def prepare_choices(choices, query_param, context, remove=[]):
    request = context['request']
    query_val = request.GET.get(query_param)
    result = [{'selected': query_val is None,
               'query_string': get_query_string(request, {}, [query_param] + remove),
               'display': 'Tout'}]
    for k, v in choices:
        result.append({'selected': smart_unicode(k) == query_val,
                       'query_string': get_query_string(request, {query_param: k}, remove),
                       'display': v})
    return result


def prepare_choices_date(field_name, context, links, remove=[]):
    request = context['request']
    params = request.GET
    field_generic = '%s__' % field_name
    date_params = dict([(k, v) for k, v in params.items() if k.startswith(field_generic)])


    
    result = []
    for title, param_dict in links:
        result.append({'selected': date_params == param_dict,
                       'query_string': get_query_string(request, param_dict, [field_generic]),
                       'display': title})
    return result

ORDER_VAR = 'o'
ORDER_TYPE_VAR = 'ot'

class SortHeaders:
    """
    Handles generation of an argument for the Django ORM's
    ``order_by`` method and generation of table headers which reflect
    the currently selected sort, based on defined table headers with
    matching sort criteria.

    Based in part on the Django Admin application's ``ChangeList``
    functionality.

    http://djangosnippets.org/snippets/308/
    """
    def __init__(self, request, headers, default_order_field=None,
            default_order_type='asc', additional_params=None):
        """
        request
            The request currently being processed - the current sort
            order field and type are determined based on GET
            parameters.

        headers
            A list of two-tuples of header text and matching ordering
            criteria for use with the Django ORM's ``order_by``
            method. A criterion of ``None`` indicates that a header
            is not sortable.

        default_order_field
            The index of the header definition to be used for default
            ordering and when an invalid or non-sortable header is
            specified in GET parameters. If not specified, the index
            of the first sortable header will be used.

        default_order_type
            The default type of ordering used - must be one of
            ``'asc`` or ``'desc'``.

        additional_params:
            Query parameters which should always appear in sort links,
            specified as a dictionary mapping parameter names to
            values. For example, this might contain the current page
            number if you're sorting a paginated list of items.
        """
        if default_order_field is None:
            for i, (header, query_lookup) in enumerate(headers):
                if query_lookup is not None:
                    default_order_field = i
                    break
        if default_order_field is None:
            raise AttributeError('No default_order_field was specified and none of the header definitions given were sortable.')
        if default_order_type not in ('asc', 'desc'):
            raise AttributeError('If given, default_order_type must be one of \'asc\' or \'desc\'.')
        if additional_params is None: additional_params = {}

        self.header_defs = headers
        self.additional_params = additional_params
        self.order_field, self.order_type = default_order_field, default_order_type

        # Determine order field and order type for the current request
        params = dict(request.GET.items())
        if ORDER_VAR in params:
            try:
                new_order_field = int(params[ORDER_VAR])
                if headers[new_order_field][1] is not None:
                    import pdb
                    pdb.set_trace()
                    self.order_field = params[ORDER_VAR]
            except (IndexError, ValueError):
                pass # Use the default
        if ORDER_TYPE_VAR in params and params[ORDER_TYPE_VAR] in ('asc', 'desc'):
            self.order_type = params[ORDER_TYPE_VAR]
        try:
            del params[ORDER_VAR]
        except KeyError:
            pass
        try:
            del params[ORDER_TYPE_VAR]
        except KeyError:
            pass
        self.additional_params = params

    def headers(self):
        """
        Generates dicts containing header and sort link details for
        all defined headers.
        """
        for i, (header, order_criterion) in enumerate(self.header_defs):
            th_classes = []
            new_order_type = 'asc'
            if i == self.order_field:
                th_classes.append('sorted %sending' % self.order_type)
                new_order_type = {'asc': 'desc', 'desc': 'asc'}[self.order_type]
            yield {
                'text': header,
                'sortable': order_criterion is not None,
                'url': self.get_query_string({ORDER_VAR: i, ORDER_TYPE_VAR: new_order_type}),
                'class_attr': (th_classes and ' '.join(th_classes) or ''),
            }

    def get_query_string(self, params):
        """
        Creates a query string from the given dictionary of
        parameters, including any additonal parameters which should
        always be present.
        """
        params.update(self.additional_params)
        if self.order_type == params[ORDER_TYPE_VAR] and self.order_field == params[ORDER_VAR]:
            params[ORDER_TYPE_VAR] = 'asc' if params[ORDER_TYPE_VAR] == "desc" else "desc"
        return '?%s' % '&'.join(['%s=%s' % (param, value) \
                                     for param, value in params.items()])

    def get_order_by(self):
        """
        Creates an ordering criterion based on the current order
        field and order type, for use with the Django ORM's
        ``order_by`` method.
        """
        return '%s%s' % (
            self.order_type == 'desc' and '-' or '',
            self.header_defs[self.order_field][1],
        )