[#1823] Première phase pour régionnalisation

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

diff --git a/project/rh/lib.py b/project/rh/lib.py
index 5d6e935..e832a89 100644 (file)
--- a/project/rh/lib.py
+++ b/project/rh/lib.py
@@ -3,9 +3,40 @@
 from django.db import models
 from django.contrib import admin
 from django.conf import settings
+from django.db.models import Q
 from auf.django.metadata.admin import AUFMetadataAdminMixin, AUFMetadataInlineAdminMixin, AUF_METADATA_READONLY_FIELDS
 from project.rh import models as rh
 from forms import DossierForm, ContratForm
+from dae.utils import get_employe_from_user
+
+
+class ProtectRegionMixin(object):
+
+    def queryset(self, request):
+        qs = super(ProtectRegionMixin, self).queryset(request)
+
+        if request.user.is_superuser:
+            return qs
+
+        employe = get_employe_from_user(request.user)
+
+        q = Q(**{self.model.prefix_implantation: employe.implantation.region})
+        qs = qs.filter(q).distinct()
+        return qs
+
+    def has_change_permission(self, request, obj=None):
+        if request.user.is_superuser:
+            return True
+
+        if obj:
+            employe = get_employe_from_user(request.user)
+            if employe.implantation.region in obj.get_regions():
+                return True
+            else:
+                return False
+
+        return True
+
 
 # Inlines
 
@@ -44,6 +75,7 @@ class DossierCommentaireInline(AUFMetadataInlineAdminMixin, admin.TabularInline)
 
 class DossierPieceInline(admin.TabularInline):
     model = models.Model  # à remplacer dans admin.py
+    extra = 4
 
 
 class EmployeInline(admin.TabularInline):
@@ -57,6 +89,7 @@ class EmployeCommentaireInline(AUFMetadataInlineAdminMixin, admin.TabularInline)
 
 class EmployePieceInline(admin.TabularInline):
     model = models.Model  # à remplacer dans admin.py
+    extra = 4
 
 
 class EvenementInline(AUFMetadataInlineAdminMixin, admin.TabularInline):
@@ -98,7 +131,7 @@ class TypePosteInline(AUFMetadataInlineAdminMixin, admin.TabularInline):
 
 # Admins
 
-class AyantDroitAdmin(AUFMetadataAdminMixin, admin.ModelAdmin):
+class AyantDroitAdmin(AUFMetadataAdminMixin, ProtectRegionMixin, admin.ModelAdmin):
     """
     L'ajout d'un nouvel ayantdroit se fait dans l'admin de l'employé.
     """
@@ -186,7 +219,7 @@ class DeviseAdmin(AUFMetadataAdminMixin, admin.ModelAdmin):
     )
 
 
-class DossierAdmin(AUFMetadataAdminMixin, admin.ModelAdmin,):
+class DossierAdmin(AUFMetadataAdminMixin, ProtectRegionMixin, admin.ModelAdmin,):
     form = DossierForm
     alphabet_filter = 'employe__nom'
     search_fields = ('employe__nom', 'employe__prenom', 'poste__nom', 'poste__nom_feminin')
@@ -250,7 +283,7 @@ class DossierCommentaireAdmin(admin.ModelAdmin):
     pass
 
 
-class EmployeAdmin(AUFMetadataAdminMixin, admin.ModelAdmin):
+class EmployeAdmin(AUFMetadataAdminMixin, ProtectRegionMixin, admin.ModelAdmin):
     alphabet_filter = 'nom'
     search_fields = ('id', 'nom', 'prenom', 'nom_affichage', 'actif', )
     list_filter = ('actif', )
@@ -305,7 +338,7 @@ class FamilleEmploiAdmin(AUFMetadataAdminMixin, admin.ModelAdmin):
         )
 
 
-class OrganismeBstgAdmin(AUFMetadataAdminMixin, admin.ModelAdmin):
+class OrganismeBstgAdmin(AUFMetadataAdminMixin, ProtectRegionMixin, admin.ModelAdmin):
     search_fields = ('nom', )
     list_display = ('nom', 'type', 'pays', )
     inlines = (DossierROInline,)
@@ -316,7 +349,7 @@ class OrganismeBstgAdmin(AUFMetadataAdminMixin, admin.ModelAdmin):
         )
 
 
-class PosteAdmin(AUFMetadataAdminMixin, admin.ModelAdmin):
+class PosteAdmin(AUFMetadataAdminMixin, ProtectRegionMixin, admin.ModelAdmin):
     alphabet_filter = 'nom'
     search_fields = ('nom', 'implantation__code', 'implantation__nom', 'implantation__region__code', 'implantation__region__nom', )
     list_display = ('nom', 'implantation', 'service', 'type_poste', 'date_debut', 'date_fin', )
@@ -365,6 +398,7 @@ class PosteAdmin(AUFMetadataAdminMixin, admin.ModelAdmin):
             instance.save()
         formset.save_m2m()
 
+
 class PosteCommentaireAdmin(admin.ModelAdmin):
     pass