DRH2
[auf_rh_dae.git] / project / rh / admin.py
index 9de6b4d..9ec9387 100644 (file)
@@ -14,9 +14,9 @@ from auf.django.metadata.admin import AUFMetadataAdminMixin, \
 from forms import ContratForm, AyantDroitForm, EmployeAdminForm, AjaxSelect
 from dae.utils import get_employe_from_user
 from change_list import ChangeList
-from groups import grp_drh
+from groups import grp_correspondants_rh
+from decorators import in_drh_or_admin
 import models as rh
-
 import auf.django.references.models as ref
 
 class RegionProxy(ref.Region):
@@ -83,15 +83,10 @@ class LinkedInline(admin.options.InlineModelAdmin):
 class ProtectRegionMixin(object):
 
     def queryset(self, request):
-        from dae.workflow import grp_drh, grp_correspondants_rh
         qs = super(ProtectRegionMixin, self).queryset(request)
 
-        if request.user.is_superuser:
-            return qs
-
         user_groups = request.user.groups.all()
-
-        if grp_drh in user_groups:
+        if in_drh_or_admin(request.user):
             return qs
 
         if grp_correspondants_rh in user_groups:
@@ -106,7 +101,7 @@ class ProtectRegionMixin(object):
         user_groups = request.user.groups.all()
 
         # Lock pour autoriser uniquement les DRH à utiliser RH
-        if not request.user.is_superuser and not grp_drh in user_groups:
+        if not in_drh_or_admin(request.user):
             return False
 
         if len(user_groups) == 0 and not request.user.is_superuser: