ajout de marquage traité pour les rapports de modif

[auf_rh_dae.git] / project / rh / historique.py

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

from django.core.urlresolvers import reverse
from django import forms
from django.db import models
from reversion.models import Revision

class ModificationTraite(models.Model):
    """
    Classe utilisée pour marquer les révisions de reversion traitée par
    les DRH.
    """
    revision = models.ForeignKey('reversion.Revision')
    

def format_revisions(qs):
    results = []
    for rev in qs:

        # Dans une révision, on loop dans les versions associées, pour trouver
        # un modèle qui dispose d'une admin (à priori, le parent) dans le cas 
        # des objets composés
        change_url = None
        history_url = None
        type = None
        objet = None
        for v in rev.version_set.all():

            try:
                change_url = reverse('admin:rh_%s_change' % v.content_type.model,
                        args=(v.object.id, ))
                history_url = reverse('admin:rh_%s_history' % v.content_type.model,
                        args=(v.object.id, ))
                type = v.content_type
                objet = v.object
                break
            except:
                continue

        results.append({
            'id': rev.id,
            'date_created': rev.date_created.strftime("%d-%m-%Y %H:%M:%S"),
            'short_date_created': rev.date_created.strftime("%d-%m-%Y"),
            'user': rev.user,
            'comment': rev.comment,
            'change_url': change_url,
            'history_url': history_url,
            'type': unicode(type).capitalize(),
            'objet': objet,

            })
    return results

def get_active_revisions():
    ids_revisions_traitees = [m.revision.id for m in
            ModificationTraite.objects.all()]
    return format_revisions(
            Revision.objects.
                exclude(user__isnull=True).
                exclude(id__in=ids_revisions_traitees).
                order_by('-date_created'))


class TodoForm(forms.Form):
    items = forms.MultipleChoiceField("Les items traités",
            widget=forms.CheckboxSelectMultiple)

    def __init__(self, *args, **kwargs):
        self.revisions = kwargs.pop('revisions')
        super(TodoForm, self).__init__(*args, **kwargs)
        self.fields['items'].choices = [(r['id'], r['id']) for r in self.revisions]