fix DRH2

[auf_rh_dae.git] / project / menu.py

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

"""
This file was generated with the custommenu management command, it contains
the classes for the admin menu, you can customize this class as you want.

To activate your custom menu add the following to your settings.py::
    ADMIN_TOOLS_MENU = 'project.menu.CustomMenu'
"""

from django.core.urlresolvers import reverse
from django.utils.translation import ugettext_lazy as _

from admin_tools.menu import items, Menu
from project.rh.decorators import in_drh_or_admin

class CustomMenu(Menu):
    """
    Custom Menu for project admin site.
    """
    def __init__(self, **kwargs):
        Menu.__init__(self, **kwargs)
        self.children += [
            items.MenuItem(_('Dashboard'), reverse('admin:index')),
            #items.Bookmarks(),
            items.AppList(
                _('Applications'),
                exclude=('django.contrib.*', 'project.rh.models.EmployeProxy')
            ),
            items.AppList(
                _('Administration'),
                models=('django.contrib.*',)
            )
        ]

    def init_with_context(self, context):
        """
        Use this method if you need to access the request context.
        """
        request = context['request']
        if in_drh_or_admin(request.user):
            self.children += [
                items.MenuItem('Rapports',
                               children=[
                                   #items.MenuItem('Rapport des postes', reverse('rhr_postes')),
                                   items.MenuItem('Rapport des contrats', reverse('rhr_contrats')),
                                   items.MenuItem(u'Rapport des employés sans contrat', reverse('rhr_employe_sans_contrat')),
                                   #items.MenuItem('Rapport de rémunération', reverse('rhr_remuneration')),
                                   items.MenuItem('Rapport des postes par service', reverse('rhr_postes_service')),
                                   items.MenuItem('Rapport des postes par implantation', reverse('rhr_postes_implantation')),
                                   #items.MenuItem('Modelisation des postes', reverse('rhr_postes_modelisation')),
                                   #items.MenuItem('Rapport hiérarchique des postes', reverse('rhr_postes_hierarchie')),
                               ]
                              ),
                items.MenuItem('Organigrammes',
                    children=[
                        items.MenuItem('Organigramme par employé', reverse('admin:rh_employeproxy_changelist')),
                        items.MenuItem('Organigramme par service', reverse('admin:rh_serviceproxy_changelist')),
                        items.MenuItem('Organigramme par implantation', reverse('admin:rh_implantationproxy_changelist')),
                        items.MenuItem('Organigramme par bureau', reverse('admin:rh_regionproxy_changelist')),
                        ]
                    ),
            ]
        super(CustomMenu, self).init_with_context(context)