Routeur: Ajout temporaire des modules workflow et dae en écriture

[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 django.db.models import Q
from reversion.models import Revision, Version


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 = []

    revisions = [r for r in qs]
    versions = Version.objects.select_related('revision', 'content_type').filter(
            Q(revision__in=revisions) &
            (Q(content_type__model='dossier') | Q(content_type__model='poste'))
            ).\
            order_by('-revision__date_created')
    #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
        #for v in rev.version_set.all():
    for v in versions:
        if v.revision in [r['id'] for r in results]:
            continue
        change_url = None
        history_url = None
        type = None
        objet = None

        try:
            klass = v.content_type.model_class()
            id = int(v.object_id)
            objet = klass.objects.get(id=id)
            change_url = reverse(
                'admin:%s_%s_change' % (
                    v.content_type.app_label, v.content_type.model),
                args=(id, ))
            history_url = reverse(
                'admin:%s_%s_history' % (
                    v.content_type.app_label, v.content_type.model),
                args=(id, ))
            type = "%s %s" % (
                v.content_type.app_label, v.content_type.model)
            rev = v.revision
            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': type,
                'objet': objet,

                })
        except Exception, e:
            continue

    return results


def get_active_revisions(limit=None):
    ids_revisions_traitees = [m.revision.id for m in
                              ModificationTraite.objects.select_related('revision').all()]
    revisions = Revision.objects.\
        filter(version__content_type__app_label="rh").\
        exclude(user__isnull=True).\
        exclude(id__in=ids_revisions_traitees).\
        order_by('-date_created').\
        distinct('id')
    
    if limit is not None:
        revisions = revisions[:limit]
    return format_revisions(revisions)


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]